fastapi-basic 0.1.2__py3-none-any.whl → 0.1.11__py3-none-any.whl

fastapi_basic/base_config.py

@@ -14,4 +14,9 @@ class BaseConfig(BaseSettings):
     AWS_SECRET_KEY: str = ""
     AWS_REGION: str = ""
     AWS_PARAMETER_PATH_PREFIX: str = ""
-    AWS_LOGGROUP_NAME: str = ""
+    AWS_LOGGROUP_NAME: str = ""
+
+    # Celery config
+    CELERY_APP: str|None = ""
+    CELERY_BROKER_URL: str|None = ""
+    CELERY_BACKEND_URL: str|None = ""

fastapi_basic/base_factory.py

@@ -3,13 +3,14 @@ from functools import lru_cache
 import os, dotenv
 
 import watchtower
+from celery import Celery
 from fastapi import FastAPI, HTTPException, Request
 from fastapi.responses import JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
 from starlette.concurrency import iterate_in_threadpool
 import logging
 
-from .const import LOG_DEFAULT_LOGGER_NAME, LOG_FMT
+from .const import LOG_DEFAULT_LOGGER_NAME, LOG_FMT, CELERY_CONFIG
 from .ext.aws import init_app as init_aws_app
 from .exception.base_exception import InternalBaseException
 from .utils import update_dict_with_cast
@@ -48,6 +49,17 @@ class BaseFactory(metaclass=ABCMeta):
             watchtower_handler.setFormatter(logging.Formatter(LOG_FMT))
             app.logger.addHandler(watchtower_handler)
 
