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)