100 lines
2.9 KiB
Python
100 lines
2.9 KiB
Python
import datetime
|
|
import logging
|
|
import matplotlib.pyplot as plt
|
|
import os
|
|
import pytz
|
|
import time
|
|
|
|
|
|
from mastodon import Mastodon
|
|
import july
|
|
|
|
# logging.basicConfig(filename='/home/captain/dev/log/heatmap.log', level=logging.INFO)
|
|
# logger = logging.getLogger('/home/captain/dev/log/heatmap.log')
|
|
banner_id = 108361439460735817
|
|
|
|
client = Mastodon(
|
|
access_token="Ug_bUMWCk3RLamOnqYIytmeB0nO6aNfjdmf06mAj2bE",
|
|
api_base_url="https://nofan.xyz",
|
|
)
|
|
|
|
fn = "./heatmap.txt"
|
|
# read from file if exists
|
|
date = []
|
|
data = []
|
|
if os.path.exists(fn):
|
|
with open(fn) as f:
|
|
data_list = f.read().splitlines()
|
|
f.close()
|
|
for data_ in data_list:
|
|
date_, count = data_.split(" ")
|
|
date.append(date_)
|
|
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:
|
|
# init
|
|
me = client.me()
|
|
tl = client.account_statuses(me.id)
|
|
zh = pytz.timezone("Asia/Shanghai")
|
|
|
|
while tl:
|
|
print(tl[0].created_at)
|
|
for status in tl:
|
|
date_ = status.created_at.astimezone(zh).date().strftime("%Y-%m-%d")
|
|
if date_ not in date:
|
|
date.append(date_)
|
|
data.append(0)
|
|
data[date.index(date_)] += 1
|
|
time.sleep(1)
|
|
tl = client.fetch_next(tl)
|
|
|
|
# write to file
|
|
with open(fn, "w") as f:
|
|
for i in range(len(date)):
|
|
f.write(date[i] + " " + str(data[i]) + "\n")
|
|
f.close()
|
|
|
|
cleaned_date, cleaned_data = july.utils.preprocess_inputs(date, data)
|
|
july.heatmap(cleaned_date, cleaned_data, cmap="BuGn")
|
|
|
|
img_path = "./heatmap.png"
|
|
plt.savefig(img_path, bbox_inches="tight")
|
|
|
|
# upload image to mastodon
|
|
media = client.media_post(img_path)
|
|
|
|
headers = {}
|
|
headers["Authorization"] = "Bearer " + client.access_token
|
|
params = {"status": "今天浑浑噩噩了吗", "media_ids": [media.id]}
|
|
kwargs = dict(headers=headers, files={}, timeout=client.request_timeout, json=params)
|
|
url = client.api_base_url + "/api/v1/statuses/%s/" % banner_id
|
|
resp = client.session.request("PUT", url, **kwargs)
|
|
if resp.status_code == 200:
|
|
client.session.request(
|
|
"GET", "https://up.tunpok.com/api/push/qku6CYquGu?msg=OK&ping="
|
|
)
|