68 lines
1.9 KiB
Python
68 lines
1.9 KiB
Python
from pathlib import Path
|
|
import json
|
|
import shutil
|
|
|
|
from warchron.model.player import Player
|
|
|
|
class Model:
|
|
def __init__(self):
|
|
self.players = {}
|
|
|
|
def new(self):
|
|
self.players.clear()
|
|
# self.wars.clear()
|
|
# self.campaigns.clear()
|
|
# self.rounds.clear()
|
|
|
|
def load(self, path: Path):
|
|
self.players.clear()
|
|
self._load_data(path)
|
|
|
|
def save(self, path: Path):
|
|
self._save_data(path)
|
|
|
|
def _load_data(self, path: Path):
|
|
if not path.exists() or path.stat().st_size == 0:
|
|
return # Start empty
|
|
try:
|
|
with open(path, "r", encoding="utf-8") as f:
|
|
data = json.load(f)
|
|
for player in data["players"] :
|
|
saved_player = Player.fromDict(player["id"], player['name'])
|
|
self.players[saved_player.id] = saved_player
|
|
for war in data["wars"]:
|
|
# placeholder
|
|
pass
|
|
except json.JSONDecodeError:
|
|
raise RuntimeError("Data file is corrupted")
|
|
|
|
def _save_data(self, path: Path):
|
|
if path.exists():
|
|
shutil.copy(path, path.with_suffix(".json.bak"))
|
|
data = {}
|
|
data['version'] = "1.0"
|
|
data['players'] = []
|
|
data['wars'] = []
|
|
for player in self.players.values():
|
|
data['players'].append(player.toDict())
|
|
with open(path, "w", encoding="utf-8") as f:
|
|
json.dump(data, f, indent=2)
|
|
|
|
def add_player(self, name):
|
|
player = Player(name)
|
|
self.players[player.id] = player
|
|
return player
|
|
|
|
def get_player(self, id):
|
|
return self.players[id]
|
|
|
|
def update_player(self, id, name):
|
|
player = self.get_player(id)
|
|
player.set_name(name)
|
|
|
|
def delete_player(self, id):
|
|
del self.players[id]
|
|
|
|
def get_all_players(self) -> list[Player]:
|
|
return list(self.players.values())
|
|
|