Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2025-12-15 14:22:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

🐛 : 用户数据迁移包括uid

Browse Source
This commit is contained in:
HibiKier 2024-07-30 00:06:29 +08:00
parent cd4da389c3
commit b4b30e59c7
2 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
zhenxun/builtin_plugins/__init__.py
View File

@ -7,6 +7,7 @@ from tortoise import Tortoise
from tortoise.exceptions import OperationalError from tortoise.exceptions import OperationalError
from zhenxun.models.goods_info import GoodsInfo from zhenxun.models.goods_info import GoodsInfo
from zhenxun.models.group_member_info import GroupInfoUser
from zhenxun.models.sign_user import SignUser from zhenxun.models.sign_user import SignUser
from zhenxun.models.user_console import UserConsole from zhenxun.models.user_console import UserConsole
from zhenxun.services.log import logger from zhenxun.services.log import logger
@ -69,6 +70,8 @@ async def _():
and not await SignUser.annotate().count() and not await SignUser.annotate().count()
): ):
try: try:
group_user = await GroupInfoUser.filter(uid__isnull=False).all()
user2uid = {u.user_id: u.uid for u in group_user}
flag = False flag = False
db = Tortoise.get_connection("default") db = Tortoise.get_connection("default")
old_sign_list = await db.execute_query_dict(SIGN_SQL) old_sign_list = await db.execute_query_dict(SIGN_SQL)
@ -79,7 +82,7 @@ async def _():
} }
create_list = [] create_list = []
sign_id_list = [] sign_id_list = []
uid = await UserConsole.get_new_uid() max_uid = max(user2uid.values()) + 1
for old_sign in old_sign_list: for old_sign in old_sign_list:
sign_id_list.append(old_sign["user_id"]) sign_id_list.append(old_sign["user_id"])
old_bag = [ old_bag = [
@ -97,16 +100,20 @@ async def _():
UserConsole( UserConsole(
user_id=old_sign["user_id"], user_id=old_sign["user_id"],
platform="qq", platform="qq",
uid=uid, uid=user2uid.get(old_sign["user_id"]) or max_uid,
props=props, props=props,
gold=old_bag["gold"], gold=old_bag["gold"],
) )
) )
if not user2uid.get(old_sign["user_id"]):
max_uid += 1
else: else:
create_list.append( create_list.append(
UserConsole(user_id=old_sign["user_id"], platform="qq", uid=uid) UserConsole(
user_id=old_sign["user_id"], platform="qq", uid=max_uid
)
) )
uid += 1 max_uid += 1
if create_list: if create_list:
logger.info("开始迁移用户数据...") logger.info("开始迁移用户数据...")
await UserConsole.bulk_create(create_list, 10) await UserConsole.bulk_create(create_list, 10)

3
zhenxun/models/user_console.py
View File

@ -131,7 +131,8 @@ class UserConsole(Model):
platform: 平台. platform: 平台.
""" """
user, _ = await cls.get_or_create( user, _ = await cls.get_or_create(
user_id=user_id, defaults={"platform": platform, "uid": cls.get_new_uid()} user_id=user_id,
defaults={"platform": platform, "uid": await cls.get_new_uid()},
) )
if goods_uuid not in user.props: if goods_uuid not in user.props:
user.props[goods_uuid] = 0 user.props[goods_uuid] = 0