From 2a4e37a5eb61ef17c228d46fa9d9075c0788f8ec Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Fri, 21 Jun 2024 16:26:33 +0200 Subject: [PATCH] main/main --- pidd/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pidd/__init__.py b/pidd/__init__.py index 9df594e..db868ff 100644 --- a/pidd/__init__.py +++ b/pidd/__init__.py @@ -115,12 +115,12 @@ def install_commands(path: str) -> None: (user / f"{COMMANDS_PREFIX}{command}").symlink_to(path) -def main() -> None: +def main(main: str) -> None: path, *arguments = sys.argv name = Path(path).name if name == "__main__.py": pidd.set_ssh(*arguments) - pidd.install_commands(__file__) + pidd.install_commands(main) else: function = getattr(pidd, name.replace("-", "_")) function(*arguments)