def _extraction(self): iblrig_pipeline.extract(self.server_folder) # Check for deletion of extract_me.flag eflags = list(self.server_folder.rglob('extract_me.flag')) self.assertTrue(eflags == []) # Check for creation of register_me.flag rflags = list(self.server_folder.rglob('register_me.flag')) self.assertTrue(rflags != []) # Check for flag in all sessions self.assertTrue(len(rflags) == len(self.sessions))
def rerun_extract(ses_path, drange, dry=True): files_error, files_error_date = _order_glob_by_session_date( ses_path.rglob('extract_me.error')) for file_error, date in zip(files_error, files_error_date): if not (date >= drange[0] and (date <= drange[1])): continue print(file_error) if dry: continue file_error.unlink() flags.create_extract_flags(file_error.parent, force=True) pipes.extract(file_error.parent)
parser = argparse.ArgumentParser(description="Description of your program") parser.add_argument("action", help="Action: " + ",".join(ALLOWED_ACTIONS)) parser.add_argument("folder", help="A Folder containing a session") parser.add_argument("--dry", help="Dry Run", required=False, default=False, type=str) parser.add_argument( "--count", help="Max number of sessions to run this on", required=False, default=False, type=int, ) args = parser.parse_args( ) # returns data from the options specified (echo) if args.dry and args.dry.lower() == "false": args.dry = False assert Path(args.folder).exists() if args.action == "extract": extract(args.folder, dry=args.dry) elif args.action == "register": register(args.folder, dry=args.dry) elif args.action == "create": create(args.folder, dry=args.dry) elif args.action == "compress_video": compress_video(args.folder, dry=args.dry, max_sessions=args.count) else: logger.error("Allowed actions are: " + ", ".join(ALLOWED_ACTIONS))
default='1999-12-12', type=str) parser.add_argument('--last', help='yyyy-mm-dd date', required=False, default='2050-12-12', type=str) args = parser.parse_args( ) # returns data from the options specified (echo) if args.dry and args.dry.lower() == 'false': args.dry = False assert (Path(args.folder).exists()) ses_path = Path(args.folder) # compute the date range including both bounds files_error, files_error_date = _order_glob_by_session_date( ses_path.rglob('extract_me.error')) drange = [parse(args.first), parse(args.last)] for file_error, date in zip(files_error, files_error_date): if not (date >= drange[0] and (date <= drange[1])): continue print(file_error) if args.dry: continue file_error.unlink() flags.create_extract_flags(file_error.parent, force=True) extract(file_error.parent)