I added new option "-e", if you use this option, the script is going …

…to update elevation. If you use script without this option, the script will skip to update elevation.

epanet/tasks.py

@@ -22,7 +22,8 @@ def __init__(self, args):
         self.main_dir = datetime.datetime.now().strftime('%Y%m%d_%H%M%S') + "_epanet_data"
         self.exportdir_list = []
         wss_list_obj = WaterSupplySystems()
-        wss_list_obj.update_elevations(self.db)
+        if args.elevation == True:
+            wss_list_obj.update_elevations(self.db)
         self.wss_list = wss_list_obj.get_wss_list(self.db)
 
     def get_tasks(self):

postgis2epanet.py

@@ -11,7 +11,7 @@ def create_argument_parser():
     parser = argparse.ArgumentParser(
         description="Create EPANET INP file each WSS/District from PostGIS database.",
         epilog="Example usage: python postgis2epanet.py -d yourdatabase -H localhost - p 5432 "
-               "-u user -w securePassword -l list_of_distID(seperated by comma)"
+               "-u user -w securePassword -l list_of_distID(seperated by comma) -e(optional)"
     )
     parser.add_argument("-d", "--database", dest="database",
                         type=str, required=True,
@@ -38,6 +38,10 @@ def create_argument_parser():
                         default="", type=str,
                         help="List of district ID which you want to export. For example, '51,52,53'")
 
+    parser.add_argument("-e", "--elevation", action="store_true",
+                        default=False,
+                        help="If you use this option, the script is going to update elevation field of all layers.")
+
     return parser.parse_args()