65 lines
1.7 KiB
Python
65 lines
1.7 KiB
Python
import json
|
|
import os
|
|
from dataclasses import dataclass
|
|
from datetime import datetime
|
|
from logging import getLogger
|
|
from typing import Dict
|
|
|
|
from rare.utils.paths import data_dir
|
|
|
|
logger = getLogger("GameMeta")
|
|
|
|
|
|
@dataclass
|
|
class GameMeta:
|
|
app_name: str
|
|
last_played: datetime = None
|
|
|
|
@classmethod
|
|
def from_json(cls, data):
|
|
return cls(
|
|
app_name=data.get("app_name", ""),
|
|
last_played=datetime.strptime(data.get("last_played", "None"), '%Y-%m-%dT%H:%M:%S.%f')
|
|
)
|
|
|
|
def __dict__(self):
|
|
return dict(
|
|
app_name=self.app_name,
|
|
last_played=self.last_played.strftime("%Y-%m-%dT%H:%M:%S.%f")
|
|
)
|
|
|
|
|
|
class RareGameMeta:
|
|
_meta: Dict[str, GameMeta] = {}
|
|
|
|
def __init__(self):
|
|
meta_data = {}
|
|
if os.path.exists(p := os.path.join(data_dir, "game_meta.json")):
|
|
try:
|
|
meta_data = json.load(open(p))
|
|
except json.JSONDecodeError:
|
|
logger.warning("Game meta json file corrupt")
|
|
else:
|
|
with open(p, "w") as file:
|
|
file.write("{}")
|
|
|
|
for app_name, data in meta_data.items():
|
|
self._meta[app_name] = GameMeta.from_json(data)
|
|
|
|
def get_games(self):
|
|
return list(self._meta.values())
|
|
|
|
def get_game(self, app_name):
|
|
return self._meta.get(app_name, GameMeta(app_name))
|
|
|
|
def set_game(self, app_name: str, game: GameMeta):
|
|
self._meta[app_name] = game
|
|
self.save_file()
|
|
|
|
def save_file(self):
|
|
json.dump(
|
|
{app_name: data.__dict__() for app_name, data in self._meta.items()},
|
|
open(os.path.join(data_dir, "game_meta.json"), "w"),
|
|
indent=4
|
|
)
|