pennywise/aocbot
1
0
Fork 0
generated from public/aoc_template
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add some logging to stdout

Browse Source
This commit is contained in:
Stefan Harmuth 2024-11-30 07:45:04 +01:00
parent 8109e8a85f
commit c44a4cf451
1 changed files with 39 additions and 67 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

106
aoc_bot.py
View File

@ -1,8 +1,6 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import argparse import argparse
import json
import os.path import os.path
import requests import requests
import sys import sys
from datetime import datetime, timedelta from datetime import datetime, timedelta
@ -44,11 +42,11 @@ class AOCBot:
self.__cache_file = config["AOC"]["cache_file"] self.__cache_file = config["AOC"]["cache_file"]
except AttributeError as e: except AttributeError as e:
print("CONFIGURATION ERROR: %s" % e) print("CONFIGURATION ERROR: %s" % e)
sys.exit(1)
def fetch_leaderboard(self, year: int = datetime.now().year) -> dict: def fetch_leaderboard(self, year: int = datetime.now().year) -> dict:
return requests.get( return requests.get(
"https://adventofcode.com/%d/leaderboard/private/view/%s.json" "https://adventofcode.com/%d/leaderboard/private/view/%s.json" % (year, self.__aoc_group_id),
% (year, self.__aoc_group_id),
headers={"User-Agent": self.__aoc_user_agent}, headers={"User-Agent": self.__aoc_user_agent},
cookies={"session": self.__aoc_session_id}, cookies={"session": self.__aoc_session_id},
).json() ).json()
@ -56,8 +54,7 @@ class AOCBot:
def command_info(self, msg_from: str, message: str): def command_info(self, msg_from: str, message: str):
self.__irc_bot.privmsg( self.__irc_bot.privmsg(
msg_from, msg_from,
"I am %s => %s" "I am %s => %s" % (self.__irc_bot.getUser().nickname, self.__irc_bot.getUser().username),
% (self.__irc_bot.getUser().nickname, self.__irc_bot.getUser().username),
) )
self.__irc_bot.privmsg(msg_from, "I am currently in the following channels:") self.__irc_bot.privmsg(msg_from, "I am currently in the following channels:")
for c in self.__irc_bot.getChannelList(): for c in self.__irc_bot.getChannelList():
@ -79,17 +76,12 @@ class AOCBot:
today_list = [] today_list = []
for member, member_data in self.__cache_data.items(): for member, member_data in self.__cache_data.items():
if ( if not member.startswith("__") and "days" in member_data and day in member_data["days"]:
not member.startswith("__")
and "days" in member_data
and day in member_data["days"]
):
today_list.append(member) today_list.append(member)
self.__irc_bot.privmsg( self.__irc_bot.privmsg(
self.__irc_channel, self.__irc_channel,
"Day %s's leaderboard (last updated: %s):" "Day %s's leaderboard (last updated: %s):" % (day, self.__cache_data["__last_update__"]),
% (day, self.__cache_data["__last_update__"]),
) )
for i, member in enumerate( for i, member in enumerate(
sorted( sorted(
@ -103,16 +95,14 @@ class AOCBot:
if "1" in self.__cache_data[member]["days"][day]: if "1" in self.__cache_data[member]["days"][day]:
p1_time = "in " + human_readable_time_from_delta( p1_time = "in " + human_readable_time_from_delta(
datetime.fromisoformat(self.__cache_data[member]["days"][day]["1"]) datetime.fromisoformat(self.__cache_data[member]["days"][day]["1"]) - day_start
- day_start
) )
else: else:
p1_time = "*not yet solved*" p1_time = "*not yet solved*"
if "2" in self.__cache_data[member]["days"][day]: if "2" in self.__cache_data[member]["days"][day]:
p2_time = "in " + human_readable_time_from_delta( p2_time = "in " + human_readable_time_from_delta(
datetime.fromisoformat(self.__cache_data[member]["days"][day]["2"]) datetime.fromisoformat(self.__cache_data[member]["days"][day]["2"]) - day_start
- day_start
) )
else: else:
p2_time = "not yet solved" p2_time = "not yet solved"
@ -137,15 +127,10 @@ class AOCBot:
sys.exit(0) sys.exit(0)
def on_raw(self, msg_from: str, msg_type: str, msg_to: str, message: str): def on_raw(self, msg_from: str, msg_type: str, msg_to: str, message: str):
print( print("[%s] <%s> (%s) -> <%s>: %s" % (datetime.now().strftime("%H:%M:%S"), msg_from, msg_type, msg_to, message))
"[%s] <%s> (%s) -> <%s>: %s"
% (datetime.now().strftime("%H:%M:%S"), msg_from, msg_type, msg_to, message)
)
def calc_scores(self): def calc_scores(self):
member_count = len( member_count = len([x for x in self.__cache_data.keys() if not x.startswith("__")])
[x for x in self.__cache_data.keys() if not x.startswith("__")]
)
for day in map(str, range(1, 26)): for day in map(str, range(1, 26)):
p1_times = [] p1_times = []
p2_times = [] p2_times = []
@ -163,22 +148,12 @@ class AOCBot:
if member.startswith("__") or day not in member_data["days"]: if member.startswith("__") or day not in member_data["days"]:
continue continue
if ( if "1" in member_data["days"][day] and member_data["days"][day]["1"] in p1_times:
"1" in member_data["days"][day] score = member_count - sorted(p1_times).index(member_data["days"][day]["1"])
and member_data["days"][day]["1"] in p1_times
):
score = member_count - sorted(p1_times).index(
member_data["days"][day]["1"]
)
self.__cache_data[member]["days"][day]["score"] += score self.__cache_data[member]["days"][day]["score"] += score
if ( if "2" in member_data["days"][day] and member_data["days"][day]["2"] in p2_times:
"2" in member_data["days"][day] score = member_count - sorted(p2_times).index(member_data["days"][day]["2"])
and member_data["days"][day]["2"] in p2_times
):
score = member_count - sorted(p2_times).index(
member_data["days"][day]["2"]
)
self.__cache_data[member]["days"][day]["score"] += score self.__cache_data[member]["days"][day]["score"] += score
def update_leaderboard(self): def update_leaderboard(self):
@ -210,30 +185,21 @@ class AOCBot:
for part in member_data["completion_day_level"][day]: for part in member_data["completion_day_level"][day]:
if part not in self.__cache_data[member]["days"][day]: if part not in self.__cache_data[member]["days"][day]:
completion_time = datetime.fromtimestamp( completion_time = datetime.fromtimestamp(
member_data["completion_day_level"][day][part][ member_data["completion_day_level"][day][part]["get_star_ts"]
"get_star_ts"
]
) )
if member_data["name"] not in new_stars: if member_data["name"] not in new_stars:
new_stars[member_data["name"]] = {} new_stars[member_data["name"]] = {}
finishing_time = human_readable_time_from_delta( finishing_time = human_readable_time_from_delta(completion_time - day_start)
completion_time - day_start new_stars[member_data["name"]]["d" + day + "p" + part] = finishing_time
) self.__cache_data[member]["days"][day][part] = completion_time.isoformat()
new_stars[member_data["name"]][
"d" + day + "p" + part
] = finishing_time
self.__cache_data[member]["days"][day][
part
] = completion_time.isoformat()
if len(new_stars) > 0: if len(new_stars) > 0:
self.__irc_bot.privmsg(self.__irc_channel, "New Stars found:") self.__irc_bot.privmsg(self.__irc_channel, "New Stars found:")
for member, parts in new_stars.items(): for member, parts in new_stars.items():
line = member + ": " line = member + ": "
line += ", ".join( line += ", ".join(
"%s (%s)" % (part, new_stars[member][part]) "%s (%s)" % (part, new_stars[member][part]) for part in sorted(new_stars[member].keys())
for part in sorted(new_stars[member].keys())
) )
self.__irc_bot.privmsg(self.__irc_channel, line) self.__irc_bot.privmsg(self.__irc_channel, line)
@ -243,11 +209,17 @@ class AOCBot:
self.calc_scores() self.calc_scores()
def start(self): def start(self):
print("[AoCBot] Loading Cache File")
self.__cache_data = JSONFile(self.__cache_file, create=True) self.__cache_data = JSONFile(self.__cache_file, create=True)
self.__aoc_session_id = (
open(self.__aoc_session_file, "r").readlines()[0].strip()
)
print("[AoCBot] Loading Session ID")
try:
self.__aoc_session_id = open(self.__aoc_session_file, "r").readlines()[0].strip()
except FileNotFoundError:
print("[AoCBot] Session ID not found: %s" % self.__aoc_session_file)
sys.exit(1)
print("[AoCBot] Starting IrcBot")
self.__irc_bot = IrcBot( self.__irc_bot = IrcBot(
server=self.__irc_server, server=self.__irc_server,
port=self.__irc_port, port=self.__irc_port,
@ -255,22 +227,22 @@ class AOCBot:
username=self.__irc_user, username=self.__irc_user,
realname=self.__irc_real_name, realname=self.__irc_real_name,
) )
print("[AoCBot] Joining Channel %s" % self.__irc_channel)
self.__irc_bot.join(self.__irc_channel) self.__irc_bot.join(self.__irc_channel)
self.__irc_bot.schedule(
"update_leaderboard", timedelta(minutes=15), self.update_leaderboard print("[AoCBot] Scheduling Leaderboard Update for Group ID %s" % self.__aoc_group_id)
) self.__irc_bot.schedule("update_leaderboard", timedelta(minutes=15), self.update_leaderboard)
print("[AoCBot] Registering Commands")
self.__irc_bot.on(ServerMessage.RAW, self.on_raw) self.__irc_bot.on(ServerMessage.RAW, self.on_raw)
self.__irc_bot.register_channel_command( self.__irc_bot.register_channel_command("!info", self.__irc_channel, self.command_info)
"!info", self.__irc_channel, self.command_info self.__irc_bot.register_channel_command("!today", self.__irc_channel, self.command_today)
) self.__irc_bot.register_channel_command("!day", self.__irc_channel, self.command_today)
self.__irc_bot.register_channel_command(
"!today", self.__irc_channel, self.command_today
)
self.__irc_bot.register_channel_command(
"!day", self.__irc_channel, self.command_today
)
self.__irc_bot.register_privmsg_command("info", self.command_info) self.__irc_bot.register_privmsg_command("info", self.command_info)
self.__irc_bot.register_privmsg_command("quit", self.command_quit) self.__irc_bot.register_privmsg_command("quit", self.command_quit)
print("[AoCBot] Starting Main Loop")
self.__irc_bot.run() self.__irc_bot.run()