Make database selectable
Signed-off-by: fly <merspieler@airmail.cc>
This commit is contained in:
parent
20da6220a2
commit
c59366776b
1 changed files with 6 additions and 1 deletions
|
@ -28,6 +28,7 @@ port = 12345
|
||||||
verbose = False
|
verbose = False
|
||||||
dbuser = ""
|
dbuser = ""
|
||||||
dbpw = ""
|
dbpw = ""
|
||||||
|
dbname = "worldbuild"
|
||||||
|
|
||||||
argc = len(sys.argv)
|
argc = len(sys.argv)
|
||||||
i = 1
|
i = 1
|
||||||
|
@ -44,6 +45,9 @@ while i < argc:
|
||||||
elif sys.argv[i] == "-p" or sys.argv[i] == "--password":
|
elif sys.argv[i] == "-p" or sys.argv[i] == "--password":
|
||||||
i += 1
|
i += 1
|
||||||
dbpw = sys.argv[i]
|
dbpw = sys.argv[i]
|
||||||
|
elif sys.argv[i] == "-d" or sys.argv[i] == "--database":
|
||||||
|
i += 1
|
||||||
|
dbname = sys.argv[i]
|
||||||
elif sys.argv[i] == "-v":
|
elif sys.argv[i] == "-v":
|
||||||
verbose = True
|
verbose = True
|
||||||
elif sys.argv[i] == "-h" or sys.argv[i] == "--help":
|
elif sys.argv[i] == "-h" or sys.argv[i] == "--help":
|
||||||
|
@ -51,6 +55,7 @@ while i < argc:
|
||||||
print("Handles job asignments and keeps track of the status")
|
print("Handles job asignments and keeps track of the status")
|
||||||
print("")
|
print("")
|
||||||
print("OPTIONS")
|
print("OPTIONS")
|
||||||
|
print(" -d, --database Database name")
|
||||||
print(" -u, --user Database user")
|
print(" -u, --user Database user")
|
||||||
print(" -p, --password Database password")
|
print(" -p, --password Database password")
|
||||||
print(" -v, --verbose Verbose printouts")
|
print(" -v, --verbose Verbose printouts")
|
||||||
|
@ -71,7 +76,7 @@ if dbpw == "":
|
||||||
print("ERROR: No database password given")
|
print("ERROR: No database password given")
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
|
|
||||||
db = pymysql.connect("localhost", dbuser, dbpw, "worldbuild")
|
db = pymysql.connect("localhost", dbuser, dbpw, dbname)
|
||||||
|
|
||||||
cursor = db.cursor()
|
cursor = db.cursor()
|
||||||
|
|
||||||
|
|
Loading…
Add table
Reference in a new issue