main.py

import logging
import os
import sys

from urllib.parse import quote
from utils.common import check_image_wet, check_location_in_message
from utils.flex import entrance
from utils.github import Github

from utils.image import SkateParkImage
from utils.weather import get_current_weather, get_weather_data, simplify_data

if os.getenv('API_ENV') != 'production':
    from dotenv import load_dotenv

    load_dotenv()


from fastapi import FastAPI, HTTPException, Request

from linebot.v3.webhook import WebhookParser
from linebot.v3.messaging import (
    AsyncApiClient,
    AsyncMessagingApi,
    Configuration,
    ReplyMessageRequest,
    TextMessage,
    ImageMessage,
    QuickReply,
    QuickReplyItem,
    MessageAction,
    ShowLoadingAnimationRequest
)
from linebot.v3.exceptions import (
    InvalidSignatureError
)
from linebot.v3.webhooks import (
    MessageEvent,
    TextMessageContent,
)

import uvicorn

logging.basicConfig(level=os.getenv('LOG', 'WARNING'))
logger = logging.getLogger(__file__)

app = FastAPI()

channel_secret = os.getenv('LINE_CHANNEL_SECRET', None)
channel_access_token = os.getenv('LINE_CHANNEL_ACCESS_TOKEN', None)
if channel_secret is None:
    print('Specify LINE_CHANNEL_SECRET as environment variable.')
    sys.exit(1)
if channel_access_token is None:
    print('Specify LINE_CHANNEL_ACCESS_TOKEN as environment variable.')
    sys.exit(1)

configuration = Configuration(
    access_token=channel_access_token
)

async_api_client = AsyncApiClient(configuration)
line_bot_api = AsyncMessagingApi(async_api_client)
parser = WebhookParser(channel_secret)


@app.get("/health")
async def health():
    return 'ok'


@app.post("/webhooks/line")
async def handle_callback(request: Request):
    signature = request.headers['X-Line-Signature']

    # get request body as text
    body = await request.body()
    body = body.decode()

    try:
        events = parser.parse(body, signature)
    except InvalidSignatureError:
        raise HTTPException(status_code=400, detail="Invalid signature")
    for event in events:
        logging.info(event)
        if not isinstance(event, MessageEvent):
            continue
        if not isinstance(event.message, TextMessageContent):
            continue
        await line_bot_api.show_loading_animation(ShowLoadingAnimationRequest(chatId=event.source.user_id, loadingSeconds=20))

        text = event.message.text
        SkatePark = SkateParkImage()
        park_list = SkatePark.get_name_list()
        logger.debug("Event message in list:" + text in park_list)
        if text == "入口":
            await line_bot_api.reply_message(
                ReplyMessageRequest(
                    reply_token=event.reply_token,
                    messages=[entrance(park_list)]
                )
            )
        elif text in park_list:
            github = Github()
            if SkatePark.url_dict.get(text) is None:
                url = "https://raw.githubusercontent.com/louis70109/ideas-tree/master/images/494737304821170505.png"
            else:
                url = f"https://raw.githubusercontent.com/{github.repo_name}/master/images/{quote(text)}/default.png"

            logger.info('Ready to push data...')
            logger.info('Crawler Weather Open Data...')

            location = check_location_in_message(text)
            logger.debug('Location is: ' + location)
            weather_data = get_weather_data(location)
            simplified_data = simplify_data(weather_data)
            current_weather = get_current_weather(simplified_data)

            logger.debug('The Data is: ' + str(current_weather))
            if current_weather is not None:
                text = f'☝️「{text}」...\n位置: {location}\n氣候: {current_weather["Wx"]}\n降雨機率: {current_weather["PoP"]}\n體感: {current_weather["CI"]}'
            else:
                text = f'「{text}」...圖片如下'
            logger.debug(url)

            if location in ['臺北市', '新北市']:
                is_wet = check_image_wet(url)
                text = text + f"\n地板濕: {is_wet}"

            await line_bot_api.reply_message(
                ReplyMessageRequest(
                    reply_token=event.reply_token,
                    messages=[
                        ImageMessage(
                            originalContentUrl=url,
                            previewImageUrl=url
                        ),
                        TextMessage(
                            text=text,
                            quick_reply=QuickReply(items=[
                                QuickReplyItem(
                                    action=MessageAction(label="入口", text="入口")
                                )]
                            )
                        ),
                    ]
                )
            )
        else:
            await line_bot_api.reply_message(
                ReplyMessageRequest(
                    reply_token=event.reply_token,
                    messages=[TextMessage(
                        text=text,
                        quick_reply=QuickReply(items=[
                            QuickReplyItem(
                                action=MessageAction(label="入口", text="入口")
                            )])
                    )
                    ]
                )
            )
    return 'OK'


if __name__ == "__main__":
    port = int(os.environ.get('PORT', default=8000))
    debug = True if os.environ.get(
        'API_ENV', default='develop') == 'develop' else False
    logging.info('Application will start...')
    uvicorn.run("main:app", host="0.0.0.0", port=port, reload=debug)