mirror of
https://github.com/ClovertaTheTrilobita/SanYeCao-Nonebot.git
synced 2026-04-01 22:04:51 +00:00
56 lines
2.2 KiB
Python
56 lines
2.2 KiB
Python
import requests
|
||
from nonebot.plugin import on_command
|
||
from nonebot.rule import to_me
|
||
from nonebot.adapters.qq import Message, MessageEvent, MessageSegment
|
||
from pathlib import Path
|
||
from src.configs.path_config import path
|
||
|
||
cf_query = on_command("cf", rule=to_me(), priority=10, block=True)
|
||
@cf_query.handle()
|
||
async def get_cf_rounds():
|
||
await cf_query.send("正在为您整理近期比赛信息哦~\n请稍等💭💡🎈")
|
||
try:
|
||
result = requests.get('https://codeforces.com/api/contest.list?gym=false').json()
|
||
except BaseException:
|
||
await cf_query.finish("API请求失败，这绝对不是咱的错，绝对不是！")
|
||
all_matches = ""
|
||
for matches in result['result']:
|
||
phase = get_match_phase(matches["phase"])
|
||
one_match = ("\n比赛：" + str(matches["name"]) + "\n状态：" + phase + "\n时长：" + str(int(matches["durationSeconds"]) / 3600) +"h\n")
|
||
all_matches = "".join([all_matches, one_match])
|
||
if phase == "未开始":
|
||
until_start_time_min = 0 - int(matches["relativeTimeSeconds"]) / 60
|
||
until_start = get_until_start_time(until_start_time_min)
|
||
all_matches = "".join([all_matches, until_start])
|
||
if matches["phase"] == "FINISHED":
|
||
break
|
||
|
||
cf_image_path = path + "/image/codeforces/cfContestQR.png"
|
||
msg = Message([
|
||
MessageSegment.file_image(Path(cf_image_path)),
|
||
MessageSegment.text(all_matches),
|
||
])
|
||
|
||
await cf_query.finish(msg)
|
||
|
||
|
||
def get_match_phase(phase):
|
||
phase_map = {
|
||
"BEFORE": "未开始",
|
||
"FINISHED": "已结束",
|
||
"CODING": "进行中",
|
||
"PENDING_SYSTEM_TEST": "等待判题",
|
||
"SYSTEM_TEST": "判题中",
|
||
}
|
||
return phase_map.get(phase, "未知")
|
||
|
||
|
||
def get_until_start_time(until_start_time_min):
|
||
if until_start_time_min <= 180:
|
||
return "距开始：" + str(int(until_start_time_min)) + "min\n"
|
||
elif 180 < until_start_time_min <= 1440:
|
||
return "距开始：" + str(int(until_start_time_min / 60)) + "h\n"
|
||
elif until_start_time_min > 1440:
|
||
return "距开始：" + str(int(until_start_time_min / 60 / 24)) + "days\n"
|
||
else:
|
||
return "距开始：未知\n"
|