mirror of
https://gitlab.ub.uni-bielefeld.de/sfb1288inf/ocr.git
synced 2025-07-01 18:00:34 +00:00
Update for unprivileged usage.
This commit is contained in:
23
wrapper/ocr
23
wrapper/ocr
@ -17,15 +17,16 @@ parser.add_argument(
|
||||
dest='output_dir',
|
||||
required=False
|
||||
)
|
||||
args, ocr_args = parser.parse_known_args()
|
||||
dirs, args = parser.parse_known_args()
|
||||
|
||||
if args.input_dir == None or args.output_dir == None:
|
||||
if args.input_dir != None:
|
||||
ocr_args.append('-i')
|
||||
ocr_args.append('/files_for_ocr')
|
||||
if args.output_dir != None:
|
||||
ocr_args.append('-o')
|
||||
ocr_args.append('/files_from_ocr')
|
||||
subprocess.run(['docker', 'run', '--rm', '-it', container_image] + ocr_args)
|
||||
else:
|
||||
subprocess.run(['docker', 'run', '--rm', '-it', '-v', args.input_dir + ':/files_for_ocr', '-v', args.output_dir + ':/files_from_ocr', container_image, '-i', '/files_for_ocr', '-o', '/files_from_ocr'] + ocr_args)
|
||||
cmd = ['docker', 'run', '--rm', '-it']
|
||||
if dirs.input_dir is not None:
|
||||
cmd += ['-v', dirs.input_dir + ':/input']
|
||||
args += ['-i', '/input']
|
||||
if dirs.output_dir is not None:
|
||||
cmd += ['-v', dirs.output_dir + ':/output']
|
||||
args += ['-o', '/output']
|
||||
cmd.append(container_image)
|
||||
cmd += args
|
||||
|
||||
subprocess.run(cmd)
|
||||
|
Reference in New Issue
Block a user