diff --git a/wobbl_tools/text/tree_view.py b/wobbl_tools/text/tree_view.py index efb05fa..8609bff 100644 --- a/wobbl_tools/text/tree_view.py +++ b/wobbl_tools/text/tree_view.py @@ -66,13 +66,13 @@ def tree_view( if __name__ == "__main__": - if len(sys.argv) == 1: # use first command line argument when set, else use the module path + if len(sys.argv) == 1: # use module path when no parameters were given path = module_path elif sys.argv[1] == "-v": # if -v parameter is set, the program will view the current directory path = os.getcwd() - else: + else: # else view the directory specified by parameter path = sys.argv[1] print(tree_view(path, exclude_names=["__pycache__"]))