Fix file mode and remove (dumb) else statement.

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