kaku/PyGetGPT
1
0
Fork 0
mirror of https://github.com/Kakune55/PyGetGPT.git synced 2025-05-06 18:29:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

移除对MySQL服务器的支持

Browse Source
This commit is contained in:
Kakune55 2023-12-15 11:20:47 +08:00
parent 7b66cbb7e4
commit afdbdec803
2 changed files with 18 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
db.py
View File

@ -1,25 +1,18 @@
import config , uuid
import config , uuid , pathlib
def getconn():
try:
if config.readConf()["db"]["type"] == "sqlite3":
import sqlite3
conn = sqlite3.connect("APPData.db")
elif config.readConf()["db"]["type"] == "mysql":
import pymysql
conn = pymysql.connect(host=config.readConf()["db"]["host"],
port=config.readConf()["db"]["port"],
user=config.readConf()["db"]["user"],
password=config.readConf()["db"]["passwd"],
database=config.readConf()["db"]["database"])
conn = sqlite3.connect(config.readConf()["db"]["path"])
return conn
except:
print("DB ERROR")
return 0
def dbIsOK():
#打开数据库连接
if "init" not in config.readConf():
config.updateConf({"init" : True})
path = pathlib.Path(config.readConf()["db"]["path"])
if not path.is_file():
init()
try:
@ -53,10 +46,7 @@ def userSurplus(userkey): #查询userkey剩余配额
cursor = db.cursor()
# 使用 execute() 方法执行 SQL 查询
if config.readConf()["db"]["type"] == "sqlite3":
cursor.execute(f"SELECT surplus FROM usersurplus WHERE userkey = ?;",[userkey])
else:
cursor.execute(f"SELECT surplus FROM usersurplus WHERE userkey = %s;",[userkey])
# 使用 fetchone() 方法获取单条数据.
data = cursor.fetchone()
@ -74,10 +64,7 @@ def reduce_value(userkey, value): # 减去对应的值
cursor = db.cursor()
# 执行 SQL 查询以获取当前值
if config.readConf()["db"]["type"] == "sqlite3":
cursor.execute(f"SELECT surplus FROM usersurplus WHERE userkey = ?;",[userkey])
else:
cursor.execute(f"SELECT surplus FROM usersurplus WHERE userkey = %s;",[userkey])
current_value = cursor.fetchone()[0]
# 如果没有找到用户，则返回错误信息
@ -89,10 +76,7 @@ def reduce_value(userkey, value): # 减去对应的值
new_value = current_value - value
# 更新数据库中的值
if config.readConf()["db"]["type"] == "sqlite3":
cursor.execute(f"UPDATE usersurplus SET surplus= ? WHERE userkey= ?;",[new_value,userkey])
else:
cursor.execute(f"UPDATE usersurplus SET surplus= %s WHERE userkey= %s;",[new_value,userkey])
# 提交事务
db.commit()
@ -127,10 +111,7 @@ def delKey(userkey):
cursor = db.cursor()
# 使用 execute() 方法执行 SQL 查询
if config.readConf()["db"]["type"] == "sqlite3":
cursor.execute(f"DELETE FROM usersurplus WHERE userkey = ?;", [userkey])
else:
cursor.execute(f"DELETE FROM usersurplus WHERE userkey = %s;", [userkey])
# 提交事务
db.commit()
@ -150,7 +131,7 @@ def createKey(quota,number=1,key="null"):
# 使用 execute() 方法执行 SQL 查询
output = []
if config.readConf()["db"]["type"] == "sqlite3":
if key == "null":
for i in range(int(number)):
key = str(uuid.uuid1())
@ -159,15 +140,6 @@ def createKey(quota,number=1,key="null"):
else:
cursor.execute(f"INSERT INTO usersurplus (userkey,surplus) VALUES (?, ?);", [key, quota])
output.append(key)
else:
if key == "null":
for i in range(int(number)):
key = str(uuid.uuid1())
output.append(key)
cursor.execute(f"INSERT INTO usersurplus (userkey,surplus) VALUES (%s, %s);", [key, quota])
else:
cursor.execute(f"INSERT INTO usersurplus (userkey,surplus) VALUES (%s, %s);", [key, quota])
output.append(key)
# 提交事务

2
main.py
View File

@ -61,7 +61,7 @@ def login():
if flask.request.method == 'GET':
return flask.render_template('login.html')
userRequest = flask.request.form
if userRequest["password"] != config.readConf()["appconf"]["adminkey"]:
if userRequest["password"] != config.readConf()["adminkey"]:
return flask.render_template('login.html')
flask.session["admin"] = True
return flask.redirect(flask.url_for('admin'))