Compare commits

...

1 Commits

Author SHA1 Message Date
4de2b8b8bb
add bot 2023-04-07 10:09:41 +10:00
3 changed files with 76 additions and 1 deletions

3
.gitignore vendored
View File

@ -8,5 +8,6 @@ tmp/
build
htmlcov
data.db
*.db
cache/
token

73
bot.py Executable file
View File

@ -0,0 +1,73 @@
#!/usr/bin/env python3
from discord.utils import setup_logging
import argparse
import discord
import logging
import os
import re
import subprocess as sp
discord.utils.setup_logging()
_log = logging.getLogger("statbot")
_GAMES = "games.txt"
_DB = "holy-heck2.db"
_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)
os.rename(_DB, _DB_DEST)
except:
_log.exception(f"failed updating db")
class BotClient(discord.Client):
async def on_ready(self):
_log.info(f"ready as {self.user}")
async def on_message(self, message: discord.Message):
content = message.content
if re.match("https://replay.pokemonshowdown.com/dl-.*", content):
_log.info(f"Recognised {content} as a League game")
_write_game(content)
_update_db()
def main():
parser = argparse.ArgumentParser()
parser.add_argument(
"-t",
"--token-file",
metavar="FILE",
default="token",
help="file containing Discord API token",
)
args = parser.parse_args()
intents = discord.Intents.default()
intents.message_content = True
client = BotClient(intents=intents)
with open(args.token_file) as f:
token = f.read().strip()
client.run(token, log_handler=None)
if __name__ == "__main__":
main()

View File

@ -13,6 +13,7 @@
buildInputs =
let
python = pkgs.python3.withPackages (ps: [
ps.discordpy
ps.requests
]);
in