Yavook.de/fftcgtool
1
0
Fork 0
mirror of https://github.com/ldericher/fftcgtool synced 2025-01-15 15:02:59 +00:00
Code Issues Releases Wiki Activity

ffdecks get_deck_data

Browse source
This commit is contained in:
Jörn-Michael Miehe 2021-09-14 00:55:26 +02:00
parent 97e52b2d83
commit 27af422ca8
1 changed files with 66 additions and 58 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

124
fftcgtool/ffdecks.py
View file

@ -10,6 +10,24 @@ from .code import Code
from .ttsdeck import TTSDeck
def _sort_cards_by_type(data: dict[str, str | int]) -> int:
key_prios = {
"Forward": 1,
"Summon": 2,
"Monster": 3,
"Backup": 5,
}
try:
return key_prios[data["type"]]
except KeyError:
return 4
def _sort_cards_by_cost(data: dict[str, str | int]) -> int:
return data["cost"]
class FFDecks(list[TTSDeck]):
__FFDECKS_API_URL = "https://ffdecks.com/api/deck"
__RE_FFDECKS_ID = re.compile(r"((https?://)?ffdecks\.com(/+api)?/+deck/+)?([0-9]+).*", flags=re.UNICODE)
@ -28,79 +46,69 @@ class FFDecks(list[TTSDeck]):
for match in matches
)
@classmethod
def get_deck_data(cls, deck_id: str) -> Optional[dict]:
logger = logging.getLogger(__name__)
# api request
if not (res := requests.get(FFDecks.__FFDECKS_API_URL, params={"deck_id": deck_id})).ok:
logger.error(f"Invalid Deck ID '{deck_id}' for FFDecks API!")
else:
name = f"{res.json()['name']}"
logger.info(f"Importing Deck {name!r}")
# pre-extract usable data
card_data = [{
"code": card["card"]["serial_number"],
"type": card["card"]["type"],
"cost": int(card["card"]["cost"]),
"count": int(card["quantity"]),
} for card in res.json()["cards"]]
# sort cards by type, then by cost
card_data.sort(key=_sort_cards_by_cost)
card_data.sort(key=_sort_cards_by_type)
# ffdecks quirk: some full-art promos in database
replace_full_arts = {
# line format:
# full-art-id: normal id,
"PR-051": "11-083",
"PR-055": "11-062",
}
# replace with normal-art cards
for card in card_data:
try:
card["code"] = replace_full_arts[card["code"]]
except KeyError:
pass
return {
"name": name,
"description": deck_id,
"cards": card_data,
}
def __init__(self, deck_ids: Iterable):
super().__init__()
logger = logging.getLogger(__name__)
def by_type(data: dict[str, str | int]) -> int:
key_prios = {
"Forward": 1,
"Summon": 2,
"Monster": 3,
"Backup": 5,
}
try:
return key_prios[data["type"]]
except KeyError:
return 4
def by_cost(data: dict[str, str | int]) -> int:
return data["cost"]
for deck_id in self.sanitized_ids(deck_ids):
if deck_id is None:
logger.error("Malformed Deck ID for FFDecks API!")
else:
# api request
res = requests.get(FFDecks.__FFDECKS_API_URL, params={"deck_id": deck_id})
if not res.ok:
logger.error(f"Invalid Deck ID '{deck_id}' for FFDecks API!")
else:
# general metadata
name = f"{res.json()['name']}"
description = deck_id
logger.info(f"Importing Deck {name!r}")
# pre-extract the used data
deck_cards = [{
"code": card["card"]["serial_number"],
"type": card["card"]["type"],
"cost": int(card["card"]["cost"]),
"count": int(card["quantity"]),
} for card in res.json()["cards"]]
# sort cards by type, then by cost
deck_cards.sort(key=by_cost)
deck_cards.sort(key=by_type)
# ffdecks quirk: some full-art promos in database
replace_full_arts = {
# line format:
# full-art-id: normal id,
"PR-051": "11-083",
"PR-055": "11-062",
}
# replace with normal-art cards
for card in deck_cards:
try:
card["code"] = replace_full_arts[card["code"]]
except KeyError:
pass
if (data := FFDecks.get_deck_data(deck_id)) is not None:
codes = [
# create list of code objects
Code(card["code"])
# for each card
for card in deck_cards
for card in data["cards"]
# repeat to meet count
for _ in range(card["count"])
]
# create deck object
self.append(TTSDeck(codes, name, description, True))
self.append(TTSDeck(codes, data["name"], data["description"], True))