internal 1.0.46__py3-none-any.whl → 1.0.48__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 = ["*"]
 
@@ -92,17 +115,11 @@ class BaseFactory(metaclass=ABCMeta):
 
         @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")
+            print("startup_db_client================run")
 
         @app.on_event("shutdown")
         async def shutdown_db_client():
-            await mongodb.close()
-            app.state.logger.info("Database disconnected")
+            print("shutdown_db_client============run")
 
         app.state.db = mongodb
         app.state.config = self.get_app_config()

internal/common_enum/event_trigger_type.py

@@ -7,6 +7,7 @@ class EventTriggerTypeEnum(str, Enum):
     MANUAL_TRACING_STOP_SERVICE_TICKET = "manual_tracking_stop_service_ticket"
     MANUAL_CREATE_SERVICE_TICKET = "manual_create_service_ticket"
     MANUAL_IMPORT_RESERVATION_SMWS = "manual_import_reservation_smws"
+    MANUAL_MODIFY_USER_SERVICE_TICKET = "manual_modify_user_service_ticket"
     MANUAL_MODIFY_ESTIMATED_ARRIVAL_TIME_SERVICE_TICKET = "manual_modify_estimated_arrival_time_service_ticket"
     MANUAL_MODIFY_ESTIMATED_DELIVERY_TIME_SERVICE_TICKET = "manual_modify_estimated_delivery_time_service_ticket"
     MANUAL_BOOKING_MESSAGE_SERVICE_TICKET = "manual_booking_message_service_ticket"
@@ -15,5 +16,10 @@ class EventTriggerTypeEnum(str, Enum):
     NOSHOW_SERVICE_TICKET_AUTO_CANCEL = "noshow_service_ticket_auto_cancel"
     IMPORT_RESERVATION_CONFLICT_AUTO_CANCEL = "import_reservation_conflict_auto_cancel"
     BOOKING_REMINDING_SERVICE_TICKET = "booking_reminding_service_ticket"
-    LPNR_IN = "lpnr_in"
+
+    LPNR_IN_ESTABLISHED = "lpnr_in_established"
+    LPNR_IN_RECEPTION = "lpnr_in_reception"
+    LPNR_IN_WORKING = "lpnr_in_working"
     LPNR_OUT = "lpnr_out"
+    LPNR_OUT_GENERAL = "lpnr_out_general"
+    LPNR_OUT_NO_SERVE = "lpnr_out_no_serve"

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,12 @@ class MongoDB:
         if not self.client:
             raise DatabaseConnectFailureException()
         return self.client[self.db_name]
+
+    def get_mongodb_uri(self) -> str:
+        uri = f"mongodb://{self.user_name}:{self.password}@{self.host}:{self.port}/{self.db_name}?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.46.dist-info → internal-1.0.48.dist-info}/METADATA

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

{internal-1.0.46.dist-info → internal-1.0.48.dist-info}/RECORD

@@ -1,15 +1,15 @@
 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=skI4MIFj5XW3Dtxu-Bp5oKoDU1RBEOCx-Sgrjt1aaqU,9115
 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
-internal/common_enum/event_trigger_type.py,sha256=NgJM4F3IIhXS4LOffufpgEp0HuOK96mLrLilMwr-VX8,1110
+internal/common_enum/event_trigger_type.py,sha256=Y5V7aHe-9UE2iCbghQJZtHkzLpHNgDUdcL96ztbQwEg,1381
 internal/common_enum/lpnr_direction.py,sha256=Rkzyu3qz4_jwkA8tFsrRUc4L8HLOz9SALfN4K_9Xd1w,90
 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=HV0F-1HiAo5SAxbgqlMGpBnUsNLY-W7xzc2PtkuAUfA,2310
 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.46.dist-info/METADATA,sha256=QLaS96nYOQKDbp4oqjcnz7SEk1fqXl2wXFcocCeaw20,625
-internal-1.0.46.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-internal-1.0.46.dist-info/RECORD,,
+internal-1.0.48.dist-info/METADATA,sha256=isqwmxO3KxWh3hlIqTcvBOv0eK6D4NJgbwcleHJbCFg,625
+internal-1.0.48.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+internal-1.0.48.dist-info/RECORD,,