ching/toot_bot
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

feat(heatmap): 计算昨日 status 数量并记录

Browse Source 
计算昨日 status 数量并记录

Signed-off-by: Ching <loooching@gmail.com>
This commit is contained in:
Ching 2022-05-25 16:32:01 +08:00
parent ffb514da65
commit 5d271f43db
1 changed files with 30 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
heatmap.py
View File

@ -7,6 +7,7 @@ from mastodon import Mastodon
import pytz import pytz
import time import time
import os import os
import datetime
# logging.basicConfig(filename='/home/captain/dev/log/heatmap.log', level=logging.INFO) # logging.basicConfig(filename='/home/captain/dev/log/heatmap.log', level=logging.INFO)
# logger = logging.getLogger('/home/captain/dev/log/heatmap.log') # logger = logging.getLogger('/home/captain/dev/log/heatmap.log')
@ -29,6 +30,28 @@ if os.path.exists(fn):
date_, count = data_.split(' ') date_, count = data_.split(' ')
date.append(date_) date.append(date_)
data.append(int(count)) data.append(int(count))
me = client.me()
tl = client.account_statuses(me.id)
zh = pytz.timezone('Asia/Shanghai')
while tl:
print(tl[0].created_at)
break_flag = False
for status in tl:
date_ = status.created_at.astimezone(zh).date().strftime('%Y-%m-%d')
yesterday = datetime.datetime.now(pytz.timezone('Asia/Shanghai')).date() - datetime.timedelta(days=1)
# break if date is not yesterday
if date_ != yesterday.strftime('%Y-%m-%d'):
break_flag = True
break
if date_ not in date:
date.append(date_)
data.append(0)
data[date.index(date_)] += 1
if break_flag:
break
time.sleep(1)
tl = client.fetch_next(tl)
else: else:
# init # init
@ -47,17 +70,17 @@ else:
time.sleep(1) time.sleep(1)
tl = client.fetch_next(tl) tl = client.fetch_next(tl)
# write to file # write to file
with open(fn, 'w') as f: with open(fn, 'w') as f:
for i in range(len(date)): for i in range(len(date)):
f.write(date[i] + ' ' + str(data[i]) + '\n') f.write(date[i] + ' ' + str(data[i]) + '\n')
f.close() f.close()
cleaned_date, cleaned_data = july.utils.preprocess_inputs(date, data) cleaned_date, cleaned_data = july.utils.preprocess_inputs(date, data)
july.heatmap(cleaned_date, cleaned_data, title='', cmap="BuGn") july.heatmap(cleaned_date, cleaned_data, cmap="BuGn")
img_path = './heatmap.png' img_path = './heatmap.png'
plt.savefig(img_path) plt.savefig(img_path, bbox_inches='tight')
# upload image to mastodon # upload image to mastodon
media = client.media_post(img_path) media = client.media_post(img_path)