heatmap.py

@@ -1,100 +0,0 @@
-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",
-)
-
-base_dir = os.path.dirname(os.path.abspath(__file__))
-fn = os.path.join(base_dir, "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", month_grid=False, weekday_label=False, month_label=False)
-
-img_path = os.path.join(base_dir, "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="
-    )