Kelder/pancakecounter
32
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Update table.py for new format

Browse source
This commit is contained in:
Midgard 2020-12-09 15:07:31 +01:00
parent 228e44838a
commit 3d1ef9309f
Signed by: midgard
GPG key ID: 511C112F1331BBB4
1 changed files with 30 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

51
table.py
View file

@ -4,51 +4,60 @@ import sys
import re import re
from datetime import datetime from datetime import datetime
from collections import defaultdict from collections import defaultdict
from functools import partial from typing import Mapping, Set
if sys.stdin.isatty(): if sys.stdin.isatty():
print("Hint: stdin is a terminal, you may want to do `./make_table.py < verifications.log` instead.", file=sys.stderr) print("Hint: stdin is a terminal, you may want to do `./make_table.py < verifications.log` instead.", file=sys.stderr)
users = {} # users[awardee][post_id][verifier]: score
users: Mapping[str, Mapping[str, Mapping[str, int]]] = \
defaultdict(lambda: defaultdict(lambda: defaultdict(lambda: 0)))
for line_nr, line in enumerate(sys.stdin, start=1): for line_nr, line in enumerate(sys.stdin, start=1):
line = line.rstrip() line = line.rstrip()
m = re.fullmatch(r"([^ ]+) ([^ ]+) at ([^ ]+) verified by ([^ ]+) at ([^ ]+)", line) # {awardee} {post['id']} at {post_time}: {awarder} verified with score {score}
m = re.fullmatch(r"([^ ]+) ([^ ]+) at ([^ ]+): ([^ ]+) verified with score ([^ ]+)", line)
if m: if m:
awardee = m.group(1) awardee = m.group(1)
post_id = m.group(2) post_id = m.group(2)
post_time = datetime.fromisoformat(m.group(3)) post_time = datetime.fromisoformat(m.group(3))
verifier = m.group(4) verifier = m.group(4)
verification_time = datetime.fromisoformat(m.group(5)) score = int(m.group(5))
if awardee not in users: users[awardee][post_id][verifier] = score
users[awardee] = {}
if post_id not in users[awardee]:
users[awardee][post_id] = defaultdict(dict)
users[awardee][post_id][verifier] = verification_time
continue continue
m = re.fullmatch(r"([^ ]+) ([^ ]+) verification removed by ([^ ]+)", line) # {awardee} {post['id']} at {post_time}: {awarder} updated their verification's score from {prev_score} to {score}
m = re.fullmatch(r"([^ ]+) ([^ ]+) at ([^ ]+): ([^ ]+) updated their verification's score from ([^ ]+) to ([^ ]+)", line)
if m: if m:
awardee = m.group(1) awardee = m.group(1)
post_id = m.group(2) post_id = m.group(2)
verifier = m.group(3) post_time = datetime.fromisoformat(m.group(3))
verifier = m.group(4)
prev_score = int(m.group(5))
score = int(m.group(6))
try: users[awardee][post_id][verifier] = score
del users[awardee][post_id][verifier] continue
except KeyError:
print(f"Line {line_nr}: trying to remove non-existing verification by {verifier}, looks like the file is corrupt!", file=sys.stderr)
# If no more verifiers left, clean up the dicts, this makes it easier to count # {awardee} {post['id']} at {post_time}: {awarder} retracted their verification
if not users[awardee][post_id]: m = re.fullmatch(r"([^ ]+) ([^ ]+) at ([^ ]+): ([^ ]+) retracted their verification", line)
del users[awardee][post_id] if m:
if not users[awardee]: awardee = m.group(1)
del users[awardee] post_id = m.group(2)
post_time = datetime.fromisoformat(m.group(3))
verifier = m.group(4)
users[awardee][post_id][verifier] = 0
continue continue
print(f"Couldn't parse line {line_nr}, looks like the file is corrupt: {line}", file=sys.stderr) print(f"Couldn't parse line {line_nr}, looks like the file is corrupt: {line}", file=sys.stderr)
for username, user_posts in users.items(): for username, user_posts in users.items():
print(f"{username}: {len(user_posts)}") score = sum(
max(post_verifications.values(), default=0)
for post_verifications in user_posts.values()
)
if score > 0:
print(f"{username}: {score}")