internal 0.1.53__tar.gz → 0.1.55__tar.gz

{internal-0.1.53 → internal-0.1.55}/PKG-INFO

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

{internal-0.1.53 → internal-0.1.55}/pyproject.toml

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

{internal-0.1.53 → internal-0.1.55}/src/internal/base_factory.py

@@ -44,6 +44,21 @@ class BaseFactory(metaclass=ABCMeta):
         Each factory should define what config it wants.
         """
 
+    async def process_exception(self, app, request: Request, exc):
+        if issubclass(type(exc), InternalBaseException):
+            detail = exc.detail
+            await async_response(data=detail.get("data"), code=detail.get("code"), message=detail.get("message"),
+                                 status_code=exc.status_code)
+        elif issubclass(type(exc), RequestValidationError):
+            return await async_response(data={"detail": exc.errors(), "body": exc.body},
+                                        code="error_unprocessable_entity", message="Validation failed",
+                                        status_code=status.HTTP_422_UNPROCESSABLE_ENTITY)
+        else:
+            app.state.logger.warn(f"Exception, request:{request.__dict__}, exc:{exc}")
+            app.state.logger.warn(traceback.format_exc())
+            return await async_response(code="error_internal_server", message="Internal server error",
+                                        status_code=status.HTTP_500_INTERNAL_SERVER_ERROR)
+
     def create_app(self, title=DEFAULT_APP_NAME) -> FastAPI:
         app = FastAPI(title=title, debug=self.get_app_config().DEBUG, version=self.API_VERSION)
 
@@ -85,24 +100,25 @@ class BaseFactory(metaclass=ABCMeta):
 
         @app.exception_handler(InternalBaseException)
         async def http_exception_handler(request: Request, exc: InternalBaseException):
-            print(exc)
-            detail = exc.detail
-            print(f"Exception detail: {detail}")
-            await async_response(data=detail.data, code=detail.code, message=detail.message,
-                                 status_code=exc.status_code)
+            await self.process_exception(app, request, exc)
+            # detail = exc.detail
+            # await async_response(data=detail.get("data"), code=detail.get("code"), message=detail.get("message"),
+            #                      status_code=exc.status_code)
 
         @app.exception_handler(RequestValidationError)
         async def validation_exception_handler(request: Request, exc: RequestValidationError):
-            return await async_response(data={"detail": exc.errors(), "body": exc.body},
-                                        code="error_unprocessable_entity", message="Validation failed",
-                                        status_code=status.HTTP_422_UNPROCESSABLE_ENTITY)
+            await self.process_exception(app, request, exc)
+            # return await async_response(data={"detail": exc.errors(), "body": exc.body},
+            #                             code="error_unprocessable_entity", message="Validation failed",
+            #                             status_code=status.HTTP_422_UNPROCESSABLE_ENTITY)
 
         @app.exception_handler(Exception)
         async def http_exception_handler(request: Request, exc: Exception):
-            app.state.logger.warn(f"Exception, request:{request.__dict__}, exc:{exc}")
-            app.state.logger.warn(traceback.format_exc())
-            return await async_response(code="error_internal_server", message="Internal server error",
-                                        status_code=status.HTTP_500_INTERNAL_SERVER_ERROR)
+            await self.process_exception(app, request, exc)
+            # app.state.logger.warn(f"Exception, request:{request.__dict__}, exc:{exc}")
+            # app.state.logger.warn(traceback.format_exc())
+            # return await async_response(code="error_internal_server", message="Internal server error",
+            #                             status_code=status.HTTP_500_INTERNAL_SERVER_ERROR)
 
         return app
 

{internal-0.1.53 → internal-0.1.55}/src/internal/http/responses.py

@@ -9,7 +9,6 @@ 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(type(data), Document):
         if isinstance(data, Document):
             link_field_list = []
             for field_name in data.__annotations__:
@@ -46,5 +45,5 @@ async def async_response(data=None, message=None, code=None, page_no=None, total
         }
     else:
         ret['data'] = data
-    print(f"Response: {ret}")
+
     return JSONResponse(status_code=status_code, content=ret)