aboutsummaryrefslogtreecommitdiff
path: root/condor_wrap.py
blob: 0aa25c406d2628f16aca798d757d50def2b31853 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
#!/usr/bin/env python
import argparse
import os
import sys
import subprocess

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--verbose", action="store_true", help="Be more verbose")
    parser.add_argument("job", action="store", help="Path to executable")
    parser.add_argument("indata", nargs='?', type=argparse.FileType('r'), default=sys.stdin, help="Standard input")
    args = parser.parse_args()

    verbose = args.verbose

    job = [os.path.normpath(args.job)]
    indata = args.indata
    output = []
    if verbose:
        print("Job: {}".format(job))

    if isinstance(indata, file):
        for line in indata.readlines():
            output.append(line)
    else:
        output = indata

    compiled = job + output
    process = subprocess.Popen(compiled, stdout=sys.stdout, stderr=sys.stderr)
    process.communicate()
    process.wait()


if __name__ == "__main__":
    main()