-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathbattle_stage.py
118 lines (101 loc) · 4.69 KB
/
battle_stage.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
import urllib
import json
from datetime import datetime, timedelta
from linebot import (
LineBotApi, WebhookHandler
)
from linebot.models import (
MessageEvent, TextMessage, TextSendMessage, ImageSendMessage, ImagemapSendMessage
)
def get_battle_stage(line_bot_api, event, rule):
req = urllib.request.Request(
"https://spla2.yuu26.com/{rule}/now".format(rule=rule))
req.add_header("user-agent", "@nohararc")
with urllib.request.urlopen(req) as res:
response_body = res.read().decode("utf-8")
response_json = json.loads(response_body.split("\n")[0])
data = response_json["result"][0]
now_start = datetime.strptime(data["start"], '%Y-%m-%dT%H:%M:%S')
now_end = datetime.strptime(data["end"], '%Y-%m-%dT%H:%M:%S')
line_bot_api.reply_message(
event.reply_token, [
TextSendMessage(text="{start} ～ {end}\n{rule}\n{stage1}\n{stage2}".format(
start=now_start.strftime("%m/%d %H:%M"), end=now_end.strftime("%m/%d %H:%M"),
rule=data["rule_ex"]["name"],
stage1=data["maps_ex"][0]["name"], stage2=data["maps_ex"][1]["name"]
))
]
)
def get_specified_battle_stage(line_bot_api, event, rule, m):
req = urllib.request.Request(
"https://spla2.yuu26.com/{rule}/schedule".format(rule=rule))
req.add_header("user-agent", "@nohararc")
with urllib.request.urlopen(req) as res:
response_body = res.read().decode("utf-8")
response_json = json.loads(response_body.split("\n")[0])
data = response_json["result"]
for d in data:
start_time = datetime.strptime(d["start"], '%Y-%m-%dT%H:%M:%S')
end_time = datetime.strptime(d["end"], '%Y-%m-%dT%H:%M:%S')
start_hour = start_time.strftime("%H")
if int(m.group(1)) == int(start_hour):
line_bot_api.reply_message(
event.reply_token, [
TextSendMessage(text="{start} ～ {end}\n{rule}\n{stage1}\n{stage2}".format(
start=start_time.strftime("%m/%d %H:%M"),
end=end_time.strftime("%m/%d %H:%M"),
rule=d["rule_ex"]["name"],
stage1=d["maps_ex"][0]["name"], stage2=d["maps_ex"][1]["name"]
))
]
)
def get_battle_stage_all(line_bot_api, event, rule):
req = urllib.request.Request(
"https://spla2.yuu26.com/{rule}/schedule".format(rule=rule))
req.add_header("user-agent", "@nohararc")
with urllib.request.urlopen(req) as res:
response_body = res.read().decode("utf-8")
response_json = json.loads(response_body.split("\n")[0])
data = response_json["result"]
schedule_all = ""
for d in data:
start_time = datetime.strptime(d["start"], '%Y-%m-%dT%H:%M:%S')
end_time = datetime.strptime(d["end"], '%Y-%m-%dT%H:%M:%S')
schedule_all += start_time.strftime("%m/%d %H:%M") + " ～ " + end_time.strftime(
"%m/%d %H:%M") + "\n■" + d["rule_ex"]["name"] + "\n" + d["maps_ex"][0]["name"] + "\n" + d["maps_ex"][1]["name"] + "\n\n"
else:
schedule_all = schedule_all[:-2]
line_bot_api.reply_message(
event.reply_token, [
TextSendMessage(text=schedule_all)
]
)
def get_stage_time_range(line_bot_api, event, m):
hours = list(map(int, m.group("hours").split(m.group("sep"))))
now = datetime.now().strftime("%Y-%m-%d")
start = datetime.strptime(f"{now}{hours[0]}", '%Y-%m-%d%H')
end = datetime.strptime(f"{now}{hours[1]}", '%Y-%m-%d%H')
if hours[0] > hours[1]:
end += timedelta(days=1)
if m.group("rule") in ["ガチマッチ", "ガチマ"]:
rule = "gachi"
else:
rule = "league"
req = urllib.request.Request("https://spla2.yuu26.com/{rule}/schedule".format(rule=rule))
req.add_header("user-agent", "@nohararc")
with urllib.request.urlopen(req) as res:
response_body = res.read().decode("utf-8")
response_json = json.loads(response_body.split("\n")[0])
data = response_json["result"]
text = ""
for d in data:
if start <= datetime.strptime(d["start"], '%Y-%m-%dT%H:%M:%S') <= end:
start_time = datetime.strptime(d["start"], '%Y-%m-%dT%H:%M:%S')
text += start_time.strftime("%m/%d %H:%M") + "\n" + d["rule_ex"]["name"] + "\n" + d["maps_ex"][0]["name"] + "\n" + d["maps_ex"][1]["name"] + "\n\n"
else:
text = text[:-2]
line_bot_api.reply_message(
event.reply_token, [
TextSendMessage(text=text)
]
)