Merge pull request #36 from LSSTDESC/yanza/command_update_packages

command-update

src/rail/cli/commands.py

@@ -29,8 +29,18 @@ def clone_source(outdir, git_mode, dry_run, package_file, **kwargs):
     """Install packages from source"""
     scripts.clone_source(outdir, git_mode, dry_run, package_file)
     return 0
+
+
+@cli.command()
+@options.outdir(default='..')
+@options.dry_run()
+@options.package_file()
+def update_source(outdir, dry_run, package_file, **kwargs):
+    """Update packages from source"""
+    scripts.update_source(outdir, dry_run, package_file)
+    return 0
+
 
-
 @cli.command()
 @options.outdir(default='..')
 @options.dry_run()

src/rail/cli/scripts.py

@@ -71,6 +71,27 @@ def clone_source(outdir, git_mode, dry_run, package_file):
         else:
             os.system(com_line)
 
+
+def update_source(outdir, dry_run, package_file):
+
+    with open(package_file) as pfile:
+        package_dict = yaml.safe_load(pfile)
+
+    currentpath = os.path.abspath('.')    
+    for key, val in package_dict.items():
+        abspath = os.path.abspath(f"{outdir}/{key}")
+
+        if os.path.exists(f"{outdir}/{key}") is not True:
+            print(f"Package {outdir}/{key} does not exist!")
+            continue            
+
+        com_line = f"cd {abspath} && git pull && cd {currentpath}"
+
+        if dry_run:
+            print(com_line)
+        else:
+            os.system(com_line)
+
 
 def install(outdir, from_source, dry_run, package_file):
 

tests/cli/test_scripts.py

@@ -13,6 +13,10 @@ def test_clone_source():
     scripts.clone_source('..', GitMode.https, True, 'rail_packages.yml')
     scripts.clone_source('..', GitMode.cli, True, 'rail_packages.yml')
 
+
+def test_update_source():
+    scripts.update_source('..', True, 'rail_packages.yml')
+
 
 def test_install():
     scripts.install('..', False, True, 'rail_packages.yml')