maleo-foundation 0.0.21__tar.gz → 0.0.22__tar.gz

{maleo_foundation-0.0.21 → maleo_foundation-0.0.22}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: maleo_foundation
-Version: 0.0.21
+Version: 0.0.22
 Summary: Foundation package for Maleo
 Author-email: Agra Bima Yuda <agra@nexmedis.com>
 License: MIT

maleo_foundation-0.0.22/maleo_foundation/query.py

@@ -0,0 +1,81 @@
+from sqlalchemy import Column, Table
+from sqlalchemy.orm import Query
+from sqlalchemy.orm.attributes import InstrumentedAttribute
+from sqlalchemy.sql.expression import or_, asc, cast, desc
+from sqlalchemy.types import DATE, String, TEXT, TIMESTAMP
+from typing import Optional, Type
+from maleo_foundation.db import DatabaseManager
+from maleo_foundation.models.enums import BaseEnums
+from maleo_foundation.models.schemas.general import BaseGeneralSchemas
+
+class BaseQuery:
+    @staticmethod
+    def filter_ids(query:Query, table:Type[DatabaseManager.Base], column:str, ids:Optional[list[int]]) -> Query:
+        if ids is not None:
+            column_attr = getattr(table, column, None)
+            if column_attr:
+                id_filters = [column_attr == id for id in ids]
+                query = query.filter(or_(*id_filters))
+        return query
+    
+    @staticmethod
+    def filter_timestamps(query:Query, table:Type[DatabaseManager.Base], date_filters:list[BaseGeneralSchemas.DateFilter]) -> Query:
+        if date_filters and len(date_filters) > 0:
+            for date_filter in date_filters:
+                try:
+                    table:Table = table.__table__
+                    column:Column = table.columns[date_filter.name]
+                    column_attr:InstrumentedAttribute = getattr(table, date_filter.name)
+                    if isinstance(column.type, (TIMESTAMP, DATE)):
+                        if date_filter.from_date and date_filter.to_date:
+                            query = query.filter(column_attr.between(date_filter.from_date, date_filter.to_date))
+                        elif date_filter.from_date:
+                            query = query.filter(column_attr >= date_filter.from_date)
+                        elif date_filter.to_date:
+                            query = query.filter(column_attr <= date_filter.to_date)
+                except KeyError:
+                    continue
+        return query
+    
+    @staticmethod
+    def filter_statuses(query:Query, table:Type[DatabaseManager.Base], statuses:Optional[list[BaseEnums.StatusType]]) -> Query:
+        if statuses is not None:
+            status_filters = [table.status == status for status in statuses]
+            query = query.filter(or_(*status_filters))
+        return query
+    
+    @staticmethod
+    def filter_search(query:Query, table:Type[DatabaseManager.Base], search:Optional[str]) -> Query:
+        if search:
+            search_term = f"%{search}%" #* Use wildcard for partial matching
+            search_filters = []
+            sqla_table:Table = table.__table__
+            for name, attr in vars(table).items():
+                try:
+                    column: Column = sqla_table.columns[name]
+                    if not isinstance(attr, InstrumentedAttribute):
+                        continue
+                    if isinstance(column.type, (String, TEXT)):
+                        search_filters.append(cast(attr, TEXT).ilike(search_term))
+                except KeyError:
+                    continue
+            if search_filters:
+                query = query.filter(or_(*search_filters))
+        return query
+    
+    @staticmethod
+    def sort(query:Query, table:Type[DatabaseManager.Base], sort_columns:list[BaseGeneralSchemas.SortColumn]) -> Query:
+        for sort_column in sort_columns:
+            try:
+                sort_col = getattr(table, sort_column.name)
+                sort_col = asc(sort_col) if sort_column.order.value.lower() == "asc" else desc(sort_col)
+                query = query.order_by(sort_col)
+            except AttributeError:
+                continue
+        return query
+
+    @staticmethod
+    def paginate(query:Query, page:int, limit:int) -> Query:
+        offset:int = (page - 1) * limit #* Calculate offset based on page
+        query = query.limit(limit=limit).offset(offset=offset)
+        return query

{maleo_foundation-0.0.21 → maleo_foundation-0.0.22}/maleo_foundation.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: maleo_foundation
-Version: 0.0.21
+Version: 0.0.22
 Summary: Foundation package for Maleo
 Author-email: Agra Bima Yuda <agra@nexmedis.com>
 License: MIT

{maleo_foundation-0.0.21 → maleo_foundation-0.0.22}/maleo_foundation.egg-info/SOURCES.txt

@@ -3,6 +3,7 @@ pyproject.toml
 maleo_foundation/__init__.py
 maleo_foundation/constants.py
 maleo_foundation/controller.py
+maleo_foundation/query.py
 maleo_foundation.egg-info/PKG-INFO
 maleo_foundation.egg-info/SOURCES.txt
 maleo_foundation.egg-info/dependency_links.txt

{maleo_foundation-0.0.21 → maleo_foundation-0.0.22}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "maleo_foundation"
-version = "0.0.21"
+version = "0.0.22"
 description = "Foundation package for Maleo"
 authors = [
     { name = "Agra Bima Yuda", email = "agra@nexmedis.com" }