aboutsummaryrefslogtreecommitdiff
path: root/htcondor_utils/wrap.py
blob: 19b949672f6102fbcc1233e3eb858e469baf3e05 (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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
#!/usr/bin/env python
#
# This file is part of htcondor_utils.
#
# htcondor_utils is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# htcondor_utils is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with htcondor_utils.  If not, see <http://www.gnu.org/licenses/>.

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.abspath(args.job)]
    indata = args.indata
    output = []
    if isinstance(indata, file):
        for line in indata.readlines():
            line = line.strip()
            line = os.path.abspath(line)
            output.append(line)
    else:
        output = indata

    compiled = job + output
    if verbose:
        print("Job: {0}".format(compiled))

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


if __name__ == "__main__":
    main()