internal 0.1.32__tar.gz → 0.1.34__tar.gz

{internal-0.1.32 → internal-0.1.34}/PKG-INFO

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

{internal-0.1.32 → internal-0.1.34}/pyproject.toml

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

internal-0.1.34/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.34/src/internal/http/requests.py

@@ -0,0 +1,23 @@
+import httpx
+
+from fastapi import FastAPI
+
+from ..exception.internal_exception import GatewayTimeoutException, BadGatewayException
+
+
+async def async_request(app: FastAPI, method, url, **kwargs):
+    timeout = httpx.Timeout(connect=app.state.config.REQUEST_CONN_TIMEOUT, read=app.state.config.REQUEST_READ_TIMEOUT,
+                            write=app.state.config.REQUEST_WRITE_TIMEOUT, pool=app.state.config.REQUEST_POOL_TIMEOUT)
+
+    try:
+        async with httpx.AsyncClient(timeout=timeout) as client:
+            response = await client.request(method, url, **kwargs)
+            return response
+    except httpx.TimeoutException as exc:
+        app.state.logger.error(
+            f"async_request(), TimeoutException, exc: {exc}, url: {url}, method: {method}, kwargs: {kwargs}")
+        raise GatewayTimeoutException()
+    except Exception as exc:
+        app.state.logger.error(
+            f"async_request(), Exception, exc: {exc}, url: {url}, method: {method}, kwargs: {kwargs}")
+        raise BadGatewayException()

internal-0.1.32/src/internal/http/requests.py

@@ -1,12 +0,0 @@
-import httpx
-
-from fastapi import FastAPI
-
-
-async def async_request(app: FastAPI, method, url, **kwargs):
-    timeout = httpx.Timeout(connect=app.state.config.REQUEST_CONN_TIMEOUT, read=app.state.config.REQUEST_READ_TIMEOUT,
-                            write=app.state.config.REQUEST_WRITE_TIMEOUT, pool=app.state.config.REQUEST_POOL_TIMEOUT)
-
-    async with httpx.AsyncClient(timeout=timeout) as client:
-        response = await client.request(method, url, **kwargs)
-        return response