feat: 同步服务端代码
This commit is contained in:
15
src/plugins/weather/__init__.py
Normal file
15
src/plugins/weather/__init__.py
Normal file
@@ -0,0 +1,15 @@
|
||||
from pathlib import Path
|
||||
|
||||
import nonebot
|
||||
from nonebot import get_driver
|
||||
|
||||
from .config import Config
|
||||
|
||||
global_config = get_driver().config
|
||||
config = Config.parse_obj(global_config)
|
||||
|
||||
_sub_plugins = set()
|
||||
_sub_plugins |= nonebot.load_plugins(
|
||||
str((Path(__file__).parent / "plugins").resolve())
|
||||
)
|
||||
|
||||
5
src/plugins/weather/config.py
Normal file
5
src/plugins/weather/config.py
Normal file
@@ -0,0 +1,5 @@
|
||||
from pydantic import BaseModel, Extra
|
||||
|
||||
|
||||
class Config(BaseModel, extra=Extra.ignore):
|
||||
"""Plugin Config Here"""
|
||||
34
src/plugins/weather/plugins/MyWeather.py
Normal file
34
src/plugins/weather/plugins/MyWeather.py
Normal file
@@ -0,0 +1,34 @@
|
||||
from nonebot import on_command
|
||||
from nonebot.rule import to_me
|
||||
from nonebot.matcher import Matcher
|
||||
from nonebot.adapters import Message
|
||||
from nonebot.params import Arg, CommandArg, ArgPlainText
|
||||
|
||||
|
||||
weather = on_command("weather", rule=to_me(), aliases={"天气", "天气预报"}, priority=5)
|
||||
|
||||
@weather.handle()
|
||||
async def handle_first_receive(matcher: Matcher, args: Message = CommandArg()):
|
||||
plain_text = args.extract_plain_text() # 首次发送命令时跟随的参数,例:/天气 上海,则args为上海
|
||||
if plain_text:
|
||||
matcher.set_arg("city", args) # 如果用户发送了参数则直接赋值
|
||||
|
||||
|
||||
|
||||
|
||||
@weather.got("city", prompt="你想查询哪个城市的天气呢?")
|
||||
async def handle_city(city: Message = Arg(), city_name: str = ArgPlainText("city")):
|
||||
if city_name not in ["北京", "上海"]: # 如果参数不符合要求,则提示用户重新输入
|
||||
# 可以使用平台的 Message 类直接构造模板消息
|
||||
await weather.reject(city.template("你想查询的城市 {city} 暂不支持,请重新输入!"))
|
||||
|
||||
|
||||
city_weather = await get_weather(city_name)
|
||||
await weather.finish(city_weather)
|
||||
|
||||
|
||||
|
||||
|
||||
# 在这里编写获取天气信息的函数
|
||||
async def get_weather(city: str) -> str:
|
||||
return f"{city}的天气是..."
|
||||
21
src/plugins/zhibo8/__init__.py
Normal file
21
src/plugins/zhibo8/__init__.py
Normal file
@@ -0,0 +1,21 @@
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
import nonebot
|
||||
from nonebot import get_driver, on_message
|
||||
from nonebot.adapters.onebot.v11 import Bot, Event
|
||||
from nonebot.typing import T_State
|
||||
from nonebot.params import CommandArg
|
||||
from nonebot.message import event_preprocessor
|
||||
|
||||
from .config import Config
|
||||
from nonebot import on_command
|
||||
from nonebot.rule import to_me
|
||||
from requests_html import HTMLSession, HTML
|
||||
|
||||
global_config = get_driver().config
|
||||
config = Config.parse_obj(global_config)
|
||||
|
||||
_sub_plugins = set()
|
||||
_sub_plugins |= nonebot.load_plugins(
|
||||
str((Path(__file__).parent / "plugins").resolve())
|
||||
)
|
||||
5
src/plugins/zhibo8/config.py
Normal file
5
src/plugins/zhibo8/config.py
Normal file
@@ -0,0 +1,5 @@
|
||||
from pydantic import BaseModel, Extra
|
||||
|
||||
|
||||
class Config(BaseModel, extra=Extra.ignore):
|
||||
"""Plugin Config Here"""
|
||||
80
src/plugins/zhibo8/plugins/Foresee.py
Normal file
80
src/plugins/zhibo8/plugins/Foresee.py
Normal file
@@ -0,0 +1,80 @@
|
||||
from datetime import datetime
|
||||
from nonebot.adapters.onebot.v11 import Bot, Event
|
||||
from nonebot.typing import T_State
|
||||
from nonebot.params import CommandArg, Command
|
||||
|
||||
from nonebot import on_command
|
||||
from nonebot.rule import to_me
|
||||
from requests_html import HTMLSession, HTML
|
||||
|
||||
zhibo8Command = on_command("/zhibo8", rule=to_me(), aliases={'预测', '英超', 'zhibo8'}, priority=10)
|
||||
|
||||
xiaobianStr = r"""
|
||||
小编们预测比分:
|
||||
太能喵:
|
||||
叶小欢:
|
||||
大帝强:
|
||||
小小:
|
||||
"""
|
||||
vsMap = {}
|
||||
@zhibo8Command.handle()
|
||||
async def handle_first_receive(bot: Bot, event: Event, state: T_State, args= CommandArg()):
|
||||
# 事件预处理,获取对应的赛程数据
|
||||
vsMap.clear()
|
||||
session = HTMLSession()
|
||||
zhibo8Get = session.get("https://www.zhibo8.cc/")
|
||||
print(zhibo8Get)
|
||||
saishiContent = zhibo8Get.html.xpath("//div[@class='vct-box']/div[@class='_content']/ul/li")
|
||||
# 解析xpath
|
||||
# 解析xpath
|
||||
for item in saishiContent:
|
||||
if item.html.find("英超") >= 0:
|
||||
if item.html.find("阿森纳") >= 0 or item.html.find("曼城") >= 0 or item.html.find("曼联") >= 0 or item.html.find(
|
||||
"利物浦") >= 0 or item.html.find("切尔西") >= 0:
|
||||
rouneNum, vsStr = parserSaishiItem(item.html)
|
||||
if rouneNum not in vsMap.keys():
|
||||
vsMap[rouneNum] = []
|
||||
vsMap[rouneNum].append(vsStr)
|
||||
|
||||
plain_text = args.extract_plain_text()
|
||||
if plain_text:
|
||||
state.set_arg("round", plain_text)
|
||||
else:
|
||||
await zhibo8Command.send(message=""""选择要查看的轮次,当前支持查看的轮次为:\r\n{} \r\n""".format(getRoundPrompt(vsMap)))
|
||||
|
||||
@zhibo8Command.got(key="round", prompt="""请输入编号:""")
|
||||
async def getRoundNum(event: Event):
|
||||
round = str(event.get_message())
|
||||
if round.isdigit() == False:
|
||||
await zhibo8Command.reject(prompt="请输入编号(数字)")
|
||||
if len(vsMap.keys())>=int(round):
|
||||
resultStr = ""
|
||||
roudNum = list(vsMap.keys())[int(round)-1]
|
||||
# for roudNum in vsMap:
|
||||
resultStr += "【" + roudNum.replace("第", "联赛 第") + " 比分预测】\r\n"
|
||||
for vs in vsMap[roudNum]:
|
||||
resultStr += vs + xiaobianStr + "\r\n"
|
||||
await zhibo8Command.finish(resultStr)
|
||||
else:
|
||||
await zhibo8Command.reject_arg(key="round", prompt="输入的编号错误,请重新输入!")
|
||||
|
||||
def parserSaishiItem(saishiStr):
|
||||
html = HTML(html=saishiStr)
|
||||
# listItem = html.xpath("//b//text()")
|
||||
roundNum = html.xpath("//span[@class='_league']//text()")[0]
|
||||
homeName = html.xpath("//span[@class='_teams']//text()")[0].strip()
|
||||
awayName = html.xpath("//span[@class='_teams']//text()")[-1].strip()
|
||||
timeStr = html.xpath("./li/@data-time")[0]
|
||||
time = datetime.strptime(timeStr, "%Y-%m-%d %H:%M")
|
||||
timeTuple = time.timetuple()
|
||||
return (roundNum, "{}月{}日".format(timeTuple.tm_mon, timeTuple.tm_mday)+time.strftime("%H:%M") + " " + homeName + " vs " + awayName)
|
||||
|
||||
|
||||
def getRoundPrompt(roundMap):
|
||||
str = ""
|
||||
i = 1
|
||||
for key in roundMap:
|
||||
print(key)
|
||||
str += "{}: {} \r\n".format(i, key)
|
||||
i = i+1
|
||||
return str
|
||||
Reference in New Issue
Block a user