From 50ad57ec222b49397a98161b6742ebbd760d4b71 Mon Sep 17 00:00:00 2001 From: EillesWan Date: Thu, 22 Aug 2024 22:50:18 +0800 Subject: [PATCH] =?UTF-8?q?=E6=A0=BC=E5=BC=8F=E5=8C=96=E4=BB=A3=E7=A0=81?= =?UTF-8?q?=EF=BC=8C=E7=BB=86=E8=8A=82=E4=BC=98=E5=8C=96=EF=BC=8C=E4=BF=AE?= =?UTF-8?q?=E5=A4=8D=E4=BA=86=E5=9B=BE=E7=89=87=E8=BE=B9=E6=A1=86=E4=B8=8D?= =?UTF-8?q?=E5=AE=8C=E6=95=B4=E7=9A=84=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 17 +++--- nonebot_plugin_acgnshow/__init__.py | 7 ++- nonebot_plugin_acgnshow/acgnapis.py | 64 ++++++++++---------- nonebot_plugin_acgnshow/acgnshower.py | 36 +++++++----- nonebot_plugin_acgnshow/config.py | 5 +- nonebot_plugin_acgnshow/models.py | 4 ++ nonebot_plugin_acgnshow/res/template.html | 71 ++++++++++++++++++----- nonebot_plugin_acgnshow/util.py | 22 ++++--- pyproject.toml | 4 +- 9 files changed, 145 insertions(+), 85 deletions(-) diff --git a/README.md b/README.md index 8a37b99..294cfe7 100644 --- a/README.md +++ b/README.md @@ -10,7 +10,6 @@ _✨ 从哔哩哔哩会员购获取展览简易信息 ✨_ - license @@ -21,7 +20,6 @@ _✨ 从哔哩哔哩会员购获取展览简易信息 ✨_ - ## 📖 介绍 一个简单的 Nonebot2 插件,可以从哔哩哔哩会员购 API 获取展览(漫展等)的时间,地点,票价等信息。 @@ -44,21 +42,25 @@ _✨ 从哔哩哔哩会员购获取展览简易信息 ✨_ pip pip install nonebot-plugin-acgnshow +
pdm pdm add nonebot-plugin-acgnshow +
poetry poetry add nonebot-plugin-acgnshow +
conda conda install nonebot-plugin-acgnshow +
打开 nonebot2 项目根目录下的 `pyproject.toml` 文件, 在 `[tool.nonebot]` 部分追加写入 @@ -67,13 +69,14 @@ _✨ 从哔哩哔哩会员购获取展览简易信息 ✨_ -## 🎉 使用 +## 🎉 使用 + 发送`展览`指令可以获取使用说明 + ## ⚙️ 配置 在 nonebot2 项目的`.env`文件中添加下表中的配置 -| 配置项 | 必填 | 默认值 | 说明 | -|:-----:|:----:|:----:|:----:| -| ACGNSHOW_PAGESIZE | 否 | 8 | 单个图片的条目数,最大为20,条目数过大可能导致 Bot 无法发送 | - +| 配置项 | 必填 | 默认值 | 说明 | +| :---------------: | :--: | :----: | :----------------------------------------------------------: | +| ACGNSHOW_PAGESIZE | 否 | 8 | 单个图片的条目数,最大为 20,条目数过大可能导致 Bot 无法发送 | diff --git a/nonebot_plugin_acgnshow/__init__.py b/nonebot_plugin_acgnshow/__init__.py index 8accddf..c50cd57 100644 --- a/nonebot_plugin_acgnshow/__init__.py +++ b/nonebot_plugin_acgnshow/__init__.py @@ -1,9 +1,11 @@ from nonebot.plugin import PluginMetadata, inherit_supported_adapters, require + require("nonebot_plugin_htmlrender") require("nonebot_plugin_alconna") from .acgnshower import * from nonebot import get_driver from .config import ConfigModel + usage = """命令格式: 展览 <地区> [页码] 或 @@ -27,8 +29,8 @@ __plugin_meta__ = PluginMetadata( config=ConfigModel, homepage="https://github.com/LiteyukiStudio/nonebot-plugin-acgnshow", supported_adapters=inherit_supported_adapters("nonebot_plugin_alconna"), - #extra={"License":"MIT","Author":"Swankily"} snowykami的奇妙纠错( - extra={"License":"MIT","Author":"Asankilp"} + # extra={"License":"MIT","Author":"Swankily"} snowykami的奇妙纠错( + extra={"License": "MIT", "Author": "Asankilp"}, ) driver = get_driver() @@ -36,4 +38,3 @@ driver = get_driver() @driver.on_startup async def _(): pass - diff --git a/nonebot_plugin_acgnshow/acgnapis.py b/nonebot_plugin_acgnshow/acgnapis.py index d6de04f..e93c4d8 100644 --- a/nonebot_plugin_acgnshow/acgnapis.py +++ b/nonebot_plugin_acgnshow/acgnapis.py @@ -9,7 +9,7 @@ from .util import * CITY_API_ROOT = "https://show.bilibili.com/api/ticket/city/list?channel=3" SHOWS_API_ROOT = "https://show.bilibili.com/api/ticket/project/listV2" HEADERS = { - "user-agent": "Mozilla/5.0 (Linux; Android 14; 114514YAJU Build/UKQ1.114514.001; wv) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/125.0.6422.165 Mobile Safari/537.36 BiliApp/7810200 mobi_app/android isNotchWindow/0 NotchHeight=34 mallVersion/7810200 mVersion/242 disable_rcmd/0 7.81.0 os/android model/114514YAJU mobi_app/android build/7810200 channel/bilih5 innerVer/7810210 osVer/14 network/2" + "user-agent": "Mozilla/5.0 (Linux; Android 14; 114514YAJU Build/UKQ1.114514.001; wv) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/125.0.6422.165 Mobile Safari/537.36 BiliApp/7810200 mobi_app/android isNotchWindow/0 NotchHeight=34 mallVersion/7810200 mVersion/242 disable_rcmd/0 7.81.0 os/android model/114514YAJU mobi_app/android build/7810200 channel/bilih5 innerVer/7810210 osVer/14 network/2" } @@ -31,13 +31,8 @@ async def get_regions_dict() -> Dict[str, int]: city_data = await get_cities_data() for i in city_data.data.list: for j in i.city_list: - data.update({ - j.name: j.id - }) - data.update({ - "全国": -1, - "海外": 900000 - }) + data.update({j.name: j.id}) + data.update({"全国": -1, "海外": 900000}) return data @@ -50,13 +45,13 @@ async def get_shows_data(region_id: int, page=1, pagesize=20): pagesize: 一页最大条目数,最大20 """ param = { - "version" : 133, - "area" : region_id, - "page" : page, - "pagesize": pagesize, - "platform": "web", - "p_type" : "展览", - "style" : 1 + "version": 133, + "area": region_id, + "page": page, + "pagesize": pagesize, + "platform": "web", + "p_type": "展览", + "style": 1, } async with ClientSession() as session: async with session.get(SHOWS_API_ROOT, headers=HEADERS, params=param) as resp: @@ -108,30 +103,31 @@ def process_shows_data_to_template(shows_data: dict): district_name = i["district_name"] wish = i["wish"] cover = "https:" + i["cover"] - if district_name == None: district_name = "" + if district_name == None: + district_name = "" guests_list = i["guests"] - guests = "" if guests_list != None: - for n in guests_list: - guests += n["name"] + "," + guests = "、".join(n["name"] for n in guests_list) + else: + guests = "" item_dict = { - "name" : name, - "location" : district_name + venue_name, - "sale_flag" : sale_flag, - "id" : project_id, - "price" : price_low, - "start_time" : start_time, - "end_time" : end_time, - "wish" : wish, - "image_url" : cover, - "guests" : guests, - "page" : page, - "total_pages": total_pages + "name": name, + "location": district_name + venue_name, + "sale_flag": sale_flag, + "id": project_id, + "price": price_low, + "start_time": start_time, + "end_time": end_time, + "wish": wish, + "image_url": cover, + "guests": guests, + "page": page, + "total_pages": total_pages, } showlist.append(item_dict) global_data_dict = { - "page" : page, - "total_pages" : total_pages, - "total_results": total_results + "page": page, + "total_pages": total_pages, + "total_results": total_results, } return [showlist, global_data_dict] diff --git a/nonebot_plugin_acgnshow/acgnshower.py b/nonebot_plugin_acgnshow/acgnshower.py index 11abca1..87dd264 100644 --- a/nonebot_plugin_acgnshow/acgnshower.py +++ b/nonebot_plugin_acgnshow/acgnshower.py @@ -20,36 +20,44 @@ showcmd = on_alconna( showcmd.shortcut( r"(?P.+?)展览\s*(?P\d+)?\s*(?P.+)?", { - "prefix" : True, - "command": "展览", - "args" : ["{region}", "{page}", "{date}"], - } + "prefix": True, + "command": "展览", + "args": ["{region}", "{page}", "{date}"], + }, ) @showcmd.handle() async def find_show( - region: Optional[str] = None, page: Optional[int] = None, date: Optional[str] = None, + region: Optional[str] = None, + page: Optional[int] = None, + date: Optional[str] = None, ): - if not region: await UniMessage(__plugin_meta__.usage).send(); return - if not page: page = 1 - if not date: date = "" + if not region: + await UniMessage(__plugin_meta__.usage).send() + return + if not page: + page = 1 + if not date: + date = "" regions_dict = await get_regions_dict() regionid = regions_dict.get(region, None) - if regionid == None: await UniMessage("不支持此地区").send(); return + if regionid == None: + await UniMessage("地区未寻得或输入格式错误").send() + return # await showcmd.send("日期:"+ date) shows = await get_shows_data(regionid, page=page, pagesize=config.acgnshow_pagesize) # print(shows) try: shows_data = process_shows_data_to_template(shows) template = { - "shows" : shows_data[0], - "bgimage" : choose_random_bgimage(), - "global_data": shows_data[1] + "shows": shows_data[0], + "bgimage": choose_random_bgimage(), + "global_data": shows_data[1], } - pic = await template_to_pic(RES_PATH, TEMPLATE_NAME, template) + pic = await template_to_pic(str(RES_PATH), TEMPLATE_NAME, template) except Exception as e: - await UniMessage("发生错误").send() + await UniMessage("图片生成时产生未知错误").send() traceback.print_exc() return diff --git a/nonebot_plugin_acgnshow/config.py b/nonebot_plugin_acgnshow/config.py index c1c233e..4f5f1e6 100644 --- a/nonebot_plugin_acgnshow/config.py +++ b/nonebot_plugin_acgnshow/config.py @@ -1,11 +1,14 @@ from pathlib import Path from pydantic import BaseModel from nonebot import get_plugin_config + RES_PATH = Path(__file__).parent / "res" TEMPLATE_NAME = "template.html" BGIMAGE_PATH = RES_PATH / "bgimage" + class ConfigModel(BaseModel): acgnshow_pagesize: int = 8 -config: ConfigModel = get_plugin_config(ConfigModel) \ No newline at end of file + +config: ConfigModel = get_plugin_config(ConfigModel) diff --git a/nonebot_plugin_acgnshow/models.py b/nonebot_plugin_acgnshow/models.py index b1afa0e..25961fa 100644 --- a/nonebot_plugin_acgnshow/models.py +++ b/nonebot_plugin_acgnshow/models.py @@ -17,6 +17,7 @@ class City(BaseModel): """ 热门城市 """ + id: int type: int first_letter: str @@ -31,6 +32,7 @@ class CityDataList(BaseModel): """ 城市首字母 """ + letter: str city_list: List[City] @@ -39,6 +41,7 @@ class CityData(BaseModel): """ 城市数据 """ + hot: List[City] list: List[CityDataList] located_id: int @@ -48,6 +51,7 @@ class CityResp(BaseModel): """ 城市数据 """ + errno: int errtag: int msg: str diff --git a/nonebot_plugin_acgnshow/res/template.html b/nonebot_plugin_acgnshow/res/template.html index 57030d8..7e5fa9a 100644 --- a/nonebot_plugin_acgnshow/res/template.html +++ b/nonebot_plugin_acgnshow/res/template.html @@ -1,17 +1,20 @@ + - + Show Information +
展览信息
{% for show in shows %}
- Image + Image
{{ show.name }}
@@ -186,11 +221,17 @@
{% endfor %}
+ \ No newline at end of file diff --git a/nonebot_plugin_acgnshow/util.py b/nonebot_plugin_acgnshow/util.py index 51b5ad5..f48b85c 100644 --- a/nonebot_plugin_acgnshow/util.py +++ b/nonebot_plugin_acgnshow/util.py @@ -1,15 +1,21 @@ -from .config import BGIMAGE_PATH +import os import random import datetime +from .config import BGIMAGE_PATH -def choose_random_bgimage(): - randomfile = random.choice(list(BGIMAGE_PATH.iterdir())) - randomurl = randomfile.as_uri() + +def choose_random_bgimage() -> str: + """ + 从背景图片文件夹中随机选择一张图片,返回图片的uri地址 + """ + randomfile = random.choice(os.listdir(BGIMAGE_PATH)) + randomurl = (BGIMAGE_PATH / randomfile).as_uri() return randomurl -def convert_timestamp(timestamp): - obj = datetime.datetime.fromtimestamp(timestamp) - formatted_time = obj.strftime('%Y-%m-%d %H:%M:%S') - return formatted_time +def convert_timestamp(timestamp) -> str: + """ + 将时间戳转换为日期格式 + """ + return datetime.datetime.fromtimestamp(timestamp).strftime("%Y-%m-%d %H:%M:%S") diff --git a/pyproject.toml b/pyproject.toml index f091246..5d6b808 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,9 +4,7 @@ version = "0.1.5" description = "Nonebot2插件,从哔哩哔哩会员购获取简易展览数据" readme = "README.md" requires-python = "<4.0,>=3.9" -authors = [ - { name = "Asankilp", email = "asankilp@outlook.com" }, -] +authors = [{ name = "Asankilp", email = "asankilp@outlook.com" }] dependencies = [ "nonebot2[fastapi, websockets]>=2.2.0", "nonebot-plugin-alconna>=0.48.0",