Compare commits
No commits in common. "dfc1093b393d50cc31ef2eecce4227396d3905da" and "262a0513569e251937751e962375460a9dd68136" have entirely different histories.
dfc1093b39
...
262a051356
3
.gitignore
vendored
3
.gitignore
vendored
@ -8,6 +8,5 @@ tmp/
|
||||
build
|
||||
htmlcov
|
||||
|
||||
*.db
|
||||
data.db
|
||||
cache/
|
||||
token
|
||||
|
@ -11,8 +11,8 @@ fuck.
|
||||
## Usage
|
||||
|
||||
```sh
|
||||
$ ./index.py -h
|
||||
usage: hhirlstats [-h] [-v] [-C] [-t FILE] [-o FILE] replay [replay ...]
|
||||
$ ./main.py -h
|
||||
usage: hhirlstats [-h] [-v] [-C] [-o FILE] replay [replay ...]
|
||||
|
||||
extracts stats from a Showdown replay
|
||||
|
||||
@ -23,8 +23,6 @@ options:
|
||||
-h, --help show this help message and exit
|
||||
-v, --verbose add debugging info (default: None)
|
||||
-C, --no-cache fetch replays instead of using cache (default: False)
|
||||
-t FILE, --teams FILE
|
||||
JSON file defining players to teams (default: teams.json)
|
||||
-o FILE, --output FILE
|
||||
output data file (default: data.db)
|
||||
```
|
||||
|
74
bot.py
74
bot.py
@ -1,74 +0,0 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
from discord.utils import setup_logging
|
||||
import argparse
|
||||
import discord
|
||||
import logging
|
||||
import re
|
||||
import shutil
|
||||
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)
|
||||
shutil.move(_DB, _DB_DEST)
|
||||
_log.info("updated db")
|
||||
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()
|
@ -13,7 +13,6 @@
|
||||
buildInputs =
|
||||
let
|
||||
python = pkgs.python3.withPackages (ps: [
|
||||
ps.discordpy
|
||||
ps.requests
|
||||
]);
|
||||
in
|
||||
|
@ -358,7 +358,7 @@ def fetch(replay: str, cache: bool = True) -> Replay:
|
||||
return Replay(**data)
|
||||
|
||||
|
||||
def main():
|
||||
def main(args):
|
||||
parser = argparse.ArgumentParser(
|
||||
prog=APP,
|
||||
description="extracts stats from a Showdown replay",
|
||||
@ -389,7 +389,7 @@ def main():
|
||||
)
|
||||
parser.add_argument("replay", nargs="+", help="replay ID or URL")
|
||||
|
||||
args = parser.parse_args()
|
||||
args = parser.parse_args(args)
|
||||
if args.verbose and args.verbose > 1:
|
||||
LOG.setLevel(logging.TRACE)
|
||||
elif args.verbose:
|
||||
@ -435,4 +435,4 @@ def main():
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
main(sys.argv[1:])
|
Loading…
Reference in New Issue
Block a user