Entry.py

# -*- coding: utf-8 -*-
"""
@author: PC
FIXME   -參數
            PYTHONUNBUFFERED=1
            SAVE_PATH=./preprocess
            SQL_SERVICE_BROKER_HOST=<ip,port>
            SQL_SERVICE_LOGIN_PASSWORD=<password>
            SQL_SERVICE_LOGIN_USER=<user>
            LINE_ACCESS_TOKEN=[Fill In Your Access Token]
            LINE_SECRET_TOKEN=[Fill In Your Secret]
            GEMINI_TOKEN=[Fill In Your Token]
            GITHUB_PERSONAL_TOKEN=[Fill In Your Token]
"""
import os
from flask import request, Flask, abort
from linebot.models import MessageEvent, TextMessage, FileMessage, ImageMessage
from package.linebot import LineBotHandler

app = Flask(__name__)
linebot_token, handler, gemini_token, github_token = LineBotHandler.token_settings()
lbh = LineBotHandler(linebot_token, gemini_token, github_token)
lbh.update_webhook()
SAVE_PATH = os.environ.get('SAVE_PATH')

@app.route('/callback', methods=['POST'])
def callback():
    body = request.get_data(as_text=True)
    signature = request.headers['X-Line-Signature']
    try:
        handler.handle(body, signature)
    except Exception as e:
        print(e)
        abort(400)
    finally:
        return '', 200

@handler.add(MessageEvent, message=TextMessage)
def text_message(event: MessageEvent):
    if isinstance(event.message, TextMessage):
        lbh.process(event)

@handler.add(MessageEvent, message=FileMessage)
def file_message(event: MessageEvent):
    LineBotHandler.make_folder(SAVE_PATH)
    if isinstance(event.message, FileMessage):
        msg_id = event.message.id
        msg_name = event.message.file_name
        msg_size = LineBotHandler.make_decimal(event.message.file_size/1024/1024, '0.01')
        msg_content = linebot_token.get_message_content(msg_id).iter_content()
        file_name = f'{SAVE_PATH}/{msg_id}_{msg_name}'
        print(f'msg_id: {msg_id}, msg_name: {msg_name}, msg_size: {msg_size} MB')
        with open(file_name, 'wb') as f:
            for chunk in msg_content:
                f.write(chunk)

        lbh.process(event, file_name)

@handler.add(MessageEvent, message=ImageMessage)
def img_message(event: MessageEvent):
    LineBotHandler.make_folder(SAVE_PATH)
    if isinstance(event.message, ImageMessage):
        msg_id = event.message.id
        msg_content = linebot_token.get_message_content(msg_id).iter_content()
        file_name = f'{SAVE_PATH}/{msg_id}.jpg'
        with open(file_name, 'wb') as f:
            for chunk in msg_content:
                f.write(chunk)

        lbh.process(event, file_name)

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=5000)