beanhub-cli 1.2.0__tar.gz → 1.2.2__tar.gz

{beanhub_cli-1.2.0 → beanhub_cli-1.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: beanhub-cli
-Version: 1.2.0
+Version: 1.2.2
 Summary: Command line tools for BeanHub
 License: MIT
 Author: Fang-Pen Lin
@@ -105,8 +105,12 @@ You can run [beancount-black](https://github.com/LaunchPlatform/beancount-black)
 bh format main.bean
 ```
 
-Currently, the formatter works mostly the same as beancount-black, except it doesn't create a backup file by default.
-We are working on adding new features, such as making it possible to follow Beancount include statements to format all Beancount files.
+Without the file arguments, the format command will try to format `main.bean` in the current folder, and it will recursively follow all the `include` statements to format all the bean files.
+Like this:
+
+```bash
+bh format
+```
 
 ### More features to come
 

{beanhub_cli-1.2.0 → beanhub_cli-1.2.2}/README.md

@@ -76,8 +76,12 @@ You can run [beancount-black](https://github.com/LaunchPlatform/beancount-black)
 bh format main.bean
 ```
 
-Currently, the formatter works mostly the same as beancount-black, except it doesn't create a backup file by default.
-We are working on adding new features, such as making it possible to follow Beancount include statements to format all Beancount files.
+Without the file arguments, the format command will try to format `main.bean` in the current folder, and it will recursively follow all the `include` statements to format all the bean files.
+Like this:
+
+```bash
+bh format
+```
 
 ### More features to come
 

{beanhub_cli-1.2.0 → beanhub_cli-1.2.2}/beanhub_cli/format.py

@@ -7,6 +7,9 @@ import typing
 import click
 from beancount_black.formatter import Formatter
 from beancount_parser.parser import make_parser
+from beancount_parser.parser import traverse
+from lark import Lark
+from lark import Tree
 
 from .cli import cli
 from .environment import Environment
@@ -32,6 +35,16 @@ def create_backup(src: pathlib.Path, suffix: str) -> pathlib.Path:
         return backup_path
 
 
+def file_tree_iterator(
+    parser: Lark, filenames: list[pathlib.Path]
+) -> typing.Generator[tuple[pathlib.Path, Tree], None, None]:
+    for filename in filenames:
+        with open(filename, "rt") as input_file:
+            input_content = input_file.read()
+            tree = parser.parse(input_content)
+            yield filename, tree
+
+
 @cli.command(name="format", help="Format Beancount files with beancount-black")
 @click.argument("filename", type=click.Path(exists=False, dir_okay=False), nargs=-1)
 @click.option(
@@ -53,33 +66,43 @@ def main(
     backup: bool,
 ):
     # TODO: support follow include statements
-
     parser = make_parser()
-    formatter = Formatter()
     if stdin_mode:
         env.logger.info("Processing in stdin mode")
         input_content = sys.stdin.read()
         tree = parser.parse(input_content)
+        formatter = Formatter()
         formatter.format(tree, sys.stdout)
     else:
-        for name in filename:
-            env.logger.info("Processing file %s", name)
-            with open(name, "rt") as input_file:
-                input_content = input_file.read()
-                tree = parser.parse(input_content)
+        if filename:
+            iterator = file_tree_iterator(
+                parser=parser,
+                filenames=map(lambda item: pathlib.Path(str(item)), filename),
+            )
+        else:
+            env.logger.info("No files provided, traverse starting from main.bean")
+            iterator = traverse(
+                parser=parser,
+                bean_file=pathlib.Path("main.bean"),
+                root_dir=pathlib.Path.cwd(),
+            )
+        for filepath, tree in iterator:
+            env.logger.info("Processing file %s", filepath)
             with tempfile.NamedTemporaryFile(mode="wt+", suffix=".bean") as output_file:
+                formatter = Formatter()
                 formatter.format(tree, output_file)
                 output_file.seek(0)
                 output_content = output_file.read()
+                input_content = filepath.read_text()
                 if input_content == output_content:
-                    env.logger.info("File %s is not changed, skip", name)
+                    env.logger.info("File %s is not changed, skip", filepath)
                     continue
                 if backup:
-                    backup_path = create_backup(
-                        src=pathlib.Path(str(name)), suffix=backup_suffix
+                    backup_path = create_backup(src=filepath, suffix=backup_suffix)
+                    env.logger.info(
+                        "File %s changed, backup to %s", filepath, backup_path
                     )
-                    env.logger.info("File %s changed, backup to %s", name, backup_path)
                 output_file.seek(0)
-                with open(name, "wt") as input_file:
+                with open(filepath, "wt") as input_file:
                     shutil.copyfileobj(output_file, input_file)
     env.logger.info("done")

{beanhub_cli-1.2.0 → beanhub_cli-1.2.2}/beanhub_cli/import_cli.py

@@ -126,7 +126,6 @@ def main(env: Environment, config: str, workdir: str, beanfile: str):
         "Found %s existing imported transactions in Beancount books", len(existing_txns)
     )
 
-    formatter = Formatter()
     change_sets = compute_changes(
         generated_txns=generated_txns,
         imported_txns=existing_txns,
@@ -157,6 +156,7 @@ def main(env: Environment, config: str, workdir: str, beanfile: str):
             new_tree = apply_change_set(tree=tree, change_set=change_set)
 
         with target_file.open("wt") as fo:
+            formatter = Formatter()
             formatter.format(new_tree, fo)
 
     table = Table(

{beanhub_cli-1.2.0 → beanhub_cli-1.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "beanhub-cli"
-version = "1.2.0"
+version = "1.2.2"
 description = "Command line tools for BeanHub"
 authors = ["Fang-Pen Lin <fangpen@launchplatform.com>"]
 license = "MIT"