From 1d0e51add92e6264142e8a1d6eb5ad2c9a620556 Mon Sep 17 00:00:00 2001 From: Florian Maurer <f.maurer@outlook.de> Date: Tue, 17 Sep 2024 02:09:26 +0200 Subject: [PATCH] allow setting title in main again make it possible to call the cli function directly with args --- eralchemy/main.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/eralchemy/main.py b/eralchemy/main.py index af71246..e7408e3 100644 --- a/eralchemy/main.py +++ b/eralchemy/main.py @@ -44,11 +44,11 @@ except PackageNotFoundError: __version__ = "na" -def cli() -> None: +def cli(args=None) -> None: """Entry point for the application script.""" parser = get_argparser() - args = parser.parse_args() + args = parser.parse_args(args) check_args(args) if args.v: print(f"eralchemy version {__version__}.") @@ -395,7 +395,7 @@ def render_er( exclude_columns=exclude_columns, ) intermediary_to_output = get_output_mode(output, mode) - return intermediary_to_output(tables, relationships, output) + return intermediary_to_output(tables, relationships, output, title) except ImportError as e: module_name = e.message.split()[-1] print(f'Please install {module_name} using "pip install {module_name}".') @@ -406,4 +406,5 @@ def render_er( if __name__ == "__main__": + # cli("-i example/forum.er -o test.dot".split(" ")) cli() -- GitLab