PaiGramTeam/PamGram
3
0
Fork 0
mirror of https://github.com/PaiGramTeam/PamGram.git synced 2024-11-16 03:55:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

🐛 Fix IntegrityError in setuid

Browse Source
This commit is contained in:
xtaodada 2023-10-17 20:14:57 +08:00
parent 22a8b9883b
commit 0b4dd0879a
Signed by: xtaodada
GPG Key ID: 4CBB3F4FA8C85659
1 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
plugins/account/account.py
View File

@ -250,6 +250,18 @@ class BindAccountPlugin(Plugin.Conversation):
await message.reply_markdown_v2(text, reply_markup=ReplyKeyboardMarkup(reply_keyboard, one_time_keyboard=True))
return COMMAND_RESULT
async def update_player_info(self, player: Player, nickname: str):
player_info = await self.player_info_service.get(player)
if player_info is None:
player_info = PlayerInfoSQLModel(
user_id=player.user_id,
player_id=player.player_id,
nickname=nickname,
create_time=datetime.now(),
is_update=True,
) # 不添加更新时间
await self.player_info_service.add(player_info)
@conversation.state(state=COMMAND_RESULT)
@handler.message(filters=filters.TEXT & ~filters.COMMAND, block=False)
async def command_result(self, update: "Update", context: "ContextTypes.DEFAULT_TYPE") -> int:
@ -276,14 +288,7 @@ class BindAccountPlugin(Plugin.Conversation):
is_chosen=is_chosen, # todo 多账号
)
await self.players_service.add(player)
player_info = PlayerInfoSQLModel(
user_id=player.user_id,
player_id=player.player_id,
nickname=nickname,
create_time=datetime.now(),
is_update=True,
) # 不添加更新时间
await self.player_info_service.add(player_info)
await self.update_player_info(player, nickname)
logger.success("用户 %s[%s] 绑定UID账号成功", user.full_name, user.id)
await message.reply_text("保存成功", reply_markup=ReplyKeyboardRemove())
return ConversationHandler.END