CC-2514:new-airtime-import-script

- changing set-storage-dir command to storage-dir
- adding sub command 'set' and 'get'
This commit is contained in:
james 2011-07-08 17:46:07 -04:00
parent c0a68e20c2
commit 6ad3bb306e
1 changed files with 13 additions and 8 deletions

View File

@ -35,11 +35,11 @@ except Exception, e:
api_client = api_client.api_client_factory(config)
def import_copy(args):
dest = get_stor_dir()+"/organize/"
dest = helper_get_stor_dir()+"/organize/"
copy_or_move_files_to(args.path, dest, 'copy')
def import_move(args):
dest = get_stor_dir()+"/organize/"
dest = helper_get_stor_dir()+"/organize/"
copy_or_move_files_to(args.path, dest, 'move')
def watch_add(args):
@ -79,7 +79,6 @@ def watch_remove(args):
def set_stor_dir(args):
if(os.path.isdir(args.path)):
res = api_client.set_storage_dir(args.path)
print res
# sucess
if(res == '[]'):
print "Successfully set storage folder to %s" % args.path
@ -88,6 +87,8 @@ def set_stor_dir(args):
else:
print "Given path is not a directory: %s" % args.path
def get_stor_dir(args):
print helper_get_stor_dir()
#helper functions
# copy or move files
@ -116,7 +117,7 @@ def copy_or_move_files_to(paths, dest, flag):
else:
print "Cannot find file or path: %s" % path
def get_stor_dir():
def helper_get_stor_dir():
res = api_client.list_all_watched_dirs()
return res['dirs']['1']
@ -146,10 +147,14 @@ parser_add.set_defaults(func=watch_add)
parser_list.set_defaults(func=watch_list)
# for subcommand set-storage-dir
parser_set_stor_dir = subparsers.add_parser('set-storage-dir', help='operations on watch directory')
parser_set_stor_dir.add_argument('-f', '--force', action='store_true', help='bypass confirmation')
parser_set_stor_dir.add_argument('path', help='path to the directory')
parser_set_stor_dir.set_defaults(func=set_stor_dir)
parser_stor_dir = subparsers.add_parser('storage-dir', help='operations on storage directory')
storage_subparsers = parser_stor_dir.add_subparsers()
parser_set = storage_subparsers.add_parser('set', help='set a storage directory')
parser_get = storage_subparsers.add_parser('get', help='get the current storage directory')
parser_set.add_argument('-f', '--force', action='store_true', help='bypass confirmation')
parser_set.add_argument('path', help='path to the directory')
parser_set.set_defaults(func=set_stor_dir)
parser_get.set_defaults(func=get_stor_dir)
args = parser.parse_args()
#format args.path