internal 0.1.31__tar.gz → 0.1.33__tar.gz

{internal-0.1.31 → internal-0.1.33}/PKG-INFO

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

{internal-0.1.31 → internal-0.1.33}/pyproject.toml

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

internal-0.1.33/src/internal/exception/internal_exception.py

@@ -0,0 +1,20 @@
+from fastapi import status
+from .base_exception import InternalBaseException
+
+
+class BadGatewayException(InternalBaseException):
+    code = "error_bad_gateway"
+    message = "Bad gateway"
+
+    def __init__(self, message: str = None, **kwargs):
+        _message = message or self.message
+        super().__init__(status.HTTP_502_BAD_GATEWAY, self.code, _message, **kwargs)
+
+
+class GatewayTimeoutException(InternalBaseException):
+    code = "error_gateway_timeout"
+    message = "Gateway timeout"
+
+    def __init__(self, message: str = None, **kwargs):
+        _message = message or self.message
+        super().__init__(status.HTTP_504_GATEWAY_TIMEOUT, self.code, _message, **kwargs)

{internal-0.1.31 → internal-0.1.33}/src/internal/http/responses.py

@@ -22,6 +22,9 @@ async def async_response(data=None, message=None, code=None, page_no=None, total
                         data[field_name] = data[field_name]["id"]
         return data
 
+    if isinstance(data, JSONResponse):
+        return data
+
     ret = {}
     if isinstance(data, list):
         data = [_serialize(d) for d in data]