diff options
Diffstat (limited to 'worker')
| -rwxr-xr-x[-rw-r--r--] | worker/worker1.py | 13 |
1 files changed, 6 insertions, 7 deletions
diff --git a/worker/worker1.py b/worker/worker1.py index 5c474b2..ae2f9d8 100644..100755 --- a/worker/worker1.py +++ b/worker/worker1.py @@ -10,27 +10,26 @@ ARGS = PARSER.parse_args() def do_work(ifile): ofile = os.path.join(ARGS.output_dir, os.path.basename(ifile)) in_data = [] - out_data = [] print('Loading {} ({} bytes)'.format(ifile, os.path.getsize(ifile))) - with open(ifile) as fp: + with open(ifile, 'r') as fp: for line in fp: line = line.rstrip() in_data.append(line) - with open(ofile) as fp: + with open(ofile, 'w+') as fp: for value in in_data: - fp.writeline(value+1) + fp.writelines('{}{}'.format(int(value)+1, os.linesep)) if __name__ == '__main__': + if not ARGS.INFILE: + print("No input file(s) received!") + exit(1) for infile in ARGS.INFILE: do_work(os.path.abspath(infile)) - else: - print("No input file(s) received!") - exit(1) exit(0)
\ No newline at end of file |