96 lines
2.9 KiB
Python
96 lines
2.9 KiB
Python
|
from .cfg import config
|
||
|
|
import uuid
|
||
|
|
import aiomysql
|
||
|
|
import sys
|
||
|
|
|
||
|
|
HOST = config["db"]["host"]
|
||
|
|
PORT = config["db"]["port"]
|
||
|
|
USER = config["db"]["user"]
|
||
|
|
PASSWD = config["db"]["passwd"]
|
||
|
|
DB = config["db"]["db"]
|
||
|
|
|
||
|
|
MAXSIZE = config["db"]["maxsize"]
|
||
|
|
MINSIZE = config["db"]["minsize"]
|
||
|
|
|
||
|
|
|
||
|
|
async def connect_db():
|
||
|
|
global db
|
||
|
|
sys.stdout.write("connect db\n")
|
||
|
|
db = await aiomysql.create_pool(
|
||
|
|
host=HOST,
|
||
|
|
port=PORT,
|
||
|
|
user=USER,
|
||
|
|
password=PASSWD,
|
||
|
|
db=DB,
|
||
|
|
maxsize=MAXSIZE,
|
||
|
|
minsize=MINSIZE
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
async def create_db():
|
||
|
|
async with db.acquire() as conn:
|
||
|
|
await (await conn.cursor()).execute(
|
||
|
|
"CREATE TABLE IF NOT EXISTS users (username VARCHAR(255) PRIMARY KEY, password VARCHAR(255), email VARCHAR(255), accoutpwd VARCHAR(255))"
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
async def get_user(username: str):
|
||
|
|
async with db.acquire() as conn:
|
||
|
|
async with conn.cursor() as cur:
|
||
|
|
await cur.execute("SELECT username, password FROM users WHERE username = %s", (username,))
|
||
|
|
result = await cur.fetchone()
|
||
|
|
if result:
|
||
|
|
return result[1]
|
||
|
|
return None
|
||
|
|
|
||
|
|
|
||
|
|
async def get_user_passwd(username: str):
|
||
|
|
async with db.acquire() as conn:
|
||
|
|
async with conn.cursor() as cur:
|
||
|
|
await cur.execute("SELECT accoutpwd FROM users WHERE username = %s", (username,))
|
||
|
|
result = await cur.fetchone()
|
||
|
|
if result:
|
||
|
|
return result[0]
|
||
|
|
return ""
|
||
|
|
|
||
|
|
|
||
|
|
async def create_user(username: str, password: str, email: str):
|
||
|
|
async with db.acquire() as conn:
|
||
|
|
async with conn.cursor() as cur:
|
||
|
|
# 检查用户是否已存在
|
||
|
|
await cur.execute("SELECT username FROM users WHERE username = %s", (username,))
|
||
|
|
if await cur.fetchone():
|
||
|
|
return 1
|
||
|
|
# 创建新用户
|
||
|
|
await cur.execute("INSERT INTO users (username, password, email, accoutpwd) VALUES (%s, %s, %s, %s)", (username, password, email, uuid.uuid4().hex))
|
||
|
|
await conn.commit()
|
||
|
|
return 0
|
||
|
|
|
||
|
|
|
||
|
|
async def check_user(username: str):
|
||
|
|
async with db.acquire() as conn:
|
||
|
|
async with conn.cursor() as cur:
|
||
|
|
# 检查用户是否已存在
|
||
|
|
await cur.execute("SELECT username FROM users WHERE username = %s", (username,))
|
||
|
|
if await cur.fetchone():
|
||
|
|
return 1
|
||
|
|
return 0
|
||
|
|
|
||
|
|
|
||
|
|
async def update_passwd(username: str, password: str):
|
||
|
|
async with db.acquire() as conn:
|
||
|
|
async with conn.cursor() as cur:
|
||
|
|
await cur.execute("UPDATE users SET password = %s WHERE username = %s", (password, username))
|
||
|
|
await conn.commit()
|
||
|
|
return 0
|
||
|
|
|
||
|
|
|
||
|
|
async def get_email(username: str):
|
||
|
|
async with db.acquire() as conn:
|
||
|
|
async with conn.cursor() as cur:
|
||
|
|
await cur.execute("SELECT email FROM users WHERE username = %s", (username,))
|
||
|
|
result = await cur.fetchone()
|
||
|
|
if result:
|
||
|
|
return result[0]
|
||
|
|
return None
|