mirror of
https://github.com/A-Minos/nonebot-plugin-tetris-stats.git
synced 2026-03-05 05:36:54 +08:00
80 lines
2.2 KiB
Python
80 lines
2.2 KiB
Python
from sqlite3 import connect, Connection
|
|
import os
|
|
|
|
from nonebot import get_driver
|
|
from nonebot.log import logger
|
|
|
|
|
|
_DB_FILE = 'data/nonebot_plugin_tetris_stats/data.db'
|
|
_DB: Connection | None = None
|
|
|
|
driver = get_driver()
|
|
|
|
|
|
@driver.on_startup
|
|
async def _():
|
|
'''初始化数据库'''
|
|
await init_db()
|
|
|
|
|
|
@driver.on_shutdown
|
|
async def _():
|
|
if isinstance(_DB, Connection):
|
|
_DB.close()
|
|
|
|
|
|
async def init_db() -> Connection:
|
|
'''初始化数据库'''
|
|
if not os.path.exists(os.path.dirname(_DB_FILE)):
|
|
if os.path.exists('data/nonebot-plugin-tetris-stats'): # 重命名旧的数据库路径
|
|
os.rename('data/nonebot-plugin-tetris-stats',
|
|
os.path.dirname(_DB_FILE))
|
|
else:
|
|
os.makedirs(os.path.dirname(_DB_FILE))
|
|
global _DB
|
|
_DB = connect(_DB_FILE)
|
|
cursor = _DB.cursor()
|
|
cursor.execute('''CREATE TABLE IF NOT EXISTS IOBIND
|
|
(QQ INTEGER NOT NULL,
|
|
USER TEXT NOT NULL)''')
|
|
cursor.execute('''CREATE TABLE IF NOT EXISTS TOPBIND
|
|
(QQ INTEGER NOT NULL,
|
|
USER TEXT NOT NULL)''')
|
|
_DB.commit()
|
|
logger.info('数据库初始化完成')
|
|
return _DB
|
|
|
|
|
|
async def get_db() -> Connection:
|
|
'''获取数据库对象'''
|
|
return _DB or await init_db()
|
|
|
|
|
|
async def query_bind_info(qq_number: str | int, game_type: str) -> str | None:
|
|
'''查询绑定信息'''
|
|
db = await get_db()
|
|
cursor = db.cursor()
|
|
cursor.execute(f'SELECT USER FROM {game_type}BIND WHERE QQ = {qq_number}')
|
|
user = cursor.fetchone()
|
|
db.commit()
|
|
if user is None:
|
|
return None
|
|
return user[0]
|
|
|
|
|
|
async def write_bind_info(qq_number: str | int, user: str, game_type: str) -> str:
|
|
'''写入绑定信息'''
|
|
bind_info = await query_bind_info(qq_number, game_type)
|
|
db = await get_db()
|
|
cursor = db.cursor()
|
|
if bind_info is not None:
|
|
cursor.execute(
|
|
f'UPDATE {game_type}BIND SET USER = ? WHERE QQ = ?', (user, qq_number))
|
|
message = '更新成功'
|
|
elif bind_info is None:
|
|
cursor.execute(
|
|
f'INSERT INTO {game_type}BIND (QQ, USER) VALUES (?, ?)', (qq_number, user))
|
|
message = '绑定成功'
|
|
db.commit()
|
|
return message
|