internal 1.0.69__py3-none-any.whl → 1.0.70__py3-none-any.whl

internal/base_config.py

@@ -58,5 +58,8 @@ class BaseConfig(BaseSettings):
     SYSTEM_BRAND: str = "cruisys"
     SYSTEM_ORGANIZATION_ID: str = "cruisys"
 
+    # Redis URL
+    REDIS_URL: str = ""
+
     class Config:
         case_sensitive = False

internal/base_factory.py

@@ -13,7 +13,7 @@ from fastapi import FastAPI, status, Request, APIRouter
 from fastapi.exceptions import RequestValidationError
 from fastapi.middleware.cors import CORSMiddleware
 
-from . import database
+from . import database, cache_redis
 from .const import LOG_FMT, LOG_FMT_NO_DT, LOG_DT_FMT, DEFAULT_LOGGER_NAME
 from .exception.base_exception import InternalBaseException
 from .exception.internal_exception import BadGatewayException, GatewayTimeoutException
@@ -42,9 +42,10 @@ class BaseFactory(metaclass=ABCMeta):
 
     @abstractmethod
     async def init_state_cache(self, app, app_config):
-        """
-        Each factory should define what model it wants.
-        """
+        if app_config.REDIS_URL:
+            redis_cache = cache_redis.CacheRedis(app_config.REDIS_URL)
+            await redis_cache.connect()
+            app.state.redis = redis_cache.client
 
     async def init_state_scheduler_app(self, app, app_config):
         """
@@ -77,6 +78,8 @@ class BaseFactory(metaclass=ABCMeta):
             # code to execute when app is shutting down
             await mongodb.close()
             app.state.logger.info("Database disconnected")
+            await app.state.redis.close()
+            app.state.logger.info("Cache disconnected")
 
 
         if title is None:

internal/cache_redis.py

@@ -0,0 +1,23 @@
+import redis.asyncio as redis
+
+from motor.motor_asyncio import AsyncIOMotorClient
+from pymongo.errors import ServerSelectionTimeoutError
+
+from .exception.internal_exception import RedisInitializeFailureException, RedisConnectFailureException
+
+
+class CacheRedis:
+    def __init__(self, redis_url: str):
+        self.client = None
+        self.redis_url = redis_url
+
+    async def connect(self):
+        try:
+            connection_pool = redis.ConnectionPool.from_url(self.redis_url)
+            self.client = redis.Redis(connection_pool=connection_pool)
+        except Exception as e:
+            raise RedisInitializeFailureException()
+
+    async def close(self):
+        if self.client:
+            self.client.close()

internal/exception/internal_exception.py

@@ -63,3 +63,21 @@ class ParamValidationException(InternalBaseException):
     def __init__(self, message: str = None, **kwargs):
         _message = message or self.message
         super().__init__(status.HTTP_422_UNPROCESSABLE_ENTITY, self.code, _message, **kwargs)
+
+
+class RedisInitializeFailureException(InternalBaseException):
+    code = "error_redis_initialize"
+    message = "Redis initialize failure"
+
+    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 RedisConnectFailureException(InternalBaseException):
+    code = "error_redis_connect"
+    message = "Redis connect failure"
+
+    def __init__(self, message: str = None, **kwargs):
+        _message = message or self.message
+        super().__init__(status.HTTP_500_INTERNAL_SERVER_ERROR, self.code, _message, **kwargs)

{internal-1.0.69.dist-info → internal-1.0.70.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: internal
-Version: 1.0.69
+Version: 1.0.70
 Summary: 
 Author: Ray
 Author-email: ray@cruisys.com
@@ -14,6 +14,7 @@ Requires-Dist: dictdiffer (>=0.9.0,<0.10.0)
 Requires-Dist: fastapi (>=0.109.0,<0.110.0)
 Requires-Dist: httpx (>=0.26.0,<0.27.0)
 Requires-Dist: pydantic-settings (>=2.1.0,<3.0.0)
+Requires-Dist: redis (>=5.1.1,<6.0.0)
 Requires-Dist: watchtower (>=3.0.1,<4.0.0)
 Description-Content-Type: text/markdown
 

{internal-1.0.69.dist-info → internal-1.0.70.dist-info}/RECORD

@@ -1,6 +1,7 @@
 internal/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-internal/base_config.py,sha256=tuEu2m98igYt4ZxMKtWjaQrFokpc85cRQcAjGtf_cI0,1954
-internal/base_factory.py,sha256=7UUiolLWYSodc9PxqLHaZ6KqDYUU3kC4g07iVggSymU,10125
+internal/base_config.py,sha256=TykK9HNF9y3OcIBPgx0h3VfVW5XH7nClLReq1-cqZCg,1995
+internal/base_factory.py,sha256=EtiGGe75QHkqIsUUw2LxjqwHWbzadUkjQV-FRJRAetI,10349
+internal/cache_redis.py,sha256=Lh6gnLIMQi3Rdp4yaUxqgrN1ZCvmHtTQ1GTep0xqcrA,721
 internal/common_enum/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 internal/common_enum/contact_type.py,sha256=7QkTQ71UxpaT1YHI40FpjmLz3r-UbRU-sd0m5ajH1as,142
 internal/common_enum/description_type.py,sha256=kGwkFQh6dMnjDl6ipWYpA-qbNRrhEpnPq3NleTsNwwk,118
@@ -15,7 +16,7 @@ internal/database.py,sha256=1u53JnFJ0jvP7H4HPY2faA2ABdPQ8SlsZigYmQAPktI,2470
 internal/exception/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 internal/exception/app_exception.py,sha256=z7BGD0iw0VW5rczQb-sjnfxk1O_E6CPWMnBUfddU-0g,3467
 internal/exception/base_exception.py,sha256=8ctp525A3up9mZrHCq_-Jj0hWr_4NfZsXGg1e4ftVuQ,473
-internal/exception/internal_exception.py,sha256=rBi70sDFwoYc2NSnxXQ9LY6r2TnxA7ILC64xjpg5h4c,2296
+internal/exception/internal_exception.py,sha256=DDJg2tZtbcy4xlAhinJNzntSU1HcpGZ_XFP_perMzzo,2957
 internal/ext/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 internal/ext/amazon/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 internal/ext/amazon/aws/__init__.py,sha256=2YFjb-rHG1JaZGZiZffYDesgTAJjDshOqQbswOYzhP8,834
@@ -32,6 +33,6 @@ internal/model/base_model.py,sha256=NS_GZvw_IqU1aZPHPpyQs6A741O34ybmzVzBPm9nGWU,
 internal/model/operate.py,sha256=QSM6yXYXpJMwrqkUGEWZLrEBaUgqHwVHY_Fi4S42hKc,3190
 internal/utils.py,sha256=CXYZs4rDDe33Zac27YeknBE4uQN437ys1YAbW3Xeyow,1517
 internal/validator_utils.py,sha256=5ZLcNIgw1hvkYOj1f6gS9AYxe7Y3ufW9FyDxUS7FsMQ,1226
-internal-1.0.69.dist-info/METADATA,sha256=w23ULwNtqlrMXJgKCVhleJG-DJpjSZupmWRbFtK2HZo,625
-internal-1.0.69.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-internal-1.0.69.dist-info/RECORD,,
+internal-1.0.70.dist-info/METADATA,sha256=SwTZzaFsrL2Ppva2YJW57pg86M5ND5xnvFT38bJVB3s,663
+internal-1.0.70.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+internal-1.0.70.dist-info/RECORD,,