mirror of
https://github.com/A-Minos/nonebot-plugin-tetris-stats.git
synced 2026-02-03 18:45:32 +08:00
c8832bd1c9
* 🔧 启用一些 ruff 的新规则 * ➕ 添加开发依赖 nonebot-adapter-qq * ➕ 添加依赖 nonebot-plugin-session * ➕ 添加依赖 nonebot-plugin-session-orm * 🔧 忽略 ruff 规则 ISC001 format 冲突风险 * 🚨 修复 ruff 警报 * ♻️ 重构！ * ♻️ 恢复定时获取 TetraLeague 数据的功能 * ✨ 统一处理需要捕获的错误 * ✨ 记录用户触发的指令
70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
"""Add redundant platform field
|
|
|
|
迁移 ID: 6c3206f90cc3
|
|
父迁移: 9f6582279ce2
|
|
创建时间: 2023-11-26 20:15:56.033892
|
|
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from alembic import op
|
|
from sqlalchemy.ext.automap import automap_base
|
|
from sqlalchemy.orm import Session
|
|
from ujson import dumps, loads
|
|
|
|
if TYPE_CHECKING:
|
|
from collections.abc import Sequence
|
|
|
|
revision: str = '6c3206f90cc3'
|
|
down_revision: str | Sequence[str] | None = '9f6582279ce2'
|
|
branch_labels: str | Sequence[str] | None = None
|
|
depends_on: str | Sequence[str] | None = None
|
|
|
|
|
|
def upgrade(name: str = '') -> None:
|
|
if name:
|
|
return
|
|
|
|
Base = automap_base() # noqa: N806
|
|
connection = op.get_bind()
|
|
Base.prepare(autoload_with=connection)
|
|
|
|
HistoricalData = Base.classes.nonebot_plugin_tetris_stats_historicaldata # noqa: N806
|
|
|
|
with Session(connection) as session:
|
|
for row in session.query(HistoricalData):
|
|
platform = row.game_platform
|
|
game_user = loads(row.game_user)
|
|
processed_data = loads(row.processed_data)
|
|
game_user['platform'] = platform
|
|
processed_data['platform'] = platform
|
|
row.game_user = dumps(game_user)
|
|
row.processed_data = dumps(processed_data)
|
|
session.add(row)
|
|
session.commit()
|
|
|
|
|
|
def downgrade(name: str = '') -> None:
|
|
if name:
|
|
return
|
|
|
|
Base = automap_base() # noqa: N806
|
|
connection = op.get_bind()
|
|
Base.prepare(autoload_with=connection)
|
|
|
|
HistoricalData = Base.classes.nonebot_plugin_tetris_stats_historicaldata # noqa: N806
|
|
|
|
with Session(connection) as session:
|
|
for row in session.query(HistoricalData):
|
|
game_user = loads(row.game_user)
|
|
processed_data = loads(row.processed_data)
|
|
game_user.pop('platform', None)
|
|
processed_data.pop('platform', None)
|
|
row.game_user = dumps(game_user)
|
|
row.processed_data = dumps(processed_data)
|
|
session.add(row)
|
|
session.commit()
|