snapshot/scheduler: Modified main() function to take arguments.

Modified the main function to take script arguments, so that this
script can be used as a module by other programs .

Change-Id: I902f0bc7ddfbf0d335cc087f51b1a7af4b7157fc
BUG: 1220670
Signed-off-by: n Darshan <dnarayan@redhat.com>
Reviewed-on: http://review.gluster.org/10760
Tested-by: Gluster Build System <jenkins@build.gluster.com>
Tested-by: NetBSD Build System <jenkins@build.gluster.org>
Reviewed-by: Aravinda VK <avishwan@redhat.com>
Reviewed-by: Krishnan Parthasarathi <kparthas@redhat.com>

diff --git a/extras/snap_scheduler/snap_scheduler.py b/extras/snap_scheduler/snap_scheduler.py
index 1c5f9380dd8fbc61f06581cfd051c4e9c6e0478a..ae5a84dacfaec647cddabe2e0bdf3276de1444e0 100755 (executable)
--- a/extras/snap_scheduler/snap_scheduler.py
+++ b/extras/snap_scheduler/snap_scheduler.py
@@ -581,7 +581,7 @@ def perform_operation(args):
     return ret
 
 
-def main():
+def main(argv):
     initLogger()
     ret = -1
     parser = argparse.ArgumentParser()
@@ -617,7 +617,7 @@ def main():
     parser_edit.add_argument("schedule", help="Schedule")
     parser_edit.add_argument("volname", help="Volume Name")
 
-    args = parser.parse_args()
+    args = parser.parse_args(argv)
 
     if not os.path.exists(SHARED_STORAGE_DIR):
         output("Failed: "+SHARED_STORAGE_DIR+" does not exist.")
@@ -675,4 +675,4 @@ def main():
 
 
 if __name__ == "__main__":
-    sys.exit(main())
+    sys.exit(main(sys.argv[1:]))