rsync_from_cluster.py

#!/usr/bin/env python3

from sys import argv,exit
import string,subprocess
from os import system
from os.path import basename,dirname,abspath,exists,expanduser
from cluster_info import *

def Help():
    print ()
    print ( argv[0]+' <cluster> <any extra rsync flags>' )
    print()
    if ( subprocess.call( ['/bin/bash','-i','-c','alias rfc']) == 1 ):
        print ("You may want to alias this command to rfc, by putting the following in your .bashrc: ")
        print (' alias rfc="rsync_from_cluster.py"')
    exit()

if len(argv)<2:
    Help()

cluster_in = argv[1]
(cluster,remotedir) = cluster_check( cluster_in )

if cluster == 'unknown':
    Help()

# handle flags like '--exclude' and '--delete' correctly
args = argv[2:]
filenames = []
extra_args = []
for m in args:
    if len( m ) > 2 and m.find( '--' ) > -1:
        extra_args.append( m )
    else:
        filenames.append( m )
if len(filenames) == 0: filenames = ['.']
dir = '.'

# strip off directory name based on local path.
clusterdir = remotedir+strip_home_dirname( abspath(dir) )

cluster_prefix = cluster+':'
if len(cluster) == 0: cluster_prefix = ''

commands = []
for filename in filenames:
    remote_filename = ' ' + cluster_prefix+clusterdir + '/' + filename
    #command = 'rsync -avzL '+ remote_filename + ' . '+' '.join(extra_args)
    command = 'rsync -avL '+ remote_filename + ' . '+' '.join(extra_args)
    print(command)
    system(command)
    commands.append( command )
print
print ('Ran the following commands: ')
for command in commands:
    print
    print(command)