From b1d9c49bb9f020b080fb1df674d9d04aa1d38cd3 Mon Sep 17 00:00:00 2001 From: Patrick Jentsch Date: Tue, 21 May 2019 12:27:36 +0200 Subject: [PATCH] Add wrapper and remove default arguments from Dockerfile --- Dockerfile | 1 - wrapper/nlp | 31 +++++++++++++++++++++++++++++++ 2 files changed, 31 insertions(+), 1 deletion(-) create mode 100755 wrapper/nlp diff --git a/Dockerfile b/Dockerfile index 7ba70b2..6a461fd 100644 --- a/Dockerfile +++ b/Dockerfile @@ -38,4 +38,3 @@ COPY nlp /usr/local/bin COPY spacy_nlp /usr/local/bin ENTRYPOINT ["nlp"] -CMD ["--help"] diff --git a/wrapper/nlp b/wrapper/nlp new file mode 100755 index 0000000..9f55803 --- /dev/null +++ b/wrapper/nlp @@ -0,0 +1,31 @@ +#!/usr/bin/env python3 +# coding=utf-8 + +import argparse +import subprocess + +container_image = 'gitlab.ub.uni-bielefeld.de:4567/sfb1288inf/nlp:latest' + +parser = argparse.ArgumentParser(add_help=False) +parser.add_argument( + '-i', + dest='input_dir', + required=False +) +parser.add_argument( + '-o', + dest='output_dir', + required=False +) +args, nlp_args = parser.parse_known_args() + +if args.input_dir == None or args.output_dir == None: + if args.input_dir != None: + nlp_args.append('-i') + nlp_args.append('/files_for_nlp') + if args.output_dir != None: + nlp_args.append('-o') + nlp_args.append('/files_from_nlp') + subprocess.run(['docker', 'run', '--rm', '-it', container_image] + nlp_args) +else: + subprocess.run(['docker', 'run', '--rm', '-it', '-v', args.input_dir + ':/files_for_nlp', '-v', args.output_dir + ':/files_from_nlp', container_image, '-i', '/files_for_nlp', '-o', '/files_from_nlp'] + nlp_args)