76 lines
2.1 KiB
Python
76 lines
2.1 KiB
Python
import os
|
|
import pickle
|
|
import subprocess
|
|
import sys
|
|
|
|
from maze.modules import bruh
|
|
from maze.modules.PlayerBase_func import databaseinit
|
|
|
|
user = password = None
|
|
|
|
with open("credentials.pickle", "rb") as f:
|
|
try:
|
|
while True:
|
|
d = pickle.load(f)
|
|
if d["credtype"] == "mysql":
|
|
user = d["user"]
|
|
password = d["pass"]
|
|
break
|
|
except:
|
|
user = password = None
|
|
|
|
|
|
def getcreds():
|
|
if user:
|
|
return user, password
|
|
else:
|
|
print(
|
|
"""MySQL username or password not initialised. Defaulting to 'root' and ''.
|
|
Run 'python starter.py initsql' to initialise credentials of your choice. """
|
|
)
|
|
with open("credentials.pickle", "rb+") as f:
|
|
try:
|
|
while True:
|
|
pos = f.tell()
|
|
d = pickle.load(f)
|
|
if d["credtype"] == "mysql":
|
|
d["user"] = "root"
|
|
d["pass"] = ""
|
|
f.seek(pos)
|
|
pickle.dump(d, f)
|
|
except:
|
|
pass
|
|
return None, None
|
|
|
|
|
|
if len(sys.argv) == 1:
|
|
getcreds()
|
|
bruh()
|
|
sys.exit()
|
|
else:
|
|
if sys.argv[1] == "dumpsample":
|
|
getcreds()
|
|
databaseinit()
|
|
subprocess.call(
|
|
f"mysql -u {user} --password={password} -D labyrinth < {os.path.abspath('dbdump.sql')}",
|
|
shell=True # ,
|
|
# stdout=subprocess.DEVNULL,
|
|
# stderr=subprocess.DEVNULL,
|
|
)
|
|
print("Successfully dumped sample data")
|
|
elif sys.argv[1] == "initsql":
|
|
user = input("Enter MySQL username: ")
|
|
password = input("Enter MySQL password: ")
|
|
with open("credentials.pickle", "rb+") as f:
|
|
try:
|
|
while True:
|
|
pos = f.tell()
|
|
d = pickle.load(f)
|
|
if d["credtype"] == "mysql":
|
|
d["user"] = user
|
|
d["pass"] = password
|
|
f.seek(pos)
|
|
pickle.dump(d, f)
|
|
except:
|
|
pass
|
|
print("Successfully set.")
|