fftcgtool/main.py

#!/usr/bin/env python3
import argparse
import logging
import os

from fftcg.book import Book
from fftcg.opus import Opus

# constants
GRID = 7, 10  # default in TTsim: 7 rows, 10 columns
RESOLUTION = 429, 600  # default in TTsim: 480x670 pixels per card


def main() -> None:
    # set up CLI
    parser = argparse.ArgumentParser(
        description='Imports FFTCG cards for TT-Sim.')

    parser.add_argument(
        'opus_id',
        default="1",
        metavar="Opus_ID",
        nargs="?",
        help='the Opus to import')

    parser.add_argument(
        '-n', '--num_threads',
        type=int,
        default=20,
        metavar="COUNT",
        help='maximum number of concurrent requests')

    args = parser.parse_args()

    # set up logging
    logging.basicConfig(level=logging.INFO, format='%(levelname)s: %(threadName)s %(message)s')

    # output directory
    if not os.path.exists("out"):
        os.mkdir("out")
    os.chdir("out")

    # main program
    opus = Opus(args.opus_id)
    book = Book(opus, GRID, RESOLUTION, "eg", args.num_threads)
    book.save(f"{opus.name}_{{}}.jpg")

    # bye
    logging.info("Done. Put the generated JSON files in your 'Saved Objects' Folder.")
    logging.info("Thanks for using fftcgtool!")


if __name__ == '__main__':
    main()