mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-09-29 16:50:02 +08:00
update code
This commit is contained in:
@@ -0,0 +1,29 @@
|
||||
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()
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user