mirror of
https://github.com/ldericher/fftcgtool
synced 2025-01-15 15:02:59 +00:00
78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
import logging
|
|
|
|
import yaml
|
|
from PIL import Image
|
|
|
|
from .cards import Cards
|
|
from .grid import Grid
|
|
from .imageloader import ImageLoader
|
|
|
|
|
|
class Book:
|
|
def __init__(self, cards: Cards, grid: tuple[int, int], resolution: tuple[int, int], language: str,
|
|
num_threads: int):
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# transform grid into Grid
|
|
grid = Grid(grid)
|
|
|
|
# sort cards by element, then alphabetically
|
|
cards.sort(key=lambda x: x.name)
|
|
cards.sort(key=lambda x: "Multi" if len(x.elements) > 1 else x.elements[0])
|
|
|
|
# all card face URLs
|
|
urls = [f"https://fftcg.cdn.sewest.net/images/cards/full/{card.code}_{language}.jpg" for card in cards]
|
|
# card back URL (image by Aurik)
|
|
urls.append(
|
|
"http://cloud-3.steamusercontent.com/ugc/948455238665576576/85063172B8C340602E8D6C783A457122F53F7843/"
|
|
)
|
|
|
|
# multi-threaded download
|
|
images = ImageLoader.load(urls, resolution, language, num_threads)
|
|
# card back Image
|
|
back_image = images.pop(-1)
|
|
|
|
self.__pages = []
|
|
for page_images, page_cards in zip(grid.chunks(images), grid.chunks(cards)):
|
|
# create book page Image
|
|
page_image = Image.new("RGB", grid * resolution)
|
|
logger.info(f"New image: {page_image.size[0]}x{page_image.size[1]}")
|
|
|
|
# paste card faces onto page
|
|
for i, image in enumerate(page_images):
|
|
grid.paste(page_image, i, image)
|
|
|
|
# paste card back in last position
|
|
grid.paste(page_image, grid.capacity, back_image)
|
|
|
|
# save page
|
|
self.__pages.append({
|
|
"image": page_image,
|
|
"cards": page_cards,
|
|
})
|
|
|
|
def __getitem__(self, index: int) -> Image.Image:
|
|
return self.__pages[index]["image"]
|
|
|
|
def save(self, filename: str) -> None:
|
|
pages: dict[str, dict[str, any]]
|
|
|
|
# load pages.yml file
|
|
try:
|
|
with open("pages.yml", "r") as file:
|
|
pages = yaml.load(file)
|
|
except FileNotFoundError:
|
|
pages = {}
|
|
|
|
# save book
|
|
for i, page in enumerate(self.__pages):
|
|
fn = f"{filename}_{i}.jpg"
|
|
# save page image
|
|
page["image"].save(fn)
|
|
# add contents of image
|
|
pages[fn] = {}
|
|
pages[fn]["cards"] = page["cards"] # [card.code for card in page["cards"]]
|
|
|
|
# update pages.yml file
|
|
with open("pages.yml", "w") as file:
|
|
yaml.dump(pages, file)
|