ccc_progress_tracker_bot/main.py

#!/usr/bin/env python

from datetime import date
import json
import os
import requests as r
import secret
from time import time
import traceback

API_CALL = 'https://suya.place/api/v1/statuses'
API_HEADERS = { 'Authorization': 'Bearer %s' % (secret.TOKEN) } # API key here!
EXCEPTION_MESSAGE_FORMAT = '@a1ba\n%s' # send an exception text to my direct messages
KOTCRAB_API = 'https://kotcrab.com/progress/proxy/ccc?x=%d'

def get_stats():
	data = r.get(KOTCRAB_API % int(time() * 1000)).json()
	return data[0]['units'], data[1]['units']

def post(visibility, status):
	resp = r.post(API_CALL, headers = API_HEADERS, json = {
		'visibility': visibility,
		'status' : status
	})
	return resp.status_code == 200

def parse_stats(stats, ref):
	data = {}

	# this stuff isn't documented by Kotcrab so let's be careful here
	if stats.keys() != ref.keys():
		raise Exception

	for k in stats.keys():
		if ref[k]['name'] != stats[k]['name']:
			raise Exception
		all_pages = len(ref[k]['data'])
		edited_pages = int(float(stats[k]['edited']) * 0.01 * all_pages)

		data[stats[k]['name']] = { 'edited': edited_pages, 'all': all_pages }
	return data

def dump_and_compare_data(new):
	s = json.dumps(new, indent = '\t')
	with open('last.json.new', 'w') as fd:
		fd.write(s)

	try:
		with open('last.json', 'r') as fd:
			s = fd.read()
		old = json.loads(s)

		for k in new:
			if k not in old:
				new[k]['update'] = True
				continue

			new[k]['update'] = old[k] != new[k]
	except OSError:
		for k in new:
			new[k]['update'] = False

	os.rename('last.json.new', 'last.json')

def compose_text(data):
#	text = 'Today\'s CCC Editing Progress: \n'
	text = 'CCC editing is finished! 🎉'
	days_until_ccc = (date(2023, 12, 25) - date.today()).days
	if days_until_ccc > 0:
		text += '\n\n%d days until Fate/Extra CCC translation release' % days_until_ccc
	elif days_until_ccc == 0:
		text += '\n\nFate/Extra CCC is going to be released TODAY! 🎉'
	elif days_until_ccc > -14:
		text += '\n\nFate/Extra CCC was released few days ago, did you played it yet?'

#	for k in data:
#		d = data[k]
#		mark = ''
#		if d['update']:
#			if d['edited'] == d['all']:
#				mark = '🎉' # my font can't render this emoji lul
#			else:
#				mark = '⚠️'
#
#		percent = d['edited'] / float(d['all']) * 100.0
#
#		text += '%s: %.2f%% (%d/%d) %s\n' % (k, percent, d['edited'], d['all'], mark)

	days_until_fsr = (date(2023, 9, 29) - date.today()).days
	if days_until_fsr > 0:
		text += '\n\n%d days until Fate/Samurai Remnant release' % days_until_fsr
	elif days_until_fsr == 0:
		text += '\n\nFate/Samurai Remnant is going to be released TODAY! 🎉'
	elif days_until_fsr > -14:
		text += '\n\nFate/Samurai Remnant was released few days ago, did you played it yet?'
	return text

def main():
	stats, ref = get_stats()
	stats = parse_stats(stats, ref)
	dump_and_compare_data(stats)

	text = compose_text(stats)
	print(text)

	# bots are forced to be unlisted on my instance anyway
	post('unlisted', text)

if __name__ == "__main__":
	try:
		main()
	except Exception as e:
		t = traceback.format_exc()
		print(t)
		post('direct', EXCEPTION_MESSAGE_FORMAT % t)