Support for hsi chgrp.

Author: zshaheen

tests/test.py

@@ -66,7 +66,6 @@ def exit():
 # TODO: Change the hpss directory to a dir that's accessable to everyone
 HPSS_PATH='/home/z/zshaheen/zstash_test'
 
-
 # Create files and directories
 print('Creating files.')
 if not os.path.exists('zstash_test'):
@@ -98,6 +97,20 @@ def exit():
 output, err = run_cmd(cmd)
 str_in(output+err, 'Transferring file to HPSS')
 
+print('Testing chgrp')
+GROUP = 'acme'
+print('First, make sure that the files are not already in the {} group'.format(GROUP))
+cmd = 'hsi ls -l {}'.format(HPSS_PATH)
+output, err = run_cmd(cmd)
+str_not_in(output+err, GROUP)
+print('Running zstash chgrp')
+cmd = 'zstash chgrp -R {} {}'.format(GROUP, HPSS_PATH)
+output, err = run_cmd(cmd)
+print('Now check that the files are in the {} group'.format(GROUP))
+cmd = 'hsi ls -l {}'.format(HPSS_PATH)
+output, err = run_cmd(cmd)
+str_in(output+err, 'acme')
+
 print('Running update on the newly created directory, nothing should happen')
 os.chdir('zstash_test')
 cmd = 'zstash update --hpss={}'.format(HPSS_PATH)
@@ -178,4 +191,3 @@ def exit():
 print('*'*40)
 print('All of the tests passed! :)')
 print('*'*40)
-

zstash/chgrp.py

@@ -0,0 +1,18 @@
+import sys
+import argparse
+from hpss import hpss_chgrp
+
+def chgrp():
+
+    # Parser
+    parser = argparse.ArgumentParser(
+        usage='zstash chgrp [<args>] group hpss_archive',
+        description='Change the group of an HPSS repository.')
+    parser.add_argument('group', type=str, help='new group name of file(s)')
+    parser.add_argument('hpss', type=str, help='path to HPSS storage')
+    parser.add_argument('-R', action='store_const', const=True, help='recurse through subdirectories')
+
+    args = parser.parse_args(sys.argv[2:])
+    recurse = True if args.R else False
+    hpss_chgrp(args.hpss, args.group, recurse)
+

zstash/hpss.py

@@ -5,8 +5,10 @@
 from subprocess import Popen, PIPE
 
 
-# Put file to hpss
 def hpss_put(hpss, file, keep=True):
+    """
+    Put a file to the HPSS archive.
+    """
 
     logging.info('Transferring file to HPSS: %s' % (file))
     path, name = os.path.split(file)
@@ -36,8 +38,10 @@ def hpss_put(hpss, file, keep=True):
         os.remove(file)
 
 
-# Get file from hpss
 def hpss_get(hpss, file):
+    """
+    Get ia file from the HPSS archive.
+    """
 
     logging.info('Transferring from HPSS: %s' % (file))
     path, name = os.path.split(file)
@@ -63,3 +67,22 @@ def hpss_get(hpss, file):
     # Back to original working directory
     if path != '':
         os.chdir(cwd)
+
+def hpss_chgrp(hpss, group, recurse=False):
+    """
+    Change the group of the HPSS archive.
+    """
+    if recurse:
+        cmd = 'hsi chgrp -R {} {}'.format(group, hpss)
+    else:
+        cmd = 'hsi chgrp {} {}'.format(group, hpss)
+
+    p1 = Popen(shlex.split(cmd), stdout=PIPE, stderr=PIPE)
+    (stdout, stderr) = p1.communicate()
+    status = p1.returncode
+    if status != 0:
+        logging.error('Changing group of HPSS archive {} to {}'.format())
+        logging.debug('stdout:\n%s', stdout)
+        logging.debug('stderr:\n%s', stderr)
+        raise Exception
+

zstash/main.py

@@ -9,6 +9,7 @@
 from create import create
 from update import update
 from extract import extract
+from chgrp import chgrp
 
 
 # -----------------------------------------------------------------------------
@@ -39,6 +40,8 @@ def main():
         update()
     elif args.command == 'extract':
         extract()
+    elif args.command == 'chgrp':
+        chgrp()
     else:
         print 'Unrecognized command'
         parser.print_help()