mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
30 lines
918 B
Python
30 lines
918 B
Python
from pathlib import Path
|
|
from services.db_context import db
|
|
from asyncpg.exceptions import DuplicateColumnError
|
|
import nonebot
|
|
|
|
driver = nonebot.get_driver()
|
|
|
|
|
|
@driver.on_startup
|
|
async def _init_database():
|
|
file = Path() / 'plugins' / 'database_scripts.py'
|
|
if file.exists():
|
|
update_sql = [
|
|
'ALTER TABLE russian_users ADD winning_streak Integer default 0;',
|
|
'ALTER TABLE russian_users ADD losing_streak Integer default 0;',
|
|
'ALTER TABLE russian_users ADD max_winning_streak Integer default 0;',
|
|
'ALTER TABLE russian_users ADD max_losing_streak Integer default 0;',
|
|
'ALTER TABLE group_info_users ADD uid Integer default 0;'
|
|
]
|
|
for sql in update_sql:
|
|
try:
|
|
query = db.text(sql)
|
|
await db.first(query)
|
|
except DuplicateColumnError:
|
|
pass
|
|
file.unlink()
|
|
|
|
|
|
|