def datasets(args): if args.list: utils.pprint_json(utils.get_datasets()) elif args.show: ds = utils.get_datasets().get(args.show, None) if not ds: raise ValueError("Dataset not found.") utils.pprint_json(ds) elif args.search: search_datasets_for_string(args.search) elif args.download: ds_name, volume = args.download try: volume = float(volume) except ValueError: raise ValueError( "Volume has to be a valid number, not `{}`".format(volume)) if not utils.get_datasets().get(ds_name, False): raise ValueError("Dataset `{}` not found.".format(ds_name)) elif args.run_list: # is the given run list a subset of the full run list? user_run_list = [int(r.strip()) for r in args.run_list.split(",")] full_run_list = [ int(r.strip()) for r in utils.get_datasets().get(ds_name) ['run_list'].split(",") ] if not set(user_run_list).issubset(full_run_list): raise ValueError( "Run {} list is not a subset of this periods run list ({})!" .format(user_run_list, full_run_list)) utils.download_dataset(ds_name, volume, args.run_list)
def datasets(args): if args.list: _pprint_json(utils.get_datasets()) elif args.show: ds = utils.get_datasets().get(args.show, None) if not ds: raise ValueError("Dataset not found.") _pprint_json(ds) elif args.search: search_datasets_for_string(args.search) elif args.download: ds_name, volume = args.download try: volume = float(volume) except ValueError: raise ValueError("Volume has to be a valid number, not `{}`".format(volume)) if not utils.get_datasets().get(ds_name, False): raise ValueError("Dataset `{}` not found.".format(ds_name)) elif args.run_list: # is the given run list a subset of the full run list? user_run_list = [int(r.strip()) for r in args.run_list.split(",")] full_run_list = [int(r.strip()) for r in utils.get_datasets().get(ds_name)['run_list'].split(",")] if not set(user_run_list).issubset(full_run_list): raise ValueError("Run {} list is not a subset of this periods run list ({})!" .format(user_run_list, full_run_list)) utils.download_dataset(ds_name, volume, args.run_list)
def test_downloaded_something(self): utils.check_alien_token() utils.download_dataset("LHC10h_AOD160", 0.001)
def test_downloaded_something(self): utils.check_alien_token() utils.download_dataset("LHC10h_AOD160", 0.001)