Use poetry for dependency management

- Introduce Google fire
- Code cleanup with ruff rules
- stop using `globals()`

Author: Your Name

beanborg/arg_parser.py

@@ -1,68 +1,110 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 
-__copyright__ = "Copyright (C) 2023  Luciano Fiandesio"
+__copyright__ = "Copyright (C) 2024  Luciano Fiandesio"
 __license__ = "GNU GPLv2"
 
 import csv
 import os
 import shutil
 import sys
 from datetime import datetime
+from typing import Optional
 
+import fire
+from fire.decorators import SetParseFns
 from rich import print as rprint
 
-from beanborg.arg_parser import eval_args
 from beanborg.config import init_config
 
 
-def main():
-
-    args = eval_args("Archives imported CVS file")
-    config = init_config(args.file, args.debug)
-
-    target_csv = os.path.join(config.csv.target, config.csv.ref + ".csv")
-
-    if not os.path.isfile(target_csv):
-        rprint(f"[red]file: {target_csv}  does not exist![red]")
-        sys.exit(-1)
-
-    if not os.path.isdir(config.csv.archive):
-        os.mkdir(config.csv.archive)
+class BeanborgArchiver:
+    """Archive imported CSV files with date range in filename.
+
+    Synopsis:
+        bb_archive --config_file=CONFIG_FILE [--debug=true]
+
+    Description:
+        Archives processed CSV files by moving them to an archive directory.
+        The archived filename includes the date range of transactions contained
+        in the file.
+
+    Arguments:
+        config_file: Path to the YAML configuration file
+        debug: Enable debug output (default: False)
+
+    Examples:
+        bb_archive --config_file=config.yaml
+        bb_archive --config_file=/path/to/config.yaml --debug=true
+    """
+
+    @SetParseFns(f=str, config_file=str, debug=bool)
+    def __call__(
+        self,
+        f: Optional[str] = None,
+        config_file: Optional[str] = None,
+        debug: bool = False,
+    ):
+        """Archive the processed CSV file.
+
+        Args:
+            f: Path to the configuration file (shorthand for --config-file)
+            config_file: Path to the configuration file
+            debug: Enable debug mode
+
+        Returns:
+            int: 0 for success, 1 for failure
+        """
+        final_config_file = f or config_file
+        if not final_config_file:
+            raise ValueError(
+                "Configuration file must be specified using either -f or --config-file"
+            )
+        config = init_config(final_config_file, debug)
+
+        target_csv = os.path.join(config.csv.target, config.csv.ref + ".csv")
+
+        if not os.path.isfile(target_csv):
+            rprint(f"[red]file: {target_csv}  does not exist![red]")
+            sys.exit(-1)
+
+        if not os.path.isdir(config.csv.archive):
+            os.mkdir(config.csv.archive)
+
+        dates = []
+        print("\u2713" + " detecting start and end date of transaction file...")
+        with open(target_csv) as csv_file:
+            csv_reader = csv.reader(csv_file, delimiter=config.csv.separator)
+            for _ in range(config.csv.skip):
+                next(csv_reader)  # skip the line
+
+            for row in csv_reader:
+                try:
+                    dates.append(
+                        datetime.strptime(
+                            row[config.indexes.date].strip(), config.csv.date_format
+                        )
+                    )
+                except Exception as ex:
+                    print("error: " + str(ex))
+
+        print("\u2713" + " moving file to archive...")
+        os.rename(
+            target_csv,
+            config.csv.archive
+            + "/"
+            + config.csv.ref
+            + "_"
+            + str(min(dates).date())
+            + "_"
+            + str(max(dates).date())
+            + ".csv",
+        )
+
+        print("\u2713" + " removing temp folder")
+        shutil.rmtree(config.csv.target)
 
-    dates = []
-    print("\u2713" + " detecting start and end date of transaction file...")
-    with open(target_csv) as csv_file:
-        csv_reader = csv.reader(csv_file, delimiter=config.csv.separator)
-        for _ in range(config.csv.skip):
-            next(csv_reader)  # skip the line
 
-        for row in csv_reader:
-            try:
-                dates.append(
-                    datetime.strptime(
-                        row[config.indexes.date].strip(), config.csv.date_format
-                    )
-                )
-            except Exception as ex:
-                print("error: " + str(ex))
-
-    print("\u2713" + " moving file to archive...")
-    os.rename(
-        target_csv,
-        config.csv.archive
-        + "/"
-        + config.csv.ref
-        + "_"
-        + str(min(dates).date())
-        + "_"
-        + str(max(dates).date())
-        + ".csv",
-    )
-
-    print("\u2713" + " removing temp folder")
-    shutil.rmtree(config.csv.target)
-
-
-if __name__ == "__main__":
-    main()
+def main():
+    """Main function to run the beanborg archiver."""
+    fire.Fire(BeanborgArchiver)

beanborg/bb_archive.py

@@ -4,14 +4,60 @@
 __copyright__ = "Copyright (C) 2024  Luciano Fiandesio"
 __license__ = "GNU GPLv2"
 
+from typing import Optional
+
+import fire
+from fire.decorators import SetParseFns
 
 from beanborg.importer import Importer
 
 
-def main():
-    imp = Importer()
-    imp.import_transactions()
+class BeanborgImporter:
+    """Parse bank CSV file and import into beancount.
+
+    Synopsis:
+        bb_import --config_file=CONFIG_FILE [--debug=true] [--fix_only=true]
+
+    Description:
+        Imports bank transactions from CSV files into beancount format.
+        Can also fix uncategorized transactions using the --fix-only flag.
+
+    Arguments:
+        config_file: Path to the YAML configuration file
+        debug: Enable debug output (default: False)
+        fix_only: Only fix transactions without an account (default: False)
 
+    Examples:
+        bb_import --config_file=config.yaml
+        bb_import --config_file=config.yaml --debug=true
+        bb_import --config_file=config.yaml --fix_only=true
+    """
 
-if __name__ == "__main__":
-    main()
+    @SetParseFns(f=str, config_file=str, debug=bool, fix_only=bool)
+    def __call__(
+        self,
+        f: Optional[str] = None,
+        config_file: Optional[str] = None,
+        debug: bool = False,
+        fix_only: bool = False,
+    ):
+        """Import transactions from CSV to beancount.
+
+        Args:
+            f: Path to the configuration file (shorthand for --config-file)
+            config_file: Path to the configuration file
+            debug: Enable debug mode
+            fix_only: Only fix uncategorized transactions
+        """
+        final_config_file = f or config_file
+        if not final_config_file:
+            raise ValueError(
+                "Configuration file must be specified using either -f or --config-file"
+            )
+        imp = Importer()
+        return imp.import_transactions(final_config_file, debug, fix_only)
+
+
+def main():
+    """Main function to run the beanborg importer."""
+    fire.Fire(BeanborgImporter)