"""CLI: Take a set of coins from a coin box.""" from argparse import ArgumentParser from sys import exit, stderr from piggybank import print_program_version from piggybank.piggybank import PiggyBank from piggybank.cli import EPILOGUE from piggybank.currencies import CURRENCIES, DEFAULT_CURRENCY, \ BaseCurrencyError, print_supported_currencies from piggybank.transaction import TYPE_OUTCOME from piggybank.util import add_common_arguments_to_parser, \ complement_array_of_coins __all__ = ["main"] def main(): """An entry point for a take command.""" parser = ArgumentParser(prog="piggybank-take", description="Take a set of coins from a coin box.", epilog=EPILOGUE) parser.add_argument("file", type=str, help="a coin box file name. Missing .cb extension will" "be added") parser.add_argument("coins", type=int, nargs="+", metavar="COIN", help="add a set of coins to a coin box. A new file" "will be created if it doesn't exist") add_common_arguments_to_parser(parser) args = parser.parse_args() if args.version: print_program_version() exit() if args.list_currencies: print_supported_currencies() exit() try: piggybank = PiggyBank.from_file(args.file) coins = complement_array_of_coins(args.coins, piggybank.currency, args.reverse) piggybank.transact(coins, TYPE_OUTCOME) piggybank.save(args.file) except BaseCurrencyError as err: print(f"{type(err).__name__}:", err, file=stderr) except ValueError as err: print(f"{type(err).__name__}:", err, file=stderr) except FileNotFoundError as err: print(f"{type(err).__name__}:", f"{err} doesn't exist.", file=stderr) except Exception as err: print(f"Something went exceptionally wrong. Error:", f"{type(err).__name__}:", err, file=stderr)