Added argparser, filename can be provided throug command line

change_image_path.py

@@ -1,4 +1,6 @@
-
+import argparse
+parser = argparse.ArgumentParser()
+parser.add_argument('--filename', required = True)
 def change_path(filename):
     f = open(filename, 'r')
     f1 = open('output.txt', 'w+')
@@ -11,5 +13,6 @@ def change_path(filename):
     f1.close()
 
 if __name__ == "__main__":
-    filename = 'test.txt'
-    change_path(filename)
+    args = parser.parse_args()
+    print(args.filename)
+    change_path(args.filename)

shuffle_file_contents.py

@@ -1,17 +1,22 @@
 import random
-
+import argparse
+import os
+parser = argparse.ArgumentParser()
+parser.add_argument('--filename', required = True)
 class Shuffle:
 
-    def shuffle_file_elements(self, filepath, output_file_path):
+    def shuffle_file_elements(self, filepath):
         with open(filepath, 'r') as f:
             read_data = f.read()
             items = read_data.split('\n')
             random.shuffle(items)
-        with open(output_file_path, 'w') as output:
+        os.remove(filepath)
+        with open(filepath, 'w') as output:
             for line in items:
                 output.write(line)
                 output.write('\n')
 
 if __name__ == "__main__":
+    args = parser.parse_args()
     shuffle_obj = Shuffle()
-    shuffle_obj.shuffle_file_elements('train_v0_3.txt', 'output_train_v0_3.txt')
+    shuffle_obj.shuffle_file_elements(args.filename)