PaiGramTeam/PaiGram
3
0
Fork 0
mirror of https://github.com/PaiGramTeam/PaiGram.git synced 2024-11-21 22:58:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

🐛 修复之前绑定另一个用户后导致的 cookies 写入问题

Browse Source
This commit is contained in:
洛水居室 2022-09-05 10:33:56 +08:00
parent 77c02e03d8
commit 86503671ed
No known key found for this signature in database
GPG Key ID: C9DE87DA724B88FC
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
plugins/genshin/adduser.py
View File

@ -3,6 +3,7 @@ from typing import Optional
import genshin import genshin
from genshin import InvalidCookies, GenshinException, DataNotPublic from genshin import InvalidCookies, GenshinException, DataNotPublic
from sqlalchemy.exc import NoResultFound
from telegram import Update, ReplyKeyboardRemove, ReplyKeyboardMarkup, TelegramObject from telegram import Update, ReplyKeyboardRemove, ReplyKeyboardMarkup, TelegramObject
from telegram.ext import CallbackContext, CommandHandler, MessageHandler, filters, ConversationHandler from telegram.ext import CallbackContext, CommandHandler, MessageHandler, filters, ConversationHandler
from telegram.helpers import escape_markdown from telegram.helpers import escape_markdown
@ -211,8 +212,13 @@ class AddUser(BasePlugins):
await update.message.reply_text("数据错误") await update.message.reply_text("数据错误")
return ConversationHandler.END return ConversationHandler.END
await self.user_service.update_user(user_db) await self.user_service.update_user(user_db)
# 临时解决错误
try:
await self.cookies_service.update_cookies(user.id, add_user_command_data.cookies, await self.cookies_service.update_cookies(user.id, add_user_command_data.cookies,
add_user_command_data.region) add_user_command_data.region)
except NoResultFound:
await self.cookies_service.add_cookies(user.id, add_user_command_data.cookies,
add_user_command_data.region)
Log.info(f"用户 {user.full_name}[{user.id}] 绑定账号成功") Log.info(f"用户 {user.full_name}[{user.id}] 绑定账号成功")
await update.message.reply_text("保存成功", reply_markup=ReplyKeyboardRemove()) await update.message.reply_text("保存成功", reply_markup=ReplyKeyboardRemove())
return ConversationHandler.END return ConversationHandler.END