fftcgtool/fftcg/book.py

import logging
import os

from PIL import Image

from .cards import Cards
from .imageloader import ImageLoader
from .utils import GRID, RESOLUTION, CARD_BACK_URL, IMAGES_DIR_NAME


class Book:
    def __init__(self, cards: Cards, language: str, num_threads: int):
        logger = logging.getLogger(__name__)

        # 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
        urls.append(CARD_BACK_URL)

        # multi-threaded download
        images = ImageLoader.load(urls, num_threads)
        # card back Image
        back_image = images.pop(-1)

        self.__pages = []

        for page_num, (page_images, page_cards) in enumerate(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)

            # set card indices
            for i, card in enumerate(page_cards):
                card.index = i

            # save page
            self.__pages.append({
                "file_name": f"{cards.file_name}_{page_num}.jpg",
                "image": page_image,
                "cards": page_cards,
            })

    def save(self) -> None:
        if not os.path.exists(IMAGES_DIR_NAME):
            os.mkdir(IMAGES_DIR_NAME)

        # save images
        for page in self.__pages:
            page["file_name"] = os.path.join(IMAGES_DIR_NAME, page["file_name"])
            page["image"].save(page["file_name"])

        # ask for upload
        for page in self.__pages:
            face_url = input(f"Upload '{page['file_name']}' and paste URL: ")

            if not face_url:
                face_url = f"file://{os.path.abspath(page['file_name'])}"

            for card in page["cards"]:
                card.face_url = face_url