+    def __setup_celery_app(self, app):
+        if not all([app.state.config.get(key) for key in CELERY_CONFIG]):
+            app.logger.info("Lack Celery config keys, ignore get Celery_app")
+            return None
+
+        app.state.celery_app = Celery(
+            app.state.config.get("CELERY_APP"),
+            broker=app.state.config.get("CELERY_BROKER_URL"),
+            backend=app.state.config.get("CELERY_BACKEND_URL")
+        )
+
     def create_app(self):
         """
         Create an application instance.
@@ -68,8 +80,11 @@ class BaseFactory(metaclass=ABCMeta):
         app.state.aws_session = init_aws_app(app)
         self.__setup_aws_cloud_log(app)
 
+        self.__setup_celery_app(app)
+
         @app.exception_handler(InternalBaseException)
         async def http_exception_handler(request: Request, exc: InternalBaseException):
+            app.logger.warning(f'message: {exc.detail["message"]}')
             return JSONResponse(
                 status_code=exc.status_code,
                 content={

fastapi_basic/const.py

@@ -1,2 +1,4 @@
 LOG_DEFAULT_LOGGER_NAME = "uvicorn"
-LOG_FMT = '%(asctime)s %(filename)s %(levelname)s: %(message)s'
+LOG_FMT = '%(asctime)s %(filename)s %(levelname)s: %(message)s'
+
+CELERY_CONFIG = ["CELERY_APP", "CELERY_BROKER_URL", "CELERY_BACKEND_URL"]

fastapi_basic/exception/base_exception.py

@@ -11,13 +11,3 @@ class InternalBaseException(HTTPException):
             "data": kwargs,
         }
         super().__init__(status_code=status_code, detail=detail)
-
-class LineBotException(HTTPException):
-    def __init__(self, status_code: int = status.HTTP_502_BAD_GATEWAY, code: str = "line_bot_error",
-                 message: str = "Line bot error", **kwargs):
-        detail = {
-            "code": code,
-            "message": message,
-            "data": kwargs,
-        }
-        super().__init__(status_code=status_code, detail=detail)

fastapi_basic/ext/line/__init__.py

@@ -1,29 +1,56 @@
+from abc import ABCMeta, abstractmethod
+import asyncio
+
 from linebot import LineBotApi, WebhookHandler
 from linebot.exceptions import LineBotApiError
-from linebot.models import TextMessage, TextSendMessage
+from linebot.models import TextMessage, TextSendMessage, MessageEvent
 
-from ...exception.base_exception import LineBotException
+from .exception import LineBotException
 
-class LineBot:
-    def __init__(self, channel_access_token: str, channel_secret: str):
+class LineBot(metaclass=ABCMeta):
+    def __init__(self, channel_access_token: str, channel_secret: str, logger=None):
         self.channel_access_token = channel_access_token
         self.channel_secret = channel_secret
         self.line_bot_api = LineBotApi(self.channel_access_token)
         self.handler = WebhookHandler(self.channel_secret)
+        self.logger = logger
+
+        self.register_handlers()
 
-    async def reply_message(self, event, message):
+    async def reply_message(self, event, text: str):
         try:
-            self.line_bot_api.reply_message(event.reply_token, TextMessage(text=message))
+            self.line_bot_api.reply_message(event.reply_token, TextMessage(text=text))
+            message = f"Send line message successful, event: {event}, text: {text}"
+            self.log_message(message)
 
         except LineBotApiError as err:
-            exception = LineBotException(str(err))
+            exception = LineBotException(message=str(err))
             raise exception
 
     async def push_message(self, line_uid: str, text: str):
         try:
             self.line_bot_api.push_message(line_uid, TextSendMessage(text=text))
+            message = f"Send line message successful, line_uid: {line_uid}, text: {text}"
+            self.log_message(message)
 
         except LineBotApiError as err:
             message = f"Send line message fail, line_uid: {line_uid}, text: {text}, err: {str(err)}"
-            exception = LineBotException(message)
-            raise exception
+            exception = LineBotException(message=message)
+            raise exception
+
+    def log_message(self, log_message: str):
+        if self.logger:
+            self.logger.info(log_message)
+
+    @abstractmethod
+    async def handle_message(self):
+        """
+        Each bot should define what handle_message it wants.
+        """
+
+    def register_handlers(self):
+        @self.handler.add(MessageEvent, message=TextMessage)
+        def handle_message(event: MessageEvent):
+            message = f'handle_message: {event}'
+            self.log_message(message)
+            asyncio.create_task(self.handle_message(event))

fastapi_basic/ext/line/exception.py

@@ -0,0 +1,12 @@
+from fastapi import status
+from ...exception.base_exception import InternalBaseException
+
+
+class LineBotException(InternalBaseException):
+    def __init__(self, message: str = "", **kwargs):
+        status_code: int = status.HTTP_502_BAD_GATEWAY
+        code: str = "line_bot_error"
+        log_message: str = "Line bot error"
+
+        message = f"{log_message}, {message}" if message else log_message
+        super().__init__(status_code=status_code, code=code, message=message, **kwargs)

{fastapi_basic-0.1.2.dist-info → fastapi_basic-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fastapi_basic
-Version: 0.1.2
+Version: 0.1.11
 Summary: A short description of your module
 Home-page: https://github.com/szx21023/fastapi-base
 Author: szx21023
@@ -10,33 +10,65 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.6
 Description-Content-Type: text/markdown
+Requires-Dist: aenum==3.1.16
+Requires-Dist: aiohappyeyeballs==2.6.1
+Requires-Dist: aiohttp==3.10.5
+Requires-Dist: aiosignal==1.3.2
+Requires-Dist: amqp==5.3.1
 Requires-Dist: annotated-types==0.7.0
 Requires-Dist: anyio==4.9.0
+Requires-Dist: attrs==25.3.0
+Requires-Dist: beanie==1.29.0
+Requires-Dist: billiard==4.2.1
 Requires-Dist: boto3==1.37.36
 Requires-Dist: botocore==1.37.36
+Requires-Dist: celery==5.5.2
 Requires-Dist: certifi==2025.1.31
 Requires-Dist: charset-normalizer==3.4.1
 Requires-Dist: click==8.1.8
+Requires-Dist: click-didyoumean==0.3.1
+Requires-Dist: click-plugins==1.1.1
+Requires-Dist: click-repl==0.3.0
+Requires-Dist: Deprecated==1.2.18
+Requires-Dist: dnspython==2.7.0
 Requires-Dist: dotenv==0.9.9
 Requires-Dist: fastapi==0.115.12
-Requires-Dist: h11==0.14.0
+Requires-Dist: frozenlist==1.6.0
+Requires-Dist: future==1.0.0
+Requires-Dist: h11==0.16.0
 Requires-Dist: idna==3.10
 Requires-Dist: jmespath==1.0.1
+Requires-Dist: kombu==5.5.3
+Requires-Dist: lazy-model==0.2.0
+Requires-Dist: line-bot-sdk==3.12.0
+Requires-Dist: motor==3.7.1
+Requires-Dist: multidict==6.4.3
+Requires-Dist: prompt_toolkit==3.0.51
+Requires-Dist: propcache==0.3.1
 Requires-Dist: pydantic==2.11.3
+Requires-Dist: pydantic-settings==2.9.1
 Requires-Dist: pydantic_core==2.33.1
+Requires-Dist: pymongo==4.13.0
 Requires-Dist: python-dateutil==2.9.0.post0
 Requires-Dist: python-dotenv==1.1.0
 Requires-Dist: python-multipart==0.0.20
+Requires-Dist: redis==6.1.0
 Requires-Dist: requests==2.32.3
 Requires-Dist: s3transfer==0.11.5
 Requires-Dist: six==1.17.0
 Requires-Dist: sniffio==1.3.1
 Requires-Dist: starlette==0.46.1
+Requires-Dist: toml==0.10.2
 Requires-Dist: typing-inspection==0.4.0
 Requires-Dist: typing_extensions==4.13.1
+Requires-Dist: tzdata==2025.2
 Requires-Dist: urllib3==2.3.0
 Requires-Dist: uvicorn==0.34.0
+Requires-Dist: vine==5.1.0
 Requires-Dist: watchtower==3.4.0
+Requires-Dist: wcwidth==0.2.13
+Requires-Dist: wrapt==1.17.2
+Requires-Dist: yarl==1.20.0
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

fastapi_basic-0.1.11.dist-info/RECORD

@@ -0,0 +1,18 @@
+fastapi_basic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+fastapi_basic/base_config.py,sha256=WtYA7nI7yTw_bmMMIH1JdKiGOl81Vjekf2OpmwtnCww,568
+fastapi_basic/base_factory.py,sha256=a8AtByr7zp0OQhWzaTkh51KQ_Q_b7XZEF2hcl1rRYCk,4387
+fastapi_basic/const.py,sha256=jjNg7nutqtfzrqEKlU-zbzkDo3ndf3aDc603LbGcB_k,174
+fastapi_basic/utils.py,sha256=8ympyQIXsKkxLILTI_7ug85vmKWYKqX0mpADjgHekgU,306
+fastapi_basic/database/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+fastapi_basic/database/mongodb.py,sha256=XW7ApqYkR1Jf7LM1HaZ6jzYgajzuu1whKw5cpsjSO-M,1017
+fastapi_basic/exception/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+fastapi_basic/exception/base_exception.py,sha256=xkpvAI8G96sqXqHgQC_rs2ZPL1liwKVXbXBIPF_8HPQ,473
+fastapi_basic/ext/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+fastapi_basic/ext/aws/__init__.py,sha256=N_cwnbv1N-F1SY6NenBOTZe20zGaKFu3UXAu17hFAXY,816
+fastapi_basic/ext/aws/const.py,sha256=Mmb6lo11aZZDAVy-nK-v_JxtVPPKYUo9GB4ihH7XYuM,68
+fastapi_basic/ext/line/__init__.py,sha256=XkFYkV3Mx6w4TRim4ziETJrgOy9qJ_4pPrnqonBE_QU,2128
+fastapi_basic/ext/line/exception.py,sha256=AUJdg15yDUxQe7kys_NiOBqfEgmolj8-y5c2wNgfp2U,490
+fastapi_basic-0.1.11.dist-info/METADATA,sha256=DAmDHtBYTmy-rE_3JIisu_kr0CHeAhdDg_eLaGCpiRM,2753
+fastapi_basic-0.1.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+fastapi_basic-0.1.11.dist-info/top_level.txt,sha256=Q9PdwWxaB4dy135MQHiroRYOqArdUSaIeEkzYinN6W0,14
+fastapi_basic-0.1.11.dist-info/RECORD,,

{fastapi_basic-0.1.2.dist-info → fastapi_basic-0.1.11.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (78.1.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

fastapi_basic-0.1.2.dist-info/RECORD

@@ -1,17 +0,0 @@
-fastapi_basic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-fastapi_basic/base_config.py,sha256=nimiiK5LRY9kmx3dBDbN7teO4R8E423-c2iyq9WMcCg,442
-fastapi_basic/base_factory.py,sha256=xrM4tbKjrhtqpzP2hupGJxiNeGsDnXfMYe8urOcq-ss,3801
-fastapi_basic/const.py,sha256=UA7-Eefu_dbWpbFn09Ei_BPb903SExnCgVbnm8_3ALE,99
-fastapi_basic/utils.py,sha256=8ympyQIXsKkxLILTI_7ug85vmKWYKqX0mpADjgHekgU,306
-fastapi_basic/database/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-fastapi_basic/database/mongodb.py,sha256=XW7ApqYkR1Jf7LM1HaZ6jzYgajzuu1whKw5cpsjSO-M,1017
-fastapi_basic/exception/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-fastapi_basic/exception/base_exception.py,sha256=VwMqn85rBeJCF6RIOYUR1eLxbP2aSDhhtAyyhYmBI84,855
-fastapi_basic/ext/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-fastapi_basic/ext/aws/__init__.py,sha256=N_cwnbv1N-F1SY6NenBOTZe20zGaKFu3UXAu17hFAXY,816
-fastapi_basic/ext/aws/const.py,sha256=Mmb6lo11aZZDAVy-nK-v_JxtVPPKYUo9GB4ihH7XYuM,68
-fastapi_basic/ext/line/__init__.py,sha256=ZyCTuvW9P_iaMr2lFXp0JEEQMK773qOTnWiUdxIwAS8,1170
-fastapi_basic-0.1.2.dist-info/METADATA,sha256=AVeideHs4DOieR-m9ec0KSbnJ_2sFxG_X6btu8PCbyw,1740
-fastapi_basic-0.1.2.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-fastapi_basic-0.1.2.dist-info/top_level.txt,sha256=Q9PdwWxaB4dy135MQHiroRYOqArdUSaIeEkzYinN6W0,14
-fastapi_basic-0.1.2.dist-info/RECORD,,