mirror of
https://github.com/ldericher/fftcgtool
synced 2025-01-15 15:02:59 +00:00
79 lines
2.1 KiB
Python
79 lines
2.1 KiB
Python
from __future__ import annotations
|
|
|
|
import bz2
|
|
import pickle
|
|
|
|
from .card import Card
|
|
from .cards import Cards
|
|
from .code import Code
|
|
from .language import API_LANGS
|
|
from .utils import CARDDB_FILE_NAME
|
|
|
|
|
|
class CardDB:
|
|
__instance: CardDB = None
|
|
__cards: dict[Code, Card]
|
|
__face_to_url: dict[str, str]
|
|
|
|
def __new__(cls) -> CardDB:
|
|
if CardDB.__instance is None:
|
|
CardDB.__instance = object.__new__(cls)
|
|
CardDB.__instance.__cards = {}
|
|
CardDB.__instance.__face_to_url = {}
|
|
|
|
return CardDB.__instance
|
|
|
|
def __contains__(self, item: Code) -> bool:
|
|
return item in self.__cards
|
|
|
|
def __getitem__(self, code: Code) -> Card:
|
|
return self.__cards[code]
|
|
|
|
def __pickle(self):
|
|
# pickle db file
|
|
with bz2.BZ2File(CARDDB_FILE_NAME, "w") as file:
|
|
pickle.dump(self.__cards, file)
|
|
pickle.dump(self.__face_to_url, file)
|
|
|
|
def update(self, cards: Cards):
|
|
for card in cards:
|
|
self.__cards[card.code] = card
|
|
|
|
self.__pickle()
|
|
|
|
def get_face_url(self, face: str) -> str:
|
|
if face in self.__face_to_url:
|
|
return self.__face_to_url[face]
|
|
else:
|
|
return face
|
|
|
|
def upload_prompt(self) -> None:
|
|
faces = list(set([
|
|
card[lang].face
|
|
for card in self.__cards.values()
|
|
for lang in API_LANGS
|
|
if card[lang].face
|
|
]))
|
|
faces.sort()
|
|
|
|
for face in faces:
|
|
if face not in self.__face_to_url:
|
|
face_url = input(f"Upload '{face}' and paste URL: ")
|
|
if face_url:
|
|
self.__face_to_url[face] = face_url
|
|
|
|
self.__pickle()
|
|
|
|
def __unpickle(self):
|
|
# unpickle db file
|
|
self.__cards.clear()
|
|
self.__face_to_url.clear()
|
|
try:
|
|
with bz2.BZ2File(CARDDB_FILE_NAME, "r") as file:
|
|
self.__cards |= pickle.load(file)
|
|
self.__face_to_url |= pickle.load(file)
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
def load(self) -> None:
|
|
self.__unpickle()
|