internal 1.0.15__tar.gz → 1.0.17__tar.gz

{internal-1.0.15 → internal-1.0.17}/PKG-INFO

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

{internal-1.0.15 → internal-1.0.17}/pyproject.toml

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

{internal-1.0.15 → internal-1.0.17}/src/internal/http/responses.py

@@ -1,27 +1,39 @@
 import json
+import arrow
+
+from datetime import datetime
+
 import httpx
 
 from fastapi import status
 from fastapi.responses import JSONResponse
 from fastapi.encoders import jsonable_encoder
 from beanie import Document, Link
+from ..const import ARR_HUMAN_READ_FMT
 
 
 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):
+                         time_zone=None, time_format=ARR_HUMAN_READ_FMT, status_code=status.HTTP_200_OK):
     def _serialize(data):
         if issubclass(type(data), Document):
             link_field_list = []
+            datetime_field_list = []
             for field_name in data.__annotations__:
                 field_type = getattr(data, field_name)
                 if isinstance(field_type, Link):
                     link_field_list.append(field_name)
+                if isinstance(field_type, datetime):
+                    datetime_field_list.append(field_name)
 
             data = json.loads(data.model_dump_json(exclude="password", by_alias=False))
             if link_field_list:
                 for field_name in link_field_list:
                     if isinstance(data[field_name], dict) and "id" in data[field_name].keys():
                         data[field_name] = data[field_name]["id"]
+            for field_name in datetime_field_list:
+                if data[field_name]:
+                    data[field_name] = arrow.get(data[field_name]).to(time_zone).format(time_format)
+
         return data
 
     if isinstance(data, httpx.Response):