Merge pull request geekcomputers#116 from manishSRM/master

added parser for command line argument in batch_file_rename.py

Author: geekcomputers

batch_file_rename.py

@@ -11,7 +11,7 @@
 
 import os
 import sys
-
+import argparse
 
 def batch_rename(work_dir, old_ext, new_ext):
     '''
@@ -32,17 +32,27 @@ def batch_rename(work_dir, old_ext, new_ext):
                 os.path.join(work_dir, newfile)
             )
 
+def get_parser():
+    parser = argparse.ArgumentParser(description='change extension of files in a working directory')
+    parser.add_argument('work_dir', metavar='WORK_DIR', type=str, nargs=1, help='the directory where to change extension')
+    parser.add_argument('old_ext', metavar='OLD_EXT', type=str, nargs=1, help='old extension')
+    parser.add_argument('new_ext', metavar='NEW_EXT', type=str, nargs=1, help='new extension')
+    return parser
 
 def main():
     '''
     This will be called if the script is directly invoked.
     '''
+    # adding command line argument
+    parser = get_parser()
+    args = vars(parser.parse_args())
+
     # Set the variable work_dir with the first argument passed
-    work_dir = sys.argv[1]
+    work_dir = args['work_dir'][0]
     # Set the variable old_ext with the second argument passed
-    old_ext = sys.argv[2]
+    old_ext = args['old_ext'][0]
     # Set the variable new_ext with the third argument passed
-    new_ext = sys.argv[3]
+    new_ext = args['new_ext'][0]
 
     batch_rename(work_dir, old_ext, new_ext)