Merge pull request #8 from helxplatform/develop

fix cli
helxplatform · May 16, 2024 · 33cce37 · 33cce37
2 parents 6b366a0 + 6819aa1
commit 33cce37
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/src/avalon/__init__.py b/src/avalon/__init__.py
@@ -1 +1 @@
-__version__="1.0.0"
+__version__="1.0.2"
diff --git a/src/avalon/main.py b/src/avalon/main.py
@@ -37,8 +37,11 @@ def main(args):
         get_files(
             local_path=args.local_path,
             remote_path=args.remote_path,
-            branch=env_args['lakefs_branch'],
-            lake_fs_client=client
+            branch=args.branch,
+            lake_fs_client=client,
+            changes_only=args.changed_files_only,
+            repo=args.repository,
+
         )
 
 
@@ -49,9 +52,11 @@ def cli():
     sub_parsers = parser.add_subparsers(help="Sub commands", dest="sub_command")
 
     parser_get_file = sub_parsers.add_parser("get", help="Gets file from Lakefs repo", )
-    parser_get_file.add_argument("-r", "--remote-path", help="Remote file / dir path")
+    parser_get_file.add_argument("-p", "--remote-path", help="Remote file / dir path")
     parser_get_file.add_argument("-l", "--local-path", help="Local output dir")
     parser_get_file.add_argument("-c", "--changed-files-only", help="To get changed files only", default=False)
+    parser_get_file.add_argument("-r", "--repository", help="repository to get data from")
+    parser_get_file.add_argument("-b", "--branch", help="repository branch")
 
     parser_put_file = sub_parsers.add_parser("put", help="Puts file to Lakefs repo")
     parser_put_file.add_argument("-l", "--local-path", help="Local dir to push")