PaiGramTeam/PaiGram
3
0
Fork 0
mirror of https://github.com/PaiGramTeam/PaiGram.git synced 2024-11-16 04:35:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

🐛 Fix the incorrect account_id retrieval

Browse Source
This commit is contained in:
洛水居室 2023-03-15 19:42:16 +08:00
parent e2649f63ee
commit 3206d5dccd
No known key found for this signature in database
GPG Key ID: C9DE87DA724B88FC
1 changed files with 16 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
alembic/versions/ddcfba3c7d5c_v4.py
View File

@ -132,13 +132,15 @@ def upgrade() -> None:
y_uid = row["yuanshen_uid"]
g_uid = row["genshin_uid"]
region = row["region"]
account_id = None
cookies_row = connection.execute(
cookies_table.select().where(cookies_table.c.user_id == user_id)
).first()
if cookies_row is not None:
account_id = cookies_row["account_id"]
if y_uid:
account_id = None
cookies_row = connection.execute(
cookies_table.select()
.where(cookies_table.c.user_id == user_id)
.where(cookies_table.c.region == "HYPERION")
).first()
if cookies_row is not None:
account_id = cookies_row["account_id"]
insert = players_table.insert().values(
user_id=int(user_id),
player_id=int(y_uid),
@ -149,6 +151,14 @@ def upgrade() -> None:
with op.get_context().autocommit_block():
connection.execute(insert)
if g_uid:
account_id = None
cookies_row = connection.execute(
cookies_table.select()
.where(cookies_table.c.user_id == user_id)
.where(cookies_table.c.region == "HOYOLAB")
).first()
if cookies_row is not None:
account_id = cookies_row["account_id"]
insert = players_table.insert().values(
user_id=int(user_id),
player_id=int(g_uid),