From 245da78a417d0192956f8a3df137da829f9cee09 Mon Sep 17 00:00:00 2001 From: Gabriel Augendre Date: Sat, 22 Feb 2020 15:34:40 +0100 Subject: [PATCH] Add a ynab config edit command --- ofx_processor/main.py | 2 ++ ofx_processor/utils/ynab.py | 27 +++++++++++++++++++++++++-- 2 files changed, 27 insertions(+), 2 deletions(-) diff --git a/ofx_processor/main.py b/ofx_processor/main.py index ffb8584..1b04a85 100644 --- a/ofx_processor/main.py +++ b/ofx_processor/main.py @@ -2,6 +2,7 @@ import click import ofx_processor.bpvf_processor.main as bpvf import ofx_processor.revolut_processor.main as revolut +from ofx_processor.utils import ynab CONTEXT_SETTINGS = dict(help_option_names=["-h", "--help"]) @@ -14,6 +15,7 @@ def cli(): cli.add_command(bpvf.cli, name="bpvf") cli.add_command(revolut.cli, name="revolut") +cli.add_command(ynab.config, name="config") if __name__ == "__main__": cli() diff --git a/ofx_processor/utils/ynab.py b/ofx_processor/utils/ynab.py index 4a6f9f2..76e30ad 100644 --- a/ofx_processor/utils/ynab.py +++ b/ofx_processor/utils/ynab.py @@ -21,12 +21,31 @@ def get_default_config(): return default_config +def get_config_file_name(): + config_file = os.path.join(DEFAULT_CONFIG_DIR, DEFAULT_CONFIG_FILENAME) + return config_file + + +@click.group() +def config(): + pass + + +@click.command() +def edit_config(): + config_file = get_config_file_name() + click.edit(filename=config_file) + + +config.add_command(edit_config, "edit") + + def push_transactions(transactions, account): if not transactions: click.secho("No transaction, nothing to do.", fg="yellow") return config = configparser.ConfigParser() - config_file = os.path.join(DEFAULT_CONFIG_DIR, DEFAULT_CONFIG_FILENAME) + config_file = get_config_file_name() if not os.path.isfile(config_file): os.makedirs(DEFAULT_CONFIG_DIR, exist_ok=True) config = get_default_config() @@ -56,7 +75,11 @@ def push_transactions(transactions, account): matched_id = transaction["matched_transaction_id"] if not matched_id or matched_id not in created: created.add(transaction["id"]) - click.secho(f"{len(created)} transactions created in YNAB.", fg="green", bold=True) + + if created: + click.secho( + f"{len(created)} transactions created in YNAB.", fg="green", bold=True + ) duplicates = data["duplicate_import_ids"] if duplicates: