mirror of
https://github.com/ldericher/fftcgtool
synced 2025-01-15 15:02:59 +00:00
93 lines
2.7 KiB
Python
93 lines
2.7 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
import pickle
|
|
import zipfile
|
|
|
|
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]
|
|
|
|
__DB_FILE_NAME = "cards.pickle"
|
|
__MAPPING_FILE_NAME = "face_to_url.json"
|
|
|
|
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 get_face_url(self, face: str) -> str:
|
|
if face in self.__face_to_url:
|
|
return self.__face_to_url[face]
|
|
else:
|
|
return face
|
|
|
|
def __pickle(self) -> None:
|
|
with zipfile.ZipFile(CARDDB_FILE_NAME, "w", compression=zipfile.ZIP_LZMA) as zip_file:
|
|
# cards db
|
|
with zip_file.open(CardDB.__DB_FILE_NAME, "w") as file:
|
|
pickle.dump(self.__cards, file)
|
|
|
|
# face_to_url mapping
|
|
with zip_file.open(CardDB.__MAPPING_FILE_NAME, "w") as file:
|
|
file.write(json.dumps(self.__face_to_url, indent=2).encode("utf-8"))
|
|
|
|
def __unpickle(self) -> None:
|
|
# unpickle db file
|
|
self.__cards.clear()
|
|
self.__face_to_url.clear()
|
|
try:
|
|
with zipfile.ZipFile(CARDDB_FILE_NAME, "r") as zip_file:
|
|
# cards db
|
|
with zip_file.open(CardDB.__DB_FILE_NAME, "r") as file:
|
|
self.__cards |= pickle.load(file)
|
|
|
|
# face_to_url mapping
|
|
with zip_file.open(CardDB.__MAPPING_FILE_NAME, "r") as file:
|
|
self.__face_to_url |= json.load(file)
|
|
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
def load(self) -> None:
|
|
self.__unpickle()
|
|
|
|
def update(self, cards: Cards) -> None:
|
|
for card in cards:
|
|
self.__cards[card.code] = card
|
|
|
|
self.__pickle()
|
|
|
|
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()
|