holy-heck-i-really-like-stats/bot.py

158 lines
4.4 KiB
Python
Raw Normal View History

2023-04-07 10:01:57 +10:00
#!/usr/bin/env python3
2023-04-18 13:29:59 +10:00
from collections import deque
2023-04-07 10:01:57 +10:00
from discord.utils import setup_logging
2023-04-08 09:50:32 +10:00
from typing import Optional
2023-04-07 10:01:57 +10:00
import argparse
import discord
import logging
2023-04-08 16:53:23 +10:00
import os.path
2023-04-08 09:50:32 +10:00
import random
2023-04-07 10:01:57 +10:00
import re
2023-04-07 10:13:11 +10:00
import shutil
2023-04-07 10:01:57 +10:00
import subprocess as sp
discord.utils.setup_logging()
2023-04-08 16:53:23 +10:00
_log = logging.getLogger(__name__)
2023-04-07 10:01:57 +10:00
_GAMES = "games.txt"
2023-04-10 10:05:53 +10:00
_DB = "holy-heck.db"
2023-04-07 10:01:57 +10:00
_DB_DEST = f"/var/lib/grafana/{_DB}"
def _write_game(content: str):
try:
with open(_GAMES, "a") as f:
f.write(content)
f.write("\n")
except:
_log.exception(f"failed writing game {content}")
def _update_db():
try:
games = []
with open(_GAMES) as f:
for line in f:
games.append(line.strip())
sp.run(["./index.py", "-o", _DB] + games)
2023-04-07 10:13:11 +10:00
shutil.move(_DB, _DB_DEST)
2023-04-07 10:13:45 +10:00
_log.info("updated db")
2023-04-07 10:01:57 +10:00
except:
_log.exception(f"failed updating db")
class BotClient(discord.Client):
2023-04-08 09:50:32 +10:00
def __init__(
self,
2023-04-08 16:53:23 +10:00
intents: discord.Intents,
2023-04-08 09:50:32 +10:00
leaguefacts: Optional[list[str]] = None,
):
2023-04-08 16:53:23 +10:00
super().__init__(intents=intents)
2023-04-18 13:29:59 +10:00
self._recentfacts = deque(maxlen=min(len(leaguefacts) - 1, 5))
2023-04-08 09:50:32 +10:00
self._leaguefacts = leaguefacts or []
2023-04-18 13:12:30 +10:00
self._leaguefacts.append(
f"I know {len(self._leaguefacts)} leaguefacts but you'll never see most of them."
)
2023-04-15 11:44:59 +10:00
random.seed()
2023-04-08 09:50:32 +10:00
2023-04-07 10:01:57 +10:00
async def on_ready(self):
_log.info(f"ready as {self.user}")
async def on_message(self, message: discord.Message):
content = message.content
2023-04-26 17:01:02 +10:00
if self.is_command(message):
await self.on_command(message)
elif self.is_replay(message):
2023-04-08 09:50:32 +10:00
await self.on_replay(message)
elif self.is_leaguefact(message):
await self.on_leaguefact(message)
2023-04-26 17:01:02 +10:00
def is_command(self, message: discord.Message) -> bool:
return message.content.startswith("%")
async def on_command(self, message: discord.Message):
2023-04-26 17:06:33 +10:00
match message.content.split(" "):
case ["%calc", *args]:
reply = await self._calculate(args)
2023-04-26 17:01:02 +10:00
await message.reply(content=reply)
case _:
_log.info(f"Unrecognised command {command}")
2023-04-26 17:04:36 +10:00
async def _calculate(self, args: list[str]) -> str:
2023-04-26 17:01:02 +10:00
proc = sp.run(["node", "calc_main.js", "--"] + args)
2023-04-26 17:12:24 +10:00
if proc.returncode != 0:
_log.error(f"running calculator: {proc.stderr}")
msg = proc.stdout or ""
2023-04-26 17:01:02 +10:00
return msg.strip()
2023-04-08 16:53:23 +10:00
def is_replay(self, message: discord.Message) -> bool:
2023-04-08 09:50:32 +10:00
if re.match("https://replay.pokemonshowdown.com/dl-.*", message.content):
return True
return False
async def on_replay(self, message: discord.Message):
_log.info(f"Recognised {message.content} as a League game")
_write_game(message.content)
_update_db()
2023-04-08 16:53:23 +10:00
def is_leaguefact(self, message: discord.Message) -> bool:
2023-04-15 11:44:39 +10:00
return message.content.lower() in ["leaguefact", "leaguefacts"]
2023-04-08 09:50:32 +10:00
async def on_leaguefact(self, message: discord.Message):
2023-04-08 16:54:47 +10:00
_log.info("leaguefact requested")
2023-04-08 09:50:32 +10:00
fact = self._select_leaguefact()
if fact:
2023-04-08 16:53:23 +10:00
await message.channel.send(f"Did you know? {fact}")
else:
await message.channel.send("There are no league facts.")
2023-04-08 09:50:32 +10:00
def _select_leaguefact(self) -> Optional[str]:
if not self._leaguefacts:
return None
2023-04-18 13:29:59 +10:00
choice = None
while True:
choice = random.choice(self._leaguefacts)
if choice not in self._recentfacts:
break
self._recentfacts.append(choice)
return choice
2023-04-07 10:01:57 +10:00
def main():
parser = argparse.ArgumentParser()
parser.add_argument(
"-t",
"--token-file",
metavar="FILE",
default="token",
help="file containing Discord API token",
)
2023-04-08 09:50:32 +10:00
parser.add_argument(
"-f",
"--facts",
metavar="FILE",
default="facts.txt",
help="file containing leagefacts",
)
2023-04-07 10:01:57 +10:00
args = parser.parse_args()
2023-04-08 09:50:32 +10:00
facts = []
2023-04-08 16:53:23 +10:00
if os.path.exists(args.facts):
with open(args.facts) as f:
for line in f:
facts.append(line)
2023-04-08 09:50:32 +10:00
2023-04-07 10:01:57 +10:00
intents = discord.Intents.default()
intents.message_content = True
2023-04-08 09:50:32 +10:00
client = BotClient(leaguefacts=facts, intents=intents)
2023-04-07 10:01:57 +10:00
with open(args.token_file) as f:
token = f.read().strip()
client.run(token, log_handler=None)
if __name__ == "__main__":
main()