internal 1.1.35.2__tar.gz → 1.1.35.3__tar.gz

{internal-1.1.35.2 → internal-1.1.35.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: internal
-Version: 1.1.35.2
+Version: 1.1.35.3
 Summary: 
 Author: Ray
 Author-email: ray@cruisys.com

{internal-1.1.35.2 → internal-1.1.35.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "internal"
-version = "1.1.35.2"
+version = "1.1.35.3"
 description = ""
 authors = [{ name = "Ray", email = "ray@cruisys.com" }]
 readme = "README.md"

{internal-1.1.35.2 → internal-1.1.35.3}/src/internal/model/base_model.py

@@ -1,4 +1,5 @@
 from datetime import datetime
+from idlelib import query
 
 from typing import List, Tuple, Union, Dict, Type, Optional, Any, Literal, Union, Set, Mapping
 import typing_extensions
@@ -27,8 +28,7 @@ class InternalBaseDocument(Document):
     @classmethod
     async def get_pagination_list(cls, app: FastAPI, query: list = None, sort: List[Tuple] = None,
                                   page_size: int = DEF_PAGE_SIZE, page_no: int = DEF_PAGE_NO,
-                                  ignore_cache: bool = False, fetch_links: bool = False,
-                                  exclude_fields: List[str] = None):
+                                  ignore_cache: bool = False, fetch_links: bool = False, exclude: Dict = None):
         if not query:
             final_query = []
         else:
@@ -59,15 +59,11 @@ class InternalBaseDocument(Document):
         else:
             page_no = max(1, min(page_no, total_pages))
 
-            # 建立查詢物件用於取得分頁資料
-            data_query = cls.find(*final_query, ignore_cache=ignore_cache, fetch_links=fetch_links)
+            _query = cls.find(*final_query, ignore_cache=ignore_cache, fetch_links=fetch_links)
+            if exclude:
+                _query = _query.project(exclude)
 
-            # 如果有指定要排除的欄位，使用 project() 方法
-            if exclude_fields:
-                projection = {field: 0 for field in exclude_fields}
-                data_query = data_query.project(projection)
-
-            page_data = await data_query.sort(*final_sort).limit(page_size).skip((page_no - 1) * page_size).to_list()
+            page_data = await _query.sort(*final_sort).limit(page_size).skip((page_no - 1) * page_size).to_list()
 
         return page_no, page_size, total_num, page_data
 
@@ -96,7 +92,7 @@ class InternalBaseDocument(Document):
 
     @classmethod
     async def get_list(cls, app: FastAPI, query: list = None, sort: List[Tuple] = None, ignore_cache: bool = False,
-                       fetch_links: bool = False, exclude_fields: List[str] = None):
+                       fetch_links: bool = False, exclude: Dict = None):
         if not query:
             final_query = []
         else:
@@ -116,15 +112,12 @@ class InternalBaseDocument(Document):
                     continue
             final_sort.append((cls.id, pymongo.ASCENDING))
 
-        # 建立查詢物件
-        query_obj = cls.find(*final_query, ignore_cache=ignore_cache, fetch_links=fetch_links)
+        _query = cls.find(*final_query, ignore_cache=ignore_cache, fetch_links=fetch_links)
+        if exclude:
+            _query = _query.project(exclude)
 
-        # 如果有指定要排除的欄位，使用 project() 方法
-        if exclude_fields:
-            projection = {field: 0 for field in exclude_fields}
-            query_obj = query_obj.project(projection)
+        data = await _query.sort(*final_sort).to_list()
 
-        data = await query_obj.sort(*final_sort).to_list()
         return data
 
     def model_dump(