cleaned up generateRandomKsatDataSet_db.py / fixed type casting bug in ArgumentParser

2019-02-08 13:03:15 +01:00
parent 1e1b527566
commit 0a0fbcc46d
3 changed files with 43 additions and 35 deletions
--- a/util/script.py
+++ b/util/script.py
@@ -28,7 +28,8 @@ class ArgParser:
                         
        self.__flags[alias] = {"longFlag": longFlag,
                               "hasDefault": False,
-                               "ignoreDatabaseConfig": ignoreDatabaseConfig}
+                               "ignoreDatabaseConfig": ignoreDatabaseConfig,
+                               "type": type}
        
        if default != None:
            self.__flags[alias]["hasDefault"] = True
@@ -79,7 +80,8 @@ class ArgParser:
            tmpValue = args[longFlag]
        
        if tmpValue == None:
-            tmpValue == input("pass arguement %s: " % longFlag)
+            tmpValue = flag["type"](input("pass arguement %s: " % longFlag))
+
        
        return tmpValue