fftcgtool/main.py

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

import fftcg


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

    parser.add_argument(
        "opus_id",
        default="promo",
        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: %(processName)s %(message)s")

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

    # main program
    opus = fftcg.Opus(args.opus_id)
    book = fftcg.Book(opus, "eg", args.num_threads)
    book.save()

    # load the current carddb
    carddb = fftcg.CardDB.get()
    carddb.load()

    # create elemental decks for opus
    for deck in opus.elemental_decks:
        deck.save()

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


if __name__ == "__main__":
    main()