This repository has been archived by the owner on Jul 22, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 0
/
update.py
242 lines (215 loc) · 9.98 KB
/
update.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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
import os
import sys
import time
import tweepy
import datetime
import requests
from PIL import Image
from selenium import webdriver
def fetch_api():
api = 'https://raw.githubusercontent.com/tokyo-metropolitan-gov/covid19/master/data/daily_positive_detail.json'
req = requests.get(api)
json = req.json()
last_update = json['date']
data = json['data']
return last_update, data
def update_lastupdate(last_update, file='lastUpdate.txt'):
with open(file, mode='r') as f:
if f.read() == last_update:
return False
with open(file, mode='w') as f:
f.write(last_update)
return True
def rate_to_color(rate):
if rate >= 2800:
return '#FF0000'
if rate >= 2400:
return '#FF8000'
if rate >= 2000:
return '#C0C000'
if rate >= 1600:
return '#0000FF'
if rate >= 1200:
return '#00C0C0'
if rate >= 800:
return '#008000'
if rate >= 400:
return '#804000'
if rate > 0:
return '#808080'
return '#000000'
def rating_diff(old_rating, new_rating):
return f"{'+' if old_rating < new_rating else ''}{new_rating - old_rating}"
def convert_dict(dct):
ret = []
old_rating = 0
for e in dct:
ret.append({
"ContestName": "東京都新規陽性者数の遷移",
"OldRating": old_rating,
"NewRating": e['count'],
"EndTime": datetime.datetime.strptime(f"{e['diagnosed_date']}+0000", '%Y-%m-%d%z').timestamp(),
"Place": -1,
"StandingsUrl": "#",
})
old_rating = e['count']
return ret
def generate_ogp_html(dct, old_r, new_r, dt):
with open('ogp.html', mode='w') as f:
f.write(f'''\
<html>
<head>
<meta charset="utf-8"/>
<link href="https://fonts.googleapis.com/css?family=Lato:400,700" rel="stylesheet" type="text/css">
<link href="https://fonts.googleapis.com/css?family=Squada+One" rel="stylesheet" type="text/css">
<link rel="stylesheet" type="text/css" href="https://img.atcoder.jp/public/7cd93c2/css/bootstrap.min.css">
</head>
<body>
<div style="display:flex;">
<div style="width:640px;height:440px;">
<canvas id="ratingStatus" width="640" height="80"></canvas><br>
<canvas id="ratingGraph" width="640" height="360"></canvas><br>
</div>
<div style="width:198px;height:440px;background-color:#dddddd;text-align:center;">
<div style="font-family:'Lato';font-size:35px;font-weight:bold;margin-top:40px;margin-bottom:10px;">{last_update[5:-6]} ({"月火水木金土日"[dt.weekday()]})</div>
<div style="font-family:'Squada One';font-size:72px;color:{rate_to_color(old_r)};">{old_r}</div>
<div style="font-size:30px;text-shadow:0.3px 0.3px 0,0.3px -0.3px 0px,-0.3px 0.3px 0,-0.3px -0.3px 0px,0.3px 0px 0px,0px 0.3px 0px,-0.3px 0px 0px,0px -0.3px 0px;">↓</div>
<div style="font-family:'Squada One';font-size:72px;color:{rate_to_color(new_r)};">{new_r}</div>
<div style="font-family:'Lato';font-size:30px;">({rating_diff(old_r, new_r)})</div>
</div>
</div>
<script type="text/javascript" src="https://code.createjs.com/easeljs-0.8.2.min.js"></script>
<script type="text/javascript" src="https://img.atcoder.jp/public/fc4b538/js/lib/jquery-1.9.1.min.js"></script>
<script type="text/javascript" src="https://img.atcoder.jp/public/ad3eaad/js/rating-graph.js"></script>
<script>
const rating_history = {dct};
</script>
</body>
</html>
''')
def save_ogp_image():
driver = webdriver.Chrome()
driver.get(
'file:///home/runner/work/covid19-rating-graph/covid19-rating-graph/ogp.html')
time.sleep(20) # ???
driver.save_screenshot("img/ogp_bef.png")
driver.quit()
im = Image.open('img/ogp_bef.png')
im.crop((0, 0, 838, 440)).save('img/ogp.png', quality=95)
im = Image.open('img/ogp_bef.png')
im.crop((0, 0, 640, 440)).save('img/graph.png', quality=95)
def tweet_graph(tweet, img):
AT = os.environ['TWITTER_ACCESS_TOKEN']
AS = os.environ['TWITTER_ACCESS_TOKEN_SECRET']
CK = os.environ['TWITTER_API_KEY']
CS = os.environ['TWITTER_API_SECRET_KEY']
auth = tweepy.OAuthHandler(CK, CS)
auth.set_access_token(AT, AS)
api = tweepy.API(auth)
media_ids = [api.media_upload(img).media_id]
api.update_status(status=tweet, media_ids=media_ids)
def generate_index_html(dct, tweet):
with open('index.html', mode='w') as f:
f.write(f'''\
<html>
<head>
<title>COVID-19 Rating Graph</title>
<meta charset="utf-8"/>
<link href="https://fonts.googleapis.com/css?family=Lato:400,700" rel="stylesheet" type="text/css">
<link href="https://fonts.googleapis.com/css?family=Squada+One" rel="stylesheet" type="text/css">
<link rel="stylesheet" type="text/css" href="https://img.atcoder.jp/public/7cd93c2/css/bootstrap.min.css">
<link rel="stylesheet" type="text/css" href="style.css">
<meta property="og:url" content="https://morioprog.github.io/covid19-rating-graph/" />
<meta property="og:type" content="website" />
<meta property="og:title" content="COVID-19 Rating Graph" />
<meta property="og:description" content="東京都新規陽性者数の遷移をAtCoderのレーティンググラフ風に表示するサイトです。非公式です。" />
<meta property="og:site_name" content="COVID-19 Rating Graph" />
<meta property="og:image" content="https://raw.githubusercontent.com/morioprog/covid19-rating-graph/main/img/icon.png" />
<meta name="twitter:card" content="summary" />
<meta name="twitter:site" content="@morio_prog" />
</head>
<body>
<div>
<div class="center">
<h1><a href="https://morioprog.github.io/covid19-rating-graph/">COVID-19 Rating Graph</a></h1>
<h4>完全に<span class="emphasize">非公式</span>です。</h4>
<canvas id="ratingStatus" width="640" height="80"></canvas><br>
<canvas id="ratingGraph" width="640" height="360"></canvas><br><br>
<input type="date" id="since"> 〜
<input type="date" id="until">
<input type="button" value="表示" onclick="filterDate()">
<br><br>
</div>
<div class="ul-center">
<ul>
<li>作成者: <a href="https://twitter.com/morio_prog">@morio_prog</a></li>
<li>GitHub: <a href="https://github.com/morioprog/covid19-rating-graph">morioprog/covid19-rating-graph</a></li>
<li>データ: <a href="https://github.com/tokyo-metropolitan-gov/covid19">tokyo-metropolitan-gov/covid19</a></li>
<li>最終更新: {last_update}</li>
</ul>
<div>
<a href="https://twitter.com/share"
data-text="{tweet}"
class="twitter-share-button"
data-show-count="false">
Tweet
</a>
<a href="https://twitter.com/covid19_rating?ref_src=twsrc%5Etfw"
class="twitter-follow-button"
style="display:inline-block;"
data-show-count="false">
Follow @covid19_rating
</a>
</div>
</div>
</div>
<script type="text/javascript" src="https://code.createjs.com/easeljs-0.8.2.min.js"></script>
<script type="text/javascript" src="https://img.atcoder.jp/public/fc4b538/js/lib/jquery-1.9.1.min.js"></script>
<script type="text/javascript" src="https://img.atcoder.jp/public/ad3eaad/js/rating-graph.js"></script>
<script async src="https://platform.twitter.com/widgets.js" charset="utf-8"></script>
<script>
let rating_history = {dct};
const url = new URL(window.location.href);
const params = url.searchParams;
const since = Date.parse(params.get('since')) / 1000 || rating_history[0]['EndTime'];
const until = Date.parse(params.get('until')) / 1000 || rating_history[rating_history.length - 1]['EndTime'];
const since_form = document.getElementById('since');
const until_form = document.getElementById('until');
const sec2datestr = (sec) => new Date(sec * 1000).toISOString().substring(0, 10);
since_form.value = sec2datestr(since);
until_form.value = sec2datestr(until);
const filterDate = () => {{
params.set('since', since_form.value);
params.set('until', until_form.value);
location.href = url;
}};
const filtered_rating_history = rating_history.filter((e) => since <= e['EndTime'] && e['EndTime'] <= until);
if (filtered_rating_history.length > 0) {{
rating_history = filtered_rating_history;
}}
</script>
</body>
</html>
''')
if __name__ == '__main__':
last_update, data = fetch_api()
# No update
if not update_lastupdate(last_update):
sys.exit(1)
old_r = data[-2]['count']
new_r = data[-1]['count']
dct = convert_dict(data)
y, m, d = map(int, last_update[:-6].split('/'))
last_update_dt = datetime.datetime(year=y, month=m, day=d)
generate_ogp_html(dct, old_r, new_r, last_update_dt)
save_ogp_image()
tweet = f'''\
東京都新規陽性者数（{last_update[:-6]}更新）
レーティング：{old_r}→{new_r} ({rating_diff(old_r, new_r)}) {':|' if old_r == new_r else ':(' if old_r < new_r else ':)'}
{'Highestを更新してしまいました...' if max(e['count'] for e in data[:-1]) < data[-1]['count'] else ''}
#COVID19RatingGraph
'''
tweet_graph(
tweet + 'https://morioprog.github.io/covid19-rating-graph/', 'img/ogp.png')
generate_index_html(dct, tweet)
sys.exit(0)