Skip to content

Commit 376e8a1

Browse files
committed
add migration warning
1 parent 5199119 commit 376e8a1

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

nonebot_plugin_chatrecorder/migrations/902a51ac4032_add_session.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
"""
88
import sqlalchemy as sa
99
from alembic import op
10+
from nonebot.log import logger
1011
from sqlalchemy import insert, select, update
1112
from sqlalchemy.ext.automap import automap_base
1213
from sqlalchemy.orm import Session
@@ -25,6 +26,8 @@ def upgrade() -> None:
2526
MessageRecord = Base.classes.nonebot_plugin_chatrecorder_messagerecord
2627
SessionModel = Base.classes.nonebot_plugin_session_sessionmodel
2728
with Session(op.get_bind()) as session:
29+
logger.warning("正在迁移聊天记录数据,请不要关闭程序...")
30+
2831
statement = select(MessageRecord)
2932
messages = session.scalars(statement).all()
3033

@@ -105,6 +108,7 @@ def upgrade() -> None:
105108
)
106109
if bulk_update_messages:
107110
session.execute(update(MessageRecord), bulk_update_messages)
111+
logger.warning("聊天记录数据迁移完成!")
108112

109113
with op.batch_alter_table(
110114
"nonebot_plugin_chatrecorder_messagerecord", schema=None

0 commit comments

Comments
 (0)