108 lines
2.3 KiB
Python
108 lines
2.3 KiB
Python
#!/usr/bin/env python3
|
|
from datetime import datetime
|
|
from typing import NamedTuple
|
|
from pathlib import Path
|
|
import json
|
|
from typing import Dict, Iterator, Any
|
|
|
|
from kython import cproperty, fget
|
|
from kython.konsume import dell, zoom, keq, akeq
|
|
from kython.kerror import Res, ytry, unwrap
|
|
from kython.kdatetime import as_utc
|
|
|
|
|
|
_BDIR = Path('/L/Dropbox/data/codeforces')
|
|
|
|
|
|
Cid = int
|
|
|
|
class Contest(NamedTuple):
|
|
cid: Cid
|
|
when: datetime
|
|
|
|
@classmethod
|
|
def make(cls, j) -> 'Contest':
|
|
return cls(
|
|
cid=j['id'],
|
|
when=as_utc(j['startTimeSeconds']),
|
|
)
|
|
|
|
Cmap = Dict[Cid, Contest]
|
|
|
|
def get_contests() -> Cmap:
|
|
last = max(_BDIR.glob('allcontests*.json'))
|
|
j = json.loads(last.read_text())
|
|
d = {}
|
|
for c in j['result']:
|
|
cc = Contest.make(c)
|
|
d[cc.cid] = cc
|
|
return d
|
|
|
|
|
|
def get_latest():
|
|
last = max(_BDIR.glob('codeforces*.json'))
|
|
return json.loads(last.read_text())
|
|
|
|
|
|
class Competition(NamedTuple):
|
|
json: Dict[str, Any]
|
|
cmap: Cmap
|
|
|
|
@cproperty
|
|
def uid(self) -> str:
|
|
return self.contest_id
|
|
|
|
@property
|
|
def contest_id(self):
|
|
return self.json['contestId']
|
|
|
|
def __hash__(self):
|
|
return hash(self.contest_id)
|
|
|
|
@cproperty
|
|
def when(self) -> datetime:
|
|
return self.cmap[self.uid].when
|
|
|
|
@cproperty
|
|
def contest(self) -> str:
|
|
return self.json['contestName']
|
|
|
|
@cproperty
|
|
def summary(self) -> str:
|
|
return f'participated in {self.contest}' # TODO
|
|
|
|
@classmethod
|
|
def make(cls, cmap, json) -> Iterator[Res['Competition']]:
|
|
yield cls(cmap=cmap, json=json)
|
|
yield from ytry(lambda: akeq(json, 'contestId', 'contestName', 'rank', 'oldRating', 'newRating'))
|
|
|
|
|
|
def iter_data() -> Iterator[Res[Competition]]:
|
|
cmap = get_contests()
|
|
|
|
j = get_latest()
|
|
dell(j, 'status')
|
|
|
|
j = zoom(j, 'result')
|
|
|
|
for c in j:
|
|
dell(c, 'handle', 'ratingUpdateTimeSeconds')
|
|
yield from Competition.make(cmap=cmap, json=c)
|
|
|
|
|
|
def get_data():
|
|
return list(sorted(iter_data(), key=fget(Competition.when)))
|
|
|
|
def main():
|
|
for d in iter_data():
|
|
try:
|
|
d = unwrap(d)
|
|
except Exception as e:
|
|
print(f'ERROR! {d}')
|
|
else:
|
|
print(f'{d.when}: {d.summary}')
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|