internal 1.0.47__py3-none-any.whl → 1.0.49__py3-none-any.whl

internal/base_factory.py

@@ -3,6 +3,7 @@ import logging.handlers
 import os
 import traceback
 from abc import ABCMeta, abstractmethod
+from contextlib import asynccontextmanager
 from functools import lru_cache
 
 import dotenv
@@ -44,6 +45,12 @@ class BaseFactory(metaclass=ABCMeta):
         Each factory should define what model it wants.
         """
 
+    async def init_state_scheduler_app(self, app, app_config):
+        """
+        Each factory should define what model it wants.
+        """
+        pass
+
     @abstractmethod
     @lru_cache()
     def get_app_config(self):
@@ -52,15 +59,31 @@ class BaseFactory(metaclass=ABCMeta):
         """
 
     def create_app(self, title=None) -> FastAPI:
+
+        @asynccontextmanager
+        async def lifespan(app: FastAPI):
+            await mongodb.connect()
+            document_model_list = await self.get_document_model_list()
+            await init_beanie(database=app.state.db.get_database(), document_models=document_model_list)
+            app.state.logger.info("Database connected")
+            await self.init_state_cache(app, self.get_app_config())
+            app.state.logger.info("Initialization state cache done")
+            await self.init_state_scheduler_app(app, self.get_app_config())
+            app.state.logger.info("Initialization state scheduler app done")
+            yield
+            await mongodb.close()
+            app.state.logger.info("Database disconnected")
+
         if title is None:
             title = self.DEFAULT_APP_NAME
 
         if self.get_app_config().DEBUG:
             app = FastAPI(openapi_url=self.get_app_config().OPEN_API_URL, title=title,
-                          debug=self.get_app_config().DEBUG, version=self.API_VERSION)
+                          debug=self.get_app_config().DEBUG, version=self.API_VERSION, lifespan=lifespan)
         else:
             app = FastAPI(openapi_url=self.get_app_config().OPEN_API_URL, title=title,
-                          debug=self.get_app_config().DEBUG, version=self.API_VERSION, docs_url=None, redoc_url=None)
+                          debug=self.get_app_config().DEBUG, version=self.API_VERSION, lifespan=lifespan, docs_url=None,
+                          redoc_url=None)
 
         origins = ["*"]
 
@@ -90,20 +113,6 @@ class BaseFactory(metaclass=ABCMeta):
                                    self.get_app_config().DATABASE_AUTH_SOURCE,
                                    self.get_app_config().DATABASE_SSL, self.get_app_config().DATABASE_SSL_CA_CERTS)
 
-        @app.on_event("startup")
-        async def startup_db_client():
-            await mongodb.connect()
-            document_model_list = await self.get_document_model_list()
-            await init_beanie(database=app.state.db.get_database(), document_models=document_model_list)
-            app.state.logger.info("Database connected")
-            await self.init_state_cache(app, self.get_app_config())
-            app.state.logger.info("Initialization state cache done")
-
-        @app.on_event("shutdown")
-        async def shutdown_db_client():
-            await mongodb.close()
-            app.state.logger.info("Database disconnected")
-
         app.state.db = mongodb
         app.state.config = self.get_app_config()
         self.__init_modules(app)

internal/database.py

@@ -5,12 +5,6 @@ from .exception.internal_exception import DatabaseInitializeFailureException, Da
 
 
 class MongoDB:
-    # def __init__(self, connection_url: str, db_name: str, ssl: bool = False, ssl_ca_certs: str = None):
-    #     self.client = None
-    #     self.connection_url = connection_url
-    #     self.db_name = db_name
-    #     self.ssl = ssl
-    #     self.ssl_ca_certs = ssl_ca_certs
     def __init__(self, user_name: str, password: str, host: str, port: int, db_name: str, server_selection_timeout: int,
                  connection_timeout: int, auth_source: str, ssl: bool = False, ssl_ca_certs: str = None):
         self.client = None
