roster: extract to a separate module

This commit is contained in:
rr-
2018-11-05 21:04:39 +01:00
parent 136db3ca14
commit bec02d2b2c
5 changed files with 28 additions and 18 deletions
+23
View File
@@ -0,0 +1,23 @@
import pickle
import typing as T
from pathlib import Path
from pqcli.mechanic import Player
class Roster:
def __init__(self) -> None:
self.players: T.List[Player] = []
def add_player(self, player: Player) -> None:
self.players.append(player)
@staticmethod
def load(path: Path) -> "Roster":
if path.exists():
return pickle.loads(path.read_bytes())
return Roster()
def save(self, path: Path) -> None:
path.parent.mkdir(parents=True, exist_ok=True)
path.write_bytes(pickle.dumps(self))