track damage

This commit is contained in:
xeals 2023-04-03 21:55:15 +10:00
parent 7138482b88
commit 85296a2817
Signed by: xeals
GPG Key ID: A498C7AF27EC6B5C

74
main.py
View File

@ -7,6 +7,7 @@ from pathlib import Path
import argparse
import json
import logging
import re
import requests
import sqlite3
import sys
@ -100,7 +101,12 @@ 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);
"""
)
@ -110,6 +116,7 @@ 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,
@ -153,7 +160,12 @@ def parse_log(game: str, log: str, into: sqlite3.Connection):
(game, turn, team(player), move, user, target),
)
case ["switch", name, specie, *rest]:
case ["drag", name, specie, status, *rest]:
hp[name] = int(status.split("/")[0])
case ["switch", name, specie, status, *rest]:
hp[name] = int(status.split("/")[0])
player, name = resolve_mon(name)
conn.execute(
"""
@ -207,8 +219,43 @@ def parse_log(game: str, log: str, into: sqlite3.Connection):
LOG.debug(f"{line} <- {last_move}")
last_status_set[(mon, cond)] = last_move[0]
case ["-damage", mon, "0 fnt", from_]:
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_}")
LOG.debug(f"tracing source for {line}")
source = from_.replace("[from] ", "")
source_user = None
@ -232,12 +279,22 @@ def parse_log(game: str, log: str, into: sqlite3.Connection):
LOG.error(f"missing source for {line}")
continue
player, mon = resolve_mon(mon)
player, pkmn = resolve_mon(mon)
if source_user.startswith("p1") or source_user.startswith("p2"):
source_player, source_user = resolve_mon(source_user)
source_player, source_mon = 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)
@ -248,13 +305,16 @@ def parse_log(game: str, log: str, into: sqlite3.Connection):
game,
turn,
team(player),
mon,
pkmn,
source,
source_user,
source_mon,
team(source_player),
),
)
case ["-heal", mon, status, *rest]:
hp[mon] = int(status.split("/")[0])
case _:
# LOG.debug(f"unhandled message {chunks[0]}")
pass