LiteyukiBot/liteyuki/plugins/liteyuki_weather/qweather.py

from nonebot import require, on_endswith
from nonebot.adapters import satori
from nonebot.adapters.onebot.v11 import MessageSegment
from nonebot.internal.matcher import Matcher

from liteyuki.utils.base.config import get_config
from liteyuki.utils.base.ly_typing import T_MessageEvent

from .qw_api import *
from liteyuki.utils.base.data_manager import User, user_db
from liteyuki.utils.base.language import Language, get_user_lang
from liteyuki.utils.base.resource import get_path
from liteyuki.utils.message.html_tool import template2image
from liteyuki.utils import event as event_utils

require("nonebot_plugin_alconna")
from nonebot_plugin_alconna import on_alconna, Alconna, Args, MultiVar, Arparma


@on_alconna(
    aliases={"天气"},
    command=Alconna(
        "weather",
        Args["keywords", MultiVar(str), []],
    ),
).handle()
async def _(result: Arparma, event: T_MessageEvent, matcher: Matcher):
    """await alconna.send("weather", city)"""
    kws = result.main_args.get("keywords")
    image = await get_weather_now_card(matcher, event, kws)
    if isinstance(event, satori.event.Event):
        await matcher.finish(satori.MessageSegment.image(raw=image, mime="image/png"))
    else:
        await matcher.finish(MessageSegment.image(image))


@on_endswith(("天气", "weather")).handle()
async def _(event: T_MessageEvent, matcher: Matcher):
    """await alconna.send("weather", city)"""
    kws = event.message.extract_plain_text()
    image = await get_weather_now_card(matcher, event, [kws.replace("天气", "").replace("weather", "")], False)
    await matcher.finish(MessageSegment.image(image))


async def get_weather_now_card(matcher: Matcher, event: T_MessageEvent, keyword: list[str], tip: bool = True):
    ulang = get_user_lang(event_utils.get_user_id(event))
    qw_lang = get_qw_lang(ulang.lang_code)
    key = get_config("weather_key")
    is_dev = get_memory_data("weather.is_dev", True)
    user: User = user_db.where_one(User(), "user_id = ?", event_utils.get_user_id(event), default=User())
    # params
    unit = user.profile.get("unit", "m")
    stored_location = user.profile.get("location", None)

    if not key:
        await matcher.finish(ulang.get("weather.no_key") if tip else None)

    if keyword:
        if len(keyword) >= 2:
            adm = keyword[0]
            city = keyword[-1]
        else:
            adm = ""
            city = keyword[0]
        city_info = await city_lookup(city, key, adm=adm, lang=qw_lang)
        city_name = " ".join(keyword)
    else:
        if not stored_location:
            await matcher.finish(ulang.get("liteyuki.invalid_command", TEXT="location") if tip else None)
        city_info = await city_lookup(stored_location, key, lang=qw_lang)
        city_name = stored_location
    if city_info.code == "200":
        location_data = city_info.location[0]
    else:
        await matcher.finish(ulang.get("weather.city_not_found", CITY=city_name) if tip else None)
    weather_now = await get_weather_now(key, location_data.id, lang=qw_lang, unit=unit, dev=is_dev)
    weather_daily = await get_weather_daily(key, location_data.id, lang=qw_lang, unit=unit, dev=is_dev)
    weather_hourly = await get_weather_hourly(key, location_data.id, lang=qw_lang, unit=unit, dev=is_dev)
    aqi = await get_airquality(key, location_data.id, lang=qw_lang, dev=is_dev)

    image = await template2image(
        template=get_path("templates/weather_now.html", abs_path=True),
        templates={
            "data": {
                "params": {
                    "unit": unit,
                    "lang": ulang.lang_code,
                },
                "weatherNow": weather_now,
                "weatherDaily": weather_daily,
                "weatherHourly": weather_hourly,
                "aqi": aqi,
                "location": location_data.dump(),
                "localization": get_local_data(ulang.lang_code)
            }
        },
        debug=True,
        wait=1
    )
    return image