internal 0.1.73__tar.gz → 0.1.75__tar.gz

{internal-0.1.73 → internal-0.1.75}/PKG-INFO

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

{internal-0.1.73 → internal-0.1.75}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "internal"
-version = "0.1.73"
+version = "0.1.75"
 description = ""
 authors = ["Ray <ray@cruisys.com>"]
 readme = "README.md"

{internal-0.1.73 → internal-0.1.75}/src/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-0.1.73 → internal-0.1.75}/src/internal/http/responses.py

@@ -9,7 +9,7 @@ from beanie import Document, Link
 async def async_response(data=None, message=None, code=None, page_no=None, total_num=None, page_size=None,
                          status_code=status.HTTP_200_OK):
     def _serialize(data):
-        if issubclass(data, Document):
+        if issubclass(type(data), Document):
             link_field_list = []
             for field_name in data.__annotations__:
                 field_type = getattr(data, field_name)

{internal-0.1.73 → internal-0.1.75}/src/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"