internal 1.1.35__tar.gz → 1.1.35.1__tar.gz

{internal-1.1.35 → internal-1.1.35.1}/PKG-INFO

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

{internal-1.1.35 → internal-1.1.35.1}/pyproject.toml

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

{internal-1.1.35 → internal-1.1.35.1}/src/internal/model/base_model.py

@@ -27,8 +27,8 @@ 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):
+                                  ignore_cache: bool = False, fetch_links: bool = False,
+                                  exclude_fields: List[str] = None):
         if not query:
             final_query = []
         else:
@@ -48,6 +48,10 @@ class InternalBaseDocument(Document):
                     continue
             final_sort.append((cls.id, pymongo.ASCENDING))
 
+        pymongo_kwargs = {}
+        if exclude_fields:
+            pymongo_kwargs['projection'] = {field: 0 for field in exclude_fields}
+
         total_num = await cls.find(*final_query, ignore_cache=ignore_cache, fetch_links=fetch_links).sort(
             *final_sort).count()
 
@@ -59,8 +63,9 @@ class InternalBaseDocument(Document):
         else:
             page_no = max(1, min(page_no, total_pages))
 
-            page_data = await cls.find(*final_query, ignore_cache=ignore_cache, fetch_links=fetch_links).sort(
-                *final_sort).limit(page_size).skip((page_no - 1) * page_size).to_list()
+            page_data = await cls.find(*final_query, ignore_cache=ignore_cache, fetch_links=fetch_links,
+                                       **pymongo_kwargs).sort(*final_sort).limit(page_size).skip(
+                (page_no - 1) * page_size).to_list()
 
         return page_no, page_size, total_num, page_data
 
@@ -89,7 +94,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):
+                       fetch_links: bool = False, exclude_fields: List[str] = None):
         if not query:
             final_query = []
         else:
@@ -109,8 +114,13 @@ class InternalBaseDocument(Document):
                     continue
             final_sort.append((cls.id, pymongo.ASCENDING))
 
-        data = await cls.find(*final_query, ignore_cache=ignore_cache, fetch_links=fetch_links).sort(
-            *final_sort).to_list()
+        # 準備 pymongo 的 projection 參數
+        pymongo_kwargs = {}
+        if exclude_fields:
+            pymongo_kwargs['projection'] = {field: 0 for field in exclude_fields}
+
+        data = await cls.find(*final_query, ignore_cache=ignore_cache, fetch_links=fetch_links,
+                              **pymongo_kwargs).sort(*final_sort).to_list()
 
         return data