Genshin-bots/ArknightsUID
1
0
Fork 0
mirror of https://github.com/baiqwerdvd/ArknightsUID.git synced 2025-05-05 11:33:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

add ark databese adapter

Browse Source
This commit is contained in:
qwerdvd 2023-11-02 15:36:33 +08:00
parent 18c74cf15d
commit c647842145
No known key found for this signature in database
GPG Key ID: A3AF89C783404769
2 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ArknightsUID/arknightsuid_start/__init__.py
View File

@ -4,11 +4,13 @@ import threading
from loguru import logger from loguru import logger
from ..arknightsuid_resource import startup from ..arknightsuid_resource import startup
from ..utils.database.startup import ark_adapter
async def all_start(): async def all_start():
try: try:
await startup() await startup()
await ark_adapter()
except Exception as e: except Exception as e:
logger.exception(e) logger.exception(e)

20
ArknightsUID/utils/database/startup.py Normal file
View File

@ -0,0 +1,20 @@
from gsuid_core.server import on_core_start
from gsuid_core.utils.database.base_models import async_maker
from sqlalchemy.sql import text
exec_list = [
'ALTER TABLE arknightspush ADD COLUMN version_push BOOLEAN DEFAULT FALSE;',
'ALTER TABLE arknightspush ADD COLUMN version_is_push BOOLEAN DEFAULT FALSE;',
'ALTER TABLE arknightsuser ADD COLUMN token TEXT;',
]
@on_core_start
async def ark_adapter():
async with async_maker() as session:
for _t in exec_list:
try:
await session.execute(text(_t))
await session.commit()
except: # noqa: E722
pass