Compare commits
No commits in common. "262a0513569e251937751e962375460a9dd68136" and "641ddef2095d3d454b168e2a53036b598d2104d0" have entirely different histories.
262a051356
...
641ddef209
91
main.py
91
main.py
@ -7,7 +7,6 @@ from pathlib import Path
|
||||
import argparse
|
||||
import json
|
||||
import logging
|
||||
import re
|
||||
import requests
|
||||
import sqlite3
|
||||
import sys
|
||||
@ -54,7 +53,7 @@ def team(player: str) -> str:
|
||||
if player in TEAMS:
|
||||
return TEAMS[player]
|
||||
else:
|
||||
if not player in _logged_teams and player:
|
||||
if not player in _logged_teams:
|
||||
LOG.warning(f"missing team mapping for {player}")
|
||||
_logged_teams.append(player)
|
||||
return player
|
||||
@ -101,12 +100,7 @@ def _init_db(conn: sqlite3.Connection):
|
||||
CREATE TABLE IF NOT EXISTS games(
|
||||
id, p1, p2, format, uploadtime, winner,
|
||||
UNIQUE(id)
|
||||
);
|
||||
-- No good way to ensure idempotence for damage; just re-build it.
|
||||
DROP TABLE IF EXISTS damage;
|
||||
CREATE TABLE damage(game, player, name, value);
|
||||
DROP TABLE IF EXISTS indirect_damage;
|
||||
CREATE TABLE indirect_damage(game, player, name, value);
|
||||
)
|
||||
"""
|
||||
)
|
||||
|
||||
@ -116,7 +110,6 @@ def parse_log(game: str, log: str, into: sqlite3.Connection):
|
||||
|
||||
turn = 0
|
||||
players = {}
|
||||
hp = {}
|
||||
|
||||
# ("p2a: Edward", "p1a: Meteo")
|
||||
# memorises the user of the move that causes environment setting or status,
|
||||
@ -143,10 +136,8 @@ def parse_log(game: str, log: str, into: sqlite3.Connection):
|
||||
match chunks:
|
||||
case ["player", id, username, *rest]:
|
||||
players[id] = username
|
||||
|
||||
case ["turn", turn]:
|
||||
turn = int(turn)
|
||||
|
||||
case ["move", user, move, target]:
|
||||
last_move = (user, target)
|
||||
player, user = resolve_mon(user)
|
||||
@ -159,13 +150,7 @@ def parse_log(game: str, log: str, into: sqlite3.Connection):
|
||||
""",
|
||||
(game, turn, team(player), move, user, target),
|
||||
)
|
||||
|
||||
case ["drag", name, specie, status, *rest]:
|
||||
hp[name] = int(status.split("/")[0])
|
||||
|
||||
case ["switch", name, specie, status, *rest]:
|
||||
hp[name] = int(status.split("/")[0])
|
||||
|
||||
case ["switch", name, specie, *rest]:
|
||||
player, name = resolve_mon(name)
|
||||
conn.execute(
|
||||
"""
|
||||
@ -183,7 +168,6 @@ def parse_log(game: str, log: str, into: sqlite3.Connection):
|
||||
""",
|
||||
(game, team(player), name, specie.split(", ")[0]),
|
||||
)
|
||||
|
||||
case ["faint", mon]:
|
||||
player, mon = resolve_mon(mon)
|
||||
conn.execute(
|
||||
@ -194,7 +178,6 @@ def parse_log(game: str, log: str, into: sqlite3.Connection):
|
||||
""",
|
||||
(game, turn, team(player), mon),
|
||||
)
|
||||
|
||||
case ["win", player]:
|
||||
conn.execute(
|
||||
"""
|
||||
@ -204,60 +187,24 @@ def parse_log(game: str, log: str, into: sqlite3.Connection):
|
||||
""",
|
||||
(team(player), game),
|
||||
)
|
||||
|
||||
case ["-sidestart", side, env]:
|
||||
if not last_move:
|
||||
LOG.warning(f"missing previous move for {line}")
|
||||
continue
|
||||
LOG.debug(f"{line} <- {last_move}")
|
||||
last_env_set[(side[0:1], env.replace("move: ", ""))] = last_move[0]
|
||||
|
||||
case ["-status", mon, cond]:
|
||||
if not last_move or last_move[1] != mon:
|
||||
LOG.warning(f"missing previous move for {line}")
|
||||
continue
|
||||
LOG.debug(f"{line} <- {last_move}")
|
||||
last_status_set[(mon, cond)] = last_move[0]
|
||||
|
||||
case ["-damage", mon, status]:
|
||||
# mon takes direct (non-hazard/condition) damage
|
||||
# status can be a percentage 70/100 with or without condition,
|
||||
# or "0 fnt"
|
||||
new_hp = int(re.split("[/ ]", status)[0])
|
||||
LOG.debug(f"{mon} dropped to {new_hp} from {hp[mon]}")
|
||||
LOG.debug(f"source: {last_move}")
|
||||
|
||||
# resolve to damage source
|
||||
if last_move[1] != mon:
|
||||
LOG.warn(
|
||||
f"{mon} took direct damage but last move was not targeted at them"
|
||||
)
|
||||
continue
|
||||
user = last_move[0]
|
||||
source_player, source_mon = resolve_mon(user)
|
||||
|
||||
conn.execute(
|
||||
"""
|
||||
INSERT INTO damage(game, player, name, value)
|
||||
VALUES(?, ?, ?, ?)
|
||||
ON CONFLICT DO NOTHING
|
||||
""",
|
||||
(game, team(source_player), source_mon, hp[mon] - new_hp),
|
||||
)
|
||||
|
||||
hp[mon] = new_hp
|
||||
|
||||
case ["-damage", mon, status, from_]:
|
||||
# mon takes indirect damage
|
||||
# status can be a percentage 70/100 with or without condition,
|
||||
# or "0 fnt"
|
||||
# mon has fainted from an indirect damage source
|
||||
#
|
||||
new_hp = int(re.split("[/ ]", status)[0])
|
||||
LOG.debug(f"{mon} dropped to {new_hp} from {from_}")
|
||||
|
||||
case ["-damage", mon, *rest]:
|
||||
# rest is new_hp and sometimes a source (if not from a move)
|
||||
# in a knockout, new_hp is "0 fnt"
|
||||
if rest[0] == "0 fnt" and len(rest) > 1:
|
||||
LOG.debug(f"tracing source for {line}")
|
||||
source = from_.replace("[from] ", "")
|
||||
source = rest[1].replace("[from] ", "")
|
||||
source_user = None
|
||||
|
||||
test_hazard = last_env_set.get((mon[0:1], source))
|
||||
@ -279,22 +226,12 @@ def parse_log(game: str, log: str, into: sqlite3.Connection):
|
||||
LOG.error(f"missing source for {line}")
|
||||
continue
|
||||
|
||||
player, pkmn = resolve_mon(mon)
|
||||
player, mon = resolve_mon(mon)
|
||||
if source_user.startswith("p1") or source_user.startswith("p2"):
|
||||
source_player, source_mon = resolve_mon(source_user)
|
||||
source_player, source_user = resolve_mon(source_user)
|
||||
else:
|
||||
source_player = None
|
||||
|
||||
if source_player:
|
||||
conn.execute(
|
||||
"""
|
||||
INSERT INTO indirect_damage(game, player, name, value)
|
||||
VALUES(?, ?, ?, ?)
|
||||
""",
|
||||
(game, team(source_player), source_mon, hp[mon] - new_hp),
|
||||
)
|
||||
|
||||
if status == "0 fnt":
|
||||
conn.execute(
|
||||
"""
|
||||
INSERT INTO indirect_knockouts(game, turn, player, name, source, source_user, source_player)
|
||||
@ -305,16 +242,12 @@ def parse_log(game: str, log: str, into: sqlite3.Connection):
|
||||
game,
|
||||
turn,
|
||||
team(player),
|
||||
pkmn,
|
||||
mon,
|
||||
source,
|
||||
source_mon,
|
||||
source_user,
|
||||
team(source_player),
|
||||
),
|
||||
)
|
||||
|
||||
case ["-heal", mon, status, *rest]:
|
||||
hp[mon] = int(status.split("/")[0])
|
||||
|
||||
case _:
|
||||
# LOG.debug(f"unhandled message {chunks[0]}")
|
||||
pass
|
||||
|
Loading…
Reference in New Issue
Block a user