diff --git a/nonebot_plugin_marshoai/plugins/marshoai_bangumi/__init__.py b/nonebot_plugin_marshoai/plugins/marshoai_bangumi/__init__.py old mode 100755 new mode 100644 index 6825364d..9b6c37b2 --- a/nonebot_plugin_marshoai/plugins/marshoai_bangumi/__init__.py +++ b/nonebot_plugin_marshoai/plugins/marshoai_bangumi/__init__.py @@ -2,12 +2,41 @@ import traceback import httpx -from nonebot_plugin_marshoai.plugin import PluginMetadata +from nonebot_plugin_marshoai.plugin import PluginMetadata, on_function_call +# 定义插件元数据 __marsho_meta__ = PluginMetadata( - name="Bangumi 番剧信息", - description="Bangumi 番剧信息", - usage="Bangumi 番剧信息", - author="Liteyuki", - homepage="", + name="Bangumi日历", + author="MarshoAI", + description="这个插件可以帮助你获取Bangumi的日历信息~", ) + + +@on_function_call(description="获取Bangumi日历信息") +async def get_bangumi_news() -> str: + async def fetch_calendar(): + url = "https://api.bgm.tv/calendar" + headers = { + "User-Agent": "LiteyukiStudio/nonebot-plugin-marshoai (https://github.com/LiteyukiStudio/nonebot-plugin-marshoai)" + } + async with httpx.AsyncClient() as client: + response = await client.get(url, headers=headers) + # print(response.text) + return response.json() + + try: + result = await fetch_calendar() + info = "" + for i in result: + weekday = i["weekday"]["cn"] + # print(weekday) + info += f"{weekday}:" + items = i["items"] + for item in items: + name = item["name_cn"] + info += f"《{name}》" + info += "" + return info + except Exception as e: + traceback.print_exc() + return "" diff --git a/nonebot_plugin_marshoai/plugins/marshoai_bangumi/bangumi.py b/nonebot_plugin_marshoai/plugins/marshoai_bangumi/bangumi.py deleted file mode 100644 index 9b6c37b2..00000000 --- a/nonebot_plugin_marshoai/plugins/marshoai_bangumi/bangumi.py +++ /dev/null @@ -1,42 +0,0 @@ -import traceback - -import httpx - -from nonebot_plugin_marshoai.plugin import PluginMetadata, on_function_call - -# 定义插件元数据 -__marsho_meta__ = PluginMetadata( - name="Bangumi日历", - author="MarshoAI", - description="这个插件可以帮助你获取Bangumi的日历信息~", -) - - -@on_function_call(description="获取Bangumi日历信息") -async def get_bangumi_news() -> str: - async def fetch_calendar(): - url = "https://api.bgm.tv/calendar" - headers = { - "User-Agent": "LiteyukiStudio/nonebot-plugin-marshoai (https://github.com/LiteyukiStudio/nonebot-plugin-marshoai)" - } - async with httpx.AsyncClient() as client: - response = await client.get(url, headers=headers) - # print(response.text) - return response.json() - - try: - result = await fetch_calendar() - info = "" - for i in result: - weekday = i["weekday"]["cn"] - # print(weekday) - info += f"{weekday}:" - items = i["items"] - for item in items: - name = item["name_cn"] - info += f"《{name}》" - info += "" - return info - except Exception as e: - traceback.print_exc() - return ""