internal 1.1.35.3__py3-none-any.whl → 1.1.36__py3-none-any.whl

internal/model/base_model.py

@@ -28,7 +28,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, exclude: Dict = None):
+                                  ignore_cache: bool = False, fetch_links: bool = False,
+                                  exclude_field_list: List[str] = None):
         if not query:
             final_query = []
         else:
@@ -48,22 +49,55 @@ class InternalBaseDocument(Document):
                     continue
             final_sort.append((cls.id, pymongo.ASCENDING))
 
-        total_num = await cls.find(*final_query, ignore_cache=ignore_cache, fetch_links=fetch_links).sort(
-            *final_sort).count()
-
-        total_pages = (total_num + page_size - 1) // page_size
-
-        if total_pages == 0:
-            page_no = 1
-            page_data = []
+        if exclude_field_list:
+            # 當需要排除欄位時，使用 Motor 直接操作
+            collection = cls.get_motor_collection()
+            projection = {field: 0 for field in exclude_field_list}
+
+            # 建立查詢條件
+            mongo_query = {}
+            for q in final_query:
+                if hasattr(q, 'query'):
+                    mongo_query.update(q.query)
+
+            # 計算總數
+            total_num = await collection.count_documents(mongo_query)
+            total_pages = (total_num + page_size - 1) // page_size
+
+            if total_pages == 0:
+                page_no = 1
+                page_data = []
+            else:
+                page_no = max(1, min(page_no, total_pages))
+
+                # 執行分頁查詢
+                cursor = collection.find(mongo_query, projection).sort(final_sort).skip(
+                    (page_no - 1) * page_size).limit(page_size)
+                documents = await cursor.to_list(None)
+
+                # 轉換為 Pydantic 模型
+                page_data = []
+                for doc in documents:
+                    try:
+                        page_data.append(cls.model_validate(doc))
+                    except Exception as e:
+                        print(f"模型驗證失敗: {e}")
+                        continue
         else:
-            page_no = max(1, min(page_no, total_pages))
+            # 沒有排除欄位時使用 Beanie 的方法
+            total_num = await cls.find(*final_query, ignore_cache=ignore_cache, fetch_links=fetch_links).sort(
+                *final_sort).count()
+
+            total_pages = (total_num + page_size - 1) // page_size
 
-            _query = cls.find(*final_query, ignore_cache=ignore_cache, fetch_links=fetch_links)
-            if exclude:
-                _query = _query.project(exclude)
+            if total_pages == 0:
+                page_no = 1
+                page_data = []
+            else:
+                page_no = max(1, min(page_no, total_pages))
 
-            page_data = await _query.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).sort(
+                    *final_sort).limit(page_size).skip((page_no - 1) * page_size).to_list()
 
         return page_no, page_size, total_num, page_data
 
@@ -92,7 +126,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: Dict = None):
+                       fetch_links: bool = False, exclude_field_list: List[str] = None):
         if not query:
             final_query = []
         else:
@@ -112,11 +146,33 @@ class InternalBaseDocument(Document):
                     continue
             final_sort.append((cls.id, pymongo.ASCENDING))
 
-        _query = cls.find(*final_query, ignore_cache=ignore_cache, fetch_links=fetch_links)
-        if exclude:
-            _query = _query.project(exclude)
-
-        data = await _query.sort(*final_sort).to_list()
+        if exclude_field_list:
+            # 當需要排除欄位時，使用 Motor 直接操作
+            collection = cls.get_motor_collection()
+            projection = {field: 0 for field in exclude_field_list}
+
+            # 建立查詢條件
+            mongo_query = {}
+            for q in final_query:
+                if hasattr(q, 'query'):
+                    mongo_query.update(q.query)
+
+            # 執行查詢
+            cursor = collection.find(mongo_query, projection).sort(final_sort)
+            documents = await cursor.to_list(None)
+
+            # 轉換為 Pydantic 模型
+            data = []
+            for doc in documents:
+                try:
+                    data.append(cls.model_validate(doc))
+                except Exception as e:
+                    print(f"模型驗證失敗: {e}")
+                    continue
+        else:
+            # 沒有排除欄位時使用 Beanie 的方法
+            data = await cls.find(*final_query, ignore_cache=ignore_cache, fetch_links=fetch_links).sort(
+                *final_sort).to_list()
 
         return data
 

{internal-1.1.35.3.dist-info → internal-1.1.36.dist-info}/METADATA

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

{internal-1.1.35.3.dist-info → internal-1.1.36.dist-info}/RECORD

@@ -31,10 +31,10 @@ internal/interface/base_interface.py,sha256=3YaVjIgLi_pZpLk5SEIk8WVkuICM8qPavT8r
 internal/middleware/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 internal/middleware/log_request.py,sha256=OFwWnGfzXnllcQsBHolAWVsahXKoEhw3OQ1YWOm7RHM,2087
 internal/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-internal/model/base_model.py,sha256=V21OUoLAzeTrYRPV4cUsyGCOaDcS-OpXC49SCiHjAkg,5707
+internal/model/base_model.py,sha256=vZPDpqzH7Eigwdfw1BgCtr40mflpf41M2__-6DyUPpQ,7987
 internal/model/operate.py,sha256=QSM6yXYXpJMwrqkUGEWZLrEBaUgqHwVHY_Fi4S42hKc,3190
 internal/utils.py,sha256=wK1QumW1AaWE1ga2-WcDH2rtXRr2hSLwXzy-iI5dTzY,3962
 internal/validator_utils.py,sha256=CqjaVFoAu5MqvBG_AkTP-r7AliWawtUWB851USj4moI,1519
-internal-1.1.35.3.dist-info/METADATA,sha256=zfu2VUwh7UVMML8jA7DVB3O0IRfq3XY98SMkdpImJSo,941
-internal-1.1.35.3.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
-internal-1.1.35.3.dist-info/RECORD,,
+internal-1.1.36.dist-info/METADATA,sha256=RuaBTuS0BgSHQSI9ODnMW_JCkXcFCVQLq1pnwCGe4i0,939
+internal-1.1.36.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
+internal-1.1.36.dist-info/RECORD,,