@@ -49,3 +43,18 @@ class MongoDB:
         if not self.client:
             raise DatabaseConnectFailureException()
         return self.client[self.db_name]
+
+    async def get_mongodb_uri(self) -> str:
+        if self.user_name and self.password:
+            uri = f"mongodb://{self.user_name}:{self.password}@{self.host}:{self.port}/{self.db_name}"
+        else:
+            uri = f"mongodb://{self.host}:{self.port}/{self.db_name}"
+
+        uri += f"?authSource={self.auth_source}&connectTimeoutMS={self.connection_timeout}&serverSelectionTimeoutMS={self.server_selection_timeout}&retryWrites=false"
+
+        if self.ssl:
+            uri += "&ssl=true"
+        if self.ssl_ca_certs and self.ssl_ca_certs != "":
+            uri += f"&tlsCAFile={self.ssl_ca_certs}"
+
+        return uri

{internal-1.0.47.dist-info → internal-1.0.49.dist-info}/METADATA

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

{internal-1.0.47.dist-info → internal-1.0.49.dist-info}/RECORD

@@ -1,6 +1,6 @@
 internal/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 internal/base_config.py,sha256=ZdqvHlPsF6ozELgmV0ZiSpYehSscasIclLp8eAjVE24,1870
-internal/base_factory.py,sha256=RZAlxINve8TYgPh_2mW7r0Ro8ADjsIaOe6s4JpNYpPw,8494
+internal/base_factory.py,sha256=kdwWVIzOCrZBDdZxgMgjLpLU6WaB2PXzim9SKBormSs,8854
 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
@@ -9,7 +9,7 @@ internal/common_enum/lpnr_direction.py,sha256=Rkzyu3qz4_jwkA8tFsrRUc4L8HLOz9SALf
 internal/common_enum/operator_type.py,sha256=JU4EqloxtrIXzaUbEWuQLDMmAZy6AT8EAX8KIrJH4xs,125
 internal/common_enum/order_type.py,sha256=XwAl5JaZgFLahBQhBKG4VuMJ39Si9KZEhcSS0-DKKzQ,90
 internal/const.py,sha256=U1S9r7bjtHgad2oYQoHO19s8D4V0WDUG-L-haV4UaIw,1025
-internal/database.py,sha256=atf0iyEqeG97eYr3s4GrTHKM5LjLiKlCNWiuAWoE2vs,2124
+internal/database.py,sha256=1u53JnFJ0jvP7H4HPY2faA2ABdPQ8SlsZigYmQAPktI,2470
 internal/exception/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 internal/exception/base_exception.py,sha256=U0ZXx1FOXydAnQQImznkeAdsKVC_YaukPWvJdWqEBfg,363
 internal/exception/internal_exception.py,sha256=rBi70sDFwoYc2NSnxXQ9LY6r2TnxA7ILC64xjpg5h4c,2296
@@ -28,6 +28,6 @@ internal/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 internal/model/base_model.py,sha256=NS_GZvw_IqU1aZPHPpyQs6A741O34ybmzVzBPm9nGWU,3014
 internal/model/operate.py,sha256=luEoP_Asvso_11qz1SAE1Kn5KxAnDRptM0VylC5tNik,1493
 internal/utils.py,sha256=0SubS0iUhDvjSX1F4TykasA5-enYJzt2VH-f7_0BnjI,1509
-internal-1.0.47.dist-info/METADATA,sha256=tAuRx2hrVEZ2lhJKEJphsdu61t-1-4VDxmYky58aAqY,625
-internal-1.0.47.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-internal-1.0.47.dist-info/RECORD,,
+internal-1.0.49.dist-info/METADATA,sha256=GeWqdds8wsOsUhlzsEu6Y0H8RrBHPmUCW1msxm8qb18,625
+internal-1.0.49.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+internal-1.0.49.dist-info/RECORD,,