65 lines
1.6 KiB
Python
65 lines
1.6 KiB
Python
|
import sys
|
||
|
|
from datetime import datetime, UTC
|
||
|
|
from typing import Optional
|
||
|
|
import os.path
|
||
|
|
|
||
|
|
from massranker.state import StateT, ItemT
|
||
|
|
from massranker.storage import Storage
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
if len(sys.argv) != 2:
|
||
|
|
print("massranker <ontology>")
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
ontology = sys.argv[1]
|
||
|
|
|
||
|
|
def _ontology_path(fname):
|
||
|
|
return os.path.join("ontologies", ontology, fname)
|
||
|
|
|
||
|
|
def _write_rankings(state: StateT):
|
||
|
|
with open(_ontology_path("rankings.txt"), "wt") as f:
|
||
|
|
for ix, (name, elo) in enumerate(state.dump_rankings(), start=1):
|
||
|
|
f.write("{: >4}. {} ({})\n".format(ix, name, elo))
|
||
|
|
|
||
|
|
|
||
|
|
names = []
|
||
|
|
with open(_ontology_path("names.txt"), "rt") as f:
|
||
|
|
for line in f.read().split("\n"):
|
||
|
|
if line.strip() == "":
|
||
|
|
continue
|
||
|
|
names.append(line)
|
||
|
|
|
||
|
|
with Storage.transact(_ontology_path("rankings.db")) as storage:
|
||
|
|
state: StateT = storage.fetch_last_state()
|
||
|
|
_write_rankings(state)
|
||
|
|
|
||
|
|
while True:
|
||
|
|
state: StateT = storage.fetch_last_state()
|
||
|
|
state.do_tournament(names, _ask)
|
||
|
|
storage.store_state(state)
|
||
|
|
|
||
|
|
_write_rankings(state)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def _ask(item0: ItemT, item1: ItemT) -> Optional[ItemT]:
|
||
|
|
print("Which do you prefer?")
|
||
|
|
print(f" 0: {item0.name}")
|
||
|
|
print(f" 1: {item1.name}")
|
||
|
|
print(f" 2: neither")
|
||
|
|
|
||
|
|
while True:
|
||
|
|
i = input("0/1/2> ")
|
||
|
|
if i == "0":
|
||
|
|
return item0
|
||
|
|
elif i == "1":
|
||
|
|
return item1
|
||
|
|
elif i == "2":
|
||
|
|
return None
|
||
|
|
print(f"unexpected input: {i}")
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|