internal 0.1.74__py3-none-any.whl → 0.1.75__py3-none-any.whl

internal/exception/internal_exception.py

@@ -36,3 +36,12 @@ class DatabaseConnectFailureException(InternalBaseException):
     def __init__(self, message: str = None, **kwargs):
         _message = message or self.message
         super().__init__(status.HTTP_500_INTERNAL_SERVER_ERROR, self.code, _message, **kwargs)
+
+
+class NoChangeException(InternalBaseException):
+    code = "error_no_change"
+    message = "Document no change"
+
+    def __init__(self, message: str = None, **kwargs):
+        _message = message or self.message
+        super().__init__(status.HTTP_200_OK, self.code, _message, **kwargs)

internal/model/base_model.py

@@ -1,13 +1,16 @@
 from datetime import datetime
 from enum import Enum
-from typing import List, Tuple
+from typing import List, Tuple, Union, Dict, Type
 
+import arrow
 import pymongo
-from fastapi import FastAPI
 from beanie import Document
-from pydantic import Field
+from fastapi import FastAPI
+from pydantic import Field, BaseModel
 
+from .operate import Operate
 from ..const import DEF_PAGE_SIZE, DEF_PAGE_NO
+from ..exception.internal_exception import NoChangeException
 
 
 class InternalBaseDocument(Document):
@@ -16,7 +19,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,
+                                  page_size: int = DEF_PAGE_SIZE, page_no: int = DEF_PAGE_NO,
+                                  ignore_cache: bool = False,
                                   fetch_links: bool = False):
         if not query:
             final_query = []
@@ -46,6 +50,28 @@ class InternalBaseDocument(Document):
 
         return page_no, page_size, total_num, page_data
 
+    async def update_wrap(self, schema: Union[Dict, Type[BaseModel]]) -> Tuple[Operate, 'InternalBaseDocument']:
+        if not issubclass(type(schema), dict) and not issubclass(type(schema), BaseModel):
+            raise TypeError("Schema must be a subclass of BaseModel or dict")
+
+        original_model = self.model_copy(deep=True)
+        delta_dict = schema
+        if issubclass(type(schema), BaseModel):
+            delta_dict = schema.model_dump()
+
+        for key, value in delta_dict.items():
+            if value is not None and hasattr(self, key):
+                setattr(self, key, value)
+
+        operate = await Operate.generate_operate(original_model.model_dump(), self.model_dump())
+        if not operate.add and not operate.remove and not operate.change:
+            raise NoChangeException()
+
+        self.update_time = arrow.utcnow().datetime
+
+        await self.save()
+        return operate, self
+
 
 class OrderTypeEnum(str, Enum):
     ASC = "ASC"

{internal-0.1.74.dist-info → internal-0.1.75.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: internal
-Version: 0.1.74
+Version: 0.1.75
 Summary: 
 Author: Ray
 Author-email: ray@cruisys.com

{internal-0.1.74.dist-info → internal-0.1.75.dist-info}/RECORD

@@ -8,7 +8,7 @@ internal/const.py,sha256=U1S9r7bjtHgad2oYQoHO19s8D4V0WDUG-L-haV4UaIw,1025
 internal/database.py,sha256=jnvcj3HVM7VLXAQQq4hCkxUufL6BmI01tbsJw671S8U,838
 internal/exception/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 internal/exception/base_exception.py,sha256=U0ZXx1FOXydAnQQImznkeAdsKVC_YaukPWvJdWqEBfg,363
-internal/exception/internal_exception.py,sha256=zv9lstTRJv-OpSjWpuALb8spvN2ka8AdmhpMX_8rMMg,1362
+internal/exception/internal_exception.py,sha256=byaaH5DBZTp-M8YMzGicOS2rHMg7S4zS7Two-1r7tj8,1651
 internal/ext/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 internal/ext/amazon/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 internal/ext/amazon/aws/__init__.py,sha256=2YFjb-rHG1JaZGZiZffYDesgTAJjDshOqQbswOYzhP8,834
@@ -19,9 +19,9 @@ internal/http/responses.py,sha256=IVY-piFxA9iCUx6p_Y7OCgbx0-yJxc_laGDqrhZ4Z6I,15
 internal/interface/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 internal/interface/base_interface.py,sha256=R5MmO5oa5-ORPV_adkegLVCwsVio2zFOuQFGiby3khs,340
 internal/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-internal/model/base_model.py,sha256=gNbixWuGiK8Tbe78wHuLneZ8p3fwMwQRKurBDrGoh6s,1874
+internal/model/base_model.py,sha256=SeJtYynw1Uq2_tW95evqVbVFz8OpaLAgQfKkcPPdVzg,2962
 internal/model/operate.py,sha256=luEoP_Asvso_11qz1SAE1Kn5KxAnDRptM0VylC5tNik,1493
 internal/utils.py,sha256=0SubS0iUhDvjSX1F4TykasA5-enYJzt2VH-f7_0BnjI,1509
-internal-0.1.74.dist-info/METADATA,sha256=dmEKQ9ZufDqnpMlq_rPOtlfXm2K1jPMrag_06N5Nf3I,625
-internal-0.1.74.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-internal-0.1.74.dist-info/RECORD,,
+internal-0.1.75.dist-info/METADATA,sha256=CLrPlXt9weplJMudUQKQJH24wTgaovW_v_EPOlq-alo,625
+internal-0.1.75.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+internal-0.1.75.dist-info/RECORD,,