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

fastapi_basic/ext/line/__init__.py

@@ -1,19 +1,27 @@
+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
 
-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.logger_message(message)
 
         except LineBotApiError as err:
             exception = LineBotException(str(err))
@@ -22,8 +30,27 @@ class LineBot:
     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.logger_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
+            raise exception
+
+    async 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.logger_message(message)
+            asyncio.create_task(self.handle_message(event))

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fastapi_basic
-Version: 0.1.2
+Version: 0.1.4
 Summary: A short description of your module
 Home-page: https://github.com/szx21023/fastapi-base
 Author: szx21023
@@ -22,6 +22,7 @@ Requires-Dist: fastapi==0.115.12
 Requires-Dist: h11==0.14.0
 Requires-Dist: idna==3.10
 Requires-Dist: jmespath==1.0.1
+Requires-Dist: line-bot-sdk==3.12.0
 Requires-Dist: pydantic==2.11.3
 Requires-Dist: pydantic_core==2.33.1
 Requires-Dist: python-dateutil==2.9.0.post0

{fastapi_basic-0.1.2.dist-info → fastapi_basic-0.1.4.dist-info}/RECORD

@@ -10,8 +10,8 @@ fastapi_basic/exception/base_exception.py,sha256=VwMqn85rBeJCF6RIOYUR1eLxbP2aSDh
 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,,
+fastapi_basic/ext/line/__init__.py,sha256=LJ9w5zSj9d1CMTV0zIwaEhlhOYFSK1owONCc654yObY,2144
+fastapi_basic-0.1.4.dist-info/METADATA,sha256=7pxd6PohLr89mHqHVP_sqUBddMuS5tcLIbV9IZ3ApvI,1776
+fastapi_basic-0.1.4.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+fastapi_basic-0.1.4.dist-info/top_level.txt,sha256=Q9PdwWxaB4dy135MQHiroRYOqArdUSaIeEkzYinN6W0,14
+fastapi_basic-0.1.4.dist-info/RECORD,,