2023-02-17 07:02:21 +01:00
|
|
|
from typing import Dict, List, Optional
|
|
|
|
from sqlalchemy import Table, Column, UniqueConstraint, PrimaryKeyConstraint, and_
|
|
|
|
from sqlalchemy.types import Integer, String, TIMESTAMP, Boolean, JSON, BigInteger
|
|
|
|
from sqlalchemy.engine.base import Connection
|
|
|
|
from sqlalchemy.schema import ForeignKey
|
|
|
|
from sqlalchemy.engine import Row
|
|
|
|
from sqlalchemy.sql import func, select
|
|
|
|
from sqlalchemy.dialects.mysql import insert
|
|
|
|
|
|
|
|
from core.data.schema import BaseData, metadata
|
|
|
|
|
|
|
|
profile = Table(
|
|
|
|
"chuni_profile_data",
|
|
|
|
metadata,
|
|
|
|
Column("id", Integer, primary_key=True, nullable=False),
|
2023-03-09 17:38:58 +01:00
|
|
|
Column(
|
|
|
|
"user",
|
|
|
|
ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"),
|
|
|
|
nullable=False,
|
|
|
|
),
|
2023-02-17 07:02:21 +01:00
|
|
|
Column("version", Integer, nullable=False),
|
|
|
|
Column("exp", Integer),
|
|
|
|
Column("level", Integer),
|
|
|
|
Column("point", Integer),
|
|
|
|
Column("frameId", Integer),
|
|
|
|
Column("isMaimai", Boolean),
|
|
|
|
Column("trophyId", Integer),
|
|
|
|
Column("userName", String(25)),
|
|
|
|
Column("isWebJoin", Boolean),
|
|
|
|
Column("playCount", Integer),
|
|
|
|
Column("lastGameId", String(25)),
|
|
|
|
Column("totalPoint", BigInteger),
|
|
|
|
Column("characterId", Integer),
|
|
|
|
Column("firstGameId", String(25)),
|
|
|
|
Column("friendCount", Integer),
|
|
|
|
Column("lastPlaceId", Integer),
|
|
|
|
Column("nameplateId", Integer),
|
|
|
|
Column("totalMapNum", Integer),
|
|
|
|
Column("lastAllNetId", Integer),
|
|
|
|
Column("lastClientId", String(25)),
|
|
|
|
Column("lastPlayDate", String(25)),
|
|
|
|
Column("lastRegionId", Integer),
|
|
|
|
Column("playerRating", Integer),
|
|
|
|
Column("totalHiScore", Integer),
|
|
|
|
Column("webLimitDate", String(25)),
|
|
|
|
Column("firstPlayDate", String(25)),
|
|
|
|
Column("highestRating", Integer),
|
|
|
|
Column("lastPlaceName", String(25)),
|
|
|
|
Column("multiWinCount", Integer),
|
|
|
|
Column("acceptResCount", Integer),
|
|
|
|
Column("lastRegionName", String(25)),
|
|
|
|
Column("lastRomVersion", String(25)),
|
|
|
|
Column("multiPlayCount", Integer),
|
|
|
|
Column("firstRomVersion", String(25)),
|
|
|
|
Column("lastDataVersion", String(25)),
|
|
|
|
Column("requestResCount", Integer),
|
|
|
|
Column("successResCount", Integer),
|
|
|
|
Column("eventWatchedDate", String(25)),
|
|
|
|
Column("firstDataVersion", String(25)),
|
|
|
|
Column("reincarnationNum", Integer),
|
|
|
|
Column("playedTutorialBit", Integer),
|
|
|
|
Column("totalBasicHighScore", Integer),
|
|
|
|
Column("totalExpertHighScore", Integer),
|
|
|
|
Column("totalMasterHighScore", Integer),
|
|
|
|
Column("totalRepertoireCount", Integer),
|
|
|
|
Column("firstTutorialCancelNum", Integer),
|
|
|
|
Column("totalAdvancedHighScore", Integer),
|
|
|
|
Column("masterTutorialCancelNum", Integer),
|
2023-03-09 17:38:58 +01:00
|
|
|
Column("ext1", Integer), # Added in chunew
|
2023-02-17 07:02:21 +01:00
|
|
|
Column("ext2", Integer),
|
|
|
|
Column("ext3", Integer),
|
|
|
|
Column("ext4", Integer),
|
|
|
|
Column("ext5", Integer),
|
|
|
|
Column("ext6", Integer),
|
|
|
|
Column("ext7", Integer),
|
|
|
|
Column("ext8", Integer),
|
|
|
|
Column("ext9", Integer),
|
2023-03-09 17:38:58 +01:00
|
|
|
Column("ext10", Integer),
|
2023-02-17 07:02:21 +01:00
|
|
|
Column("extStr1", String(255)),
|
|
|
|
Column("extStr2", String(255)),
|
|
|
|
Column("extLong1", Integer),
|
|
|
|
Column("extLong2", Integer),
|
|
|
|
Column("mapIconId", Integer),
|
|
|
|
Column("compatibleCmVersion", String(25)),
|
2023-03-09 17:38:58 +01:00
|
|
|
Column("medal", Integer),
|
2023-02-17 07:02:21 +01:00
|
|
|
Column("voiceId", Integer),
|
2023-03-09 17:38:58 +01:00
|
|
|
Column(
|
|
|
|
"teamId",
|
|
|
|
Integer,
|
|
|
|
ForeignKey("chuni_profile_team.id", ondelete="SET NULL", onupdate="SET NULL"),
|
|
|
|
),
|
2023-02-17 07:02:21 +01:00
|
|
|
Column("avatarBack", Integer, server_default="0"),
|
|
|
|
Column("avatarFace", Integer, server_default="0"),
|
|
|
|
Column("eliteRankPoint", Integer, server_default="0"),
|
|
|
|
Column("stockedGridCount", Integer, server_default="0"),
|
|
|
|
Column("netBattleLoseCount", Integer, server_default="0"),
|
|
|
|
Column("netBattleHostErrCnt", Integer, server_default="0"),
|
|
|
|
Column("netBattle4thCount", Integer, server_default="0"),
|
|
|
|
Column("overPowerRate", Integer, server_default="0"),
|
|
|
|
Column("battleRewardStatus", Integer, server_default="0"),
|
|
|
|
Column("avatarPoint", Integer, server_default="0"),
|
|
|
|
Column("netBattle1stCount", Integer, server_default="0"),
|
|
|
|
Column("charaIllustId", Integer, server_default="0"),
|
|
|
|
Column("avatarItem", Integer, server_default="0"),
|
|
|
|
Column("userNameEx", String(8), server_default=""),
|
|
|
|
Column("netBattleWinCount", Integer, server_default="0"),
|
|
|
|
Column("netBattleCorrection", Integer, server_default="0"),
|
|
|
|
Column("classEmblemMedal", Integer, server_default="0"),
|
|
|
|
Column("overPowerPoint", Integer, server_default="0"),
|
|
|
|
Column("netBattleErrCnt", Integer, server_default="0"),
|
|
|
|
Column("battleRankId", Integer, server_default="0"),
|
|
|
|
Column("netBattle3rdCount", Integer, server_default="0"),
|
|
|
|
Column("netBattleConsecutiveWinCount", Integer, server_default="0"),
|
|
|
|
Column("overPowerLowerRank", Integer, server_default="0"),
|
|
|
|
Column("avatarWear", Integer, server_default="0"),
|
|
|
|
Column("classEmblemBase", Integer, server_default="0"),
|
|
|
|
Column("battleRankPoint", Integer, server_default="0"),
|
|
|
|
Column("netBattle2ndCount", Integer, server_default="0"),
|
|
|
|
Column("totalUltimaHighScore", Integer, server_default="0"),
|
|
|
|
Column("skillId", Integer, server_default="0"),
|
|
|
|
Column("lastCountryCode", String(5), server_default="JPN"),
|
|
|
|
Column("isNetBattleHost", Boolean, server_default="0"),
|
|
|
|
Column("avatarFront", Integer, server_default="0"),
|
|
|
|
Column("avatarSkin", Integer, server_default="0"),
|
|
|
|
Column("battleRewardCount", Integer, server_default="0"),
|
|
|
|
Column("battleRewardIndex", Integer, server_default="0"),
|
|
|
|
Column("netBattlePlayCount", Integer, server_default="0"),
|
|
|
|
Column("exMapLoopCount", Integer, server_default="0"),
|
|
|
|
Column("netBattleEndState", Integer, server_default="0"),
|
|
|
|
Column("avatarHead", Integer, server_default="0"),
|
|
|
|
UniqueConstraint("user", "version", name="chuni_profile_profile_uk"),
|
2023-03-09 17:38:58 +01:00
|
|
|
mysql_charset="utf8mb4",
|
2023-02-17 07:02:21 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
profile_ex = Table(
|
|
|
|
"chuni_profile_data_ex",
|
|
|
|
metadata,
|
|
|
|
Column("id", Integer, primary_key=True, nullable=False),
|
2023-03-09 17:38:58 +01:00
|
|
|
Column(
|
|
|
|
"user",
|
|
|
|
ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"),
|
|
|
|
nullable=False,
|
|
|
|
),
|
2023-02-17 07:02:21 +01:00
|
|
|
Column("version", Integer, nullable=False),
|
|
|
|
Column("ext1", Integer),
|
|
|
|
Column("ext2", Integer),
|
|
|
|
Column("ext3", Integer),
|
|
|
|
Column("ext4", Integer),
|
|
|
|
Column("ext5", Integer),
|
|
|
|
Column("ext6", Integer),
|
|
|
|
Column("ext7", Integer),
|
|
|
|
Column("ext8", Integer),
|
|
|
|
Column("ext9", Integer),
|
|
|
|
Column("ext10", Integer),
|
|
|
|
Column("ext11", Integer),
|
|
|
|
Column("ext12", Integer),
|
|
|
|
Column("ext13", Integer),
|
|
|
|
Column("ext14", Integer),
|
|
|
|
Column("ext15", Integer),
|
|
|
|
Column("ext16", Integer),
|
|
|
|
Column("ext17", Integer),
|
|
|
|
Column("ext18", Integer),
|
|
|
|
Column("ext19", Integer),
|
|
|
|
Column("ext20", Integer),
|
|
|
|
Column("medal", Integer),
|
|
|
|
Column("extStr1", String(255)),
|
|
|
|
Column("extStr2", String(255)),
|
|
|
|
Column("extStr3", String(255)),
|
|
|
|
Column("extStr4", String(255)),
|
|
|
|
Column("extStr5", String(255)),
|
|
|
|
Column("voiceId", Integer),
|
|
|
|
Column("extLong1", Integer),
|
|
|
|
Column("extLong2", Integer),
|
|
|
|
Column("extLong3", Integer),
|
|
|
|
Column("extLong4", Integer),
|
|
|
|
Column("extLong5", Integer),
|
|
|
|
Column("mapIconId", Integer),
|
|
|
|
Column("compatibleCmVersion", String(25)),
|
|
|
|
UniqueConstraint("user", "version", name="chuni_profile_data_ex_uk"),
|
2023-03-09 17:38:58 +01:00
|
|
|
mysql_charset="utf8mb4",
|
2023-02-17 07:02:21 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
option = Table(
|
|
|
|
"chuni_profile_option",
|
|
|
|
metadata,
|
|
|
|
Column("id", Integer, primary_key=True, nullable=False),
|
2023-03-09 17:38:58 +01:00
|
|
|
Column(
|
|
|
|
"user",
|
|
|
|
ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"),
|
|
|
|
nullable=False,
|
|
|
|
),
|
2023-02-17 07:02:21 +01:00
|
|
|
Column("speed", Integer),
|
|
|
|
Column("bgInfo", Integer),
|
|
|
|
Column("rating", Integer),
|
|
|
|
Column("privacy", Integer),
|
|
|
|
Column("judgePos", Integer),
|
|
|
|
Column("matching", Integer),
|
|
|
|
Column("guideLine", Integer),
|
|
|
|
Column("headphone", Integer),
|
|
|
|
Column("optionSet", Integer),
|
|
|
|
Column("fieldColor", Integer),
|
|
|
|
Column("guideSound", Integer),
|
|
|
|
Column("successAir", Integer),
|
|
|
|
Column("successTap", Integer),
|
|
|
|
Column("judgeAttack", Integer),
|
|
|
|
Column("playerLevel", Integer),
|
|
|
|
Column("soundEffect", Integer),
|
|
|
|
Column("judgeJustice", Integer),
|
|
|
|
Column("successExTap", Integer),
|
|
|
|
Column("successFlick", Integer),
|
|
|
|
Column("successSkill", Integer),
|
|
|
|
Column("successSlideHold", Integer),
|
|
|
|
Column("successTapTimbre", Integer),
|
2023-03-09 17:38:58 +01:00
|
|
|
Column("ext1", Integer), # Added in chunew
|
2023-02-17 07:02:21 +01:00
|
|
|
Column("ext2", Integer),
|
|
|
|
Column("ext3", Integer),
|
|
|
|
Column("ext4", Integer),
|
|
|
|
Column("ext5", Integer),
|
|
|
|
Column("ext6", Integer),
|
|
|
|
Column("ext7", Integer),
|
|
|
|
Column("ext8", Integer),
|
|
|
|
Column("ext9", Integer),
|
|
|
|
Column("ext10", Integer),
|
|
|
|
Column("categoryDetail", Integer, server_default="0"),
|
|
|
|
Column("judgeTimingOffset_120", Integer, server_default="0"),
|
|
|
|
Column("resultVoiceShort", Integer, server_default="0"),
|
|
|
|
Column("judgeAppendSe", Integer, server_default="0"),
|
|
|
|
Column("judgeCritical", Integer, server_default="0"),
|
|
|
|
Column("trackSkip", Integer, server_default="0"),
|
|
|
|
Column("selectMusicFilterLv", Integer, server_default="0"),
|
|
|
|
Column("sortMusicFilterLv", Integer, server_default="0"),
|
|
|
|
Column("sortMusicGenre", Integer, server_default="0"),
|
|
|
|
Column("speed_120", Integer, server_default="0"),
|
|
|
|
Column("judgeTimingOffset", Integer, server_default="0"),
|
|
|
|
Column("mirrorFumen", Integer, server_default="0"),
|
|
|
|
Column("playTimingOffset_120", Integer, server_default="0"),
|
|
|
|
Column("hardJudge", Integer, server_default="0"),
|
|
|
|
Column("notesThickness", Integer, server_default="0"),
|
|
|
|
Column("fieldWallPosition", Integer, server_default="0"),
|
|
|
|
Column("playTimingOffset", Integer, server_default="0"),
|
|
|
|
Column("fieldWallPosition_120", Integer, server_default="0"),
|
|
|
|
UniqueConstraint("user", name="chuni_profile_option_uk"),
|
2023-03-09 17:38:58 +01:00
|
|
|
mysql_charset="utf8mb4",
|
2023-02-17 07:02:21 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
option_ex = Table(
|
|
|
|
"chuni_profile_option_ex",
|
|
|
|
metadata,
|
|
|
|
Column("id", Integer, primary_key=True, nullable=False),
|
2023-03-09 17:38:58 +01:00
|
|
|
Column(
|
|
|
|
"user",
|
|
|
|
ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"),
|
|
|
|
nullable=False,
|
|
|
|
),
|
2023-02-17 07:02:21 +01:00
|
|
|
Column("ext1", Integer),
|
|
|
|
Column("ext2", Integer),
|
|
|
|
Column("ext3", Integer),
|
|
|
|
Column("ext4", Integer),
|
|
|
|
Column("ext5", Integer),
|
|
|
|
Column("ext6", Integer),
|
|
|
|
Column("ext7", Integer),
|
|
|
|
Column("ext8", Integer),
|
|
|
|
Column("ext9", Integer),
|
|
|
|
Column("ext10", Integer),
|
|
|
|
Column("ext11", Integer),
|
|
|
|
Column("ext12", Integer),
|
|
|
|
Column("ext13", Integer),
|
|
|
|
Column("ext14", Integer),
|
|
|
|
Column("ext15", Integer),
|
|
|
|
Column("ext16", Integer),
|
|
|
|
Column("ext17", Integer),
|
|
|
|
Column("ext18", Integer),
|
|
|
|
Column("ext19", Integer),
|
|
|
|
Column("ext20", Integer),
|
|
|
|
UniqueConstraint("user", name="chuni_profile_option_ex_uk"),
|
2023-03-09 17:38:58 +01:00
|
|
|
mysql_charset="utf8mb4",
|
2023-02-17 07:02:21 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
recent_rating = Table(
|
|
|
|
"chuni_profile_recent_rating",
|
|
|
|
metadata,
|
|
|
|
Column("id", Integer, primary_key=True, nullable=False),
|
2023-03-09 17:38:58 +01:00
|
|
|
Column(
|
|
|
|
"user",
|
|
|
|
ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"),
|
|
|
|
nullable=False,
|
|
|
|
),
|
2023-02-17 07:02:21 +01:00
|
|
|
Column("recentRating", JSON),
|
|
|
|
UniqueConstraint("user", name="chuni_profile_recent_rating_uk"),
|
2023-03-09 17:38:58 +01:00
|
|
|
mysql_charset="utf8mb4",
|
2023-02-17 07:02:21 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
region = Table(
|
|
|
|
"chuni_profile_region",
|
|
|
|
metadata,
|
|
|
|
Column("id", Integer, primary_key=True, nullable=False),
|
2023-03-09 17:38:58 +01:00
|
|
|
Column(
|
|
|
|
"user",
|
|
|
|
ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"),
|
|
|
|
nullable=False,
|
|
|
|
),
|
2023-02-17 07:02:21 +01:00
|
|
|
Column("regionId", Integer),
|
|
|
|
Column("playCount", Integer),
|
|
|
|
UniqueConstraint("user", "regionId", name="chuni_profile_region_uk"),
|
2023-03-09 17:38:58 +01:00
|
|
|
mysql_charset="utf8mb4",
|
2023-02-17 07:02:21 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
activity = Table(
|
|
|
|
"chuni_profile_activity",
|
|
|
|
metadata,
|
|
|
|
Column("id", Integer, primary_key=True, nullable=False),
|
2023-03-09 17:38:58 +01:00
|
|
|
Column(
|
|
|
|
"user",
|
|
|
|
ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"),
|
|
|
|
nullable=False,
|
|
|
|
),
|
2023-02-17 07:02:21 +01:00
|
|
|
Column("kind", Integer),
|
2023-03-09 17:38:58 +01:00
|
|
|
Column(
|
|
|
|
"activityId", Integer
|
|
|
|
), # Reminder: Change this to ID in base.py or the game will be sad
|
2023-02-17 07:02:21 +01:00
|
|
|
Column("sortNumber", Integer),
|
|
|
|
Column("param1", Integer),
|
|
|
|
Column("param2", Integer),
|
|
|
|
Column("param3", Integer),
|
|
|
|
Column("param4", Integer),
|
|
|
|
UniqueConstraint("user", "kind", "activityId", name="chuni_profile_activity_uk"),
|
2023-03-09 17:38:58 +01:00
|
|
|
mysql_charset="utf8mb4",
|
2023-02-17 07:02:21 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
charge = Table(
|
|
|
|
"chuni_profile_charge",
|
|
|
|
metadata,
|
|
|
|
Column("id", Integer, primary_key=True, nullable=False),
|
2023-03-09 17:38:58 +01:00
|
|
|
Column(
|
|
|
|
"user",
|
|
|
|
ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"),
|
|
|
|
nullable=False,
|
|
|
|
),
|
2023-02-17 07:02:21 +01:00
|
|
|
Column("chargeId", Integer),
|
|
|
|
Column("stock", Integer),
|
|
|
|
Column("purchaseDate", String(25)),
|
|
|
|
Column("validDate", String(25)),
|
|
|
|
Column("param1", Integer),
|
|
|
|
Column("param2", Integer),
|
|
|
|
Column("paramDate", String(25)),
|
|
|
|
UniqueConstraint("user", "chargeId", name="chuni_profile_charge_uk"),
|
2023-03-09 17:38:58 +01:00
|
|
|
mysql_charset="utf8mb4",
|
2023-02-17 07:02:21 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
emoney = Table(
|
|
|
|
"chuni_profile_emoney",
|
|
|
|
metadata,
|
|
|
|
Column("id", Integer, primary_key=True, nullable=False),
|
2023-03-09 17:38:58 +01:00
|
|
|
Column(
|
|
|
|
"user",
|
|
|
|
ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"),
|
|
|
|
nullable=False,
|
|
|
|
),
|
2023-02-17 07:02:21 +01:00
|
|
|
Column("ext1", Integer),
|
|
|
|
Column("ext2", Integer),
|
|
|
|
Column("ext3", Integer),
|
|
|
|
Column("type", Integer),
|
|
|
|
Column("emoneyBrand", Integer),
|
|
|
|
Column("emoneyCredit", Integer),
|
|
|
|
UniqueConstraint("user", "emoneyBrand", name="chuni_profile_emoney_uk"),
|
2023-03-09 17:38:58 +01:00
|
|
|
mysql_charset="utf8mb4",
|
2023-02-17 07:02:21 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
overpower = Table(
|
|
|
|
"chuni_profile_overpower",
|
|
|
|
metadata,
|
|
|
|
Column("id", Integer, primary_key=True, nullable=False),
|
2023-03-09 17:38:58 +01:00
|
|
|
Column(
|
|
|
|
"user",
|
|
|
|
ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"),
|
|
|
|
nullable=False,
|
|
|
|
),
|
2023-02-17 07:02:21 +01:00
|
|
|
Column("genreId", Integer),
|
|
|
|
Column("difficulty", Integer),
|
|
|
|
Column("rate", Integer),
|
|
|
|
Column("point", Integer),
|
|
|
|
UniqueConstraint("user", "genreId", "difficulty", name="chuni_profile_emoney_uk"),
|
2023-03-09 17:38:58 +01:00
|
|
|
mysql_charset="utf8mb4",
|
2023-02-17 07:02:21 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
team = Table(
|
|
|
|
"chuni_profile_team",
|
|
|
|
metadata,
|
|
|
|
Column("id", Integer, primary_key=True, nullable=False),
|
|
|
|
Column("teamName", String(255)),
|
|
|
|
Column("teamPoint", Integer),
|
2023-03-09 17:38:58 +01:00
|
|
|
mysql_charset="utf8mb4",
|
2023-02-17 07:02:21 +01:00
|
|
|
)
|
|
|
|
|
2023-03-09 17:38:58 +01:00
|
|
|
|
2023-02-17 07:02:21 +01:00
|
|
|
class ChuniProfileData(BaseData):
|
2023-03-09 17:38:58 +01:00
|
|
|
def put_profile_data(
|
|
|
|
self, aime_id: int, version: int, profile_data: Dict
|
|
|
|
) -> Optional[int]:
|
2023-02-17 07:02:21 +01:00
|
|
|
profile_data["user"] = aime_id
|
|
|
|
profile_data["version"] = version
|
|
|
|
if "accessCode" in profile_data:
|
|
|
|
profile_data.pop("accessCode")
|
2023-03-09 17:38:58 +01:00
|
|
|
|
2023-02-17 07:02:21 +01:00
|
|
|
profile_data = self.fix_bools(profile_data)
|
2023-03-09 17:38:58 +01:00
|
|
|
|
2023-02-17 07:02:21 +01:00
|
|
|
sql = insert(profile).values(**profile_data)
|
|
|
|
conflict = sql.on_duplicate_key_update(**profile_data)
|
|
|
|
result = self.execute(conflict)
|
|
|
|
|
|
|
|
if result is None:
|
|
|
|
self.logger.warn(f"put_profile_data: Failed to update! aime_id: {aime_id}")
|
|
|
|
return None
|
|
|
|
return result.lastrowid
|
2023-03-09 17:38:58 +01:00
|
|
|
|
2023-02-17 07:02:21 +01:00
|
|
|
def get_profile_preview(self, aime_id: int, version: int) -> Optional[Row]:
|
2023-03-09 17:38:58 +01:00
|
|
|
sql = (
|
|
|
|
select([profile, option])
|
|
|
|
.join(option, profile.c.user == option.c.user)
|
|
|
|
.filter(and_(profile.c.user == aime_id, profile.c.version == version))
|
2023-02-17 07:02:21 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
result = self.execute(sql)
|
2023-03-09 17:38:58 +01:00
|
|
|
if result is None:
|
|
|
|
return None
|
2023-02-17 07:02:21 +01:00
|
|
|
return result.fetchone()
|
|
|
|
|
|
|
|
def get_profile_data(self, aime_id: int, version: int) -> Optional[Row]:
|
2023-03-09 17:38:58 +01:00
|
|
|
sql = select(profile).where(
|
|
|
|
and_(
|
|
|
|
profile.c.user == aime_id,
|
|
|
|
profile.c.version == version,
|
|
|
|
)
|
|
|
|
)
|
2023-02-17 07:02:21 +01:00
|
|
|
|
|
|
|
result = self.execute(sql)
|
2023-03-09 17:38:58 +01:00
|
|
|
if result is None:
|
|
|
|
return None
|
2023-02-17 07:02:21 +01:00
|
|
|
return result.fetchone()
|
2023-03-09 17:38:58 +01:00
|
|
|
|
|
|
|
def put_profile_data_ex(
|
|
|
|
self, aime_id: int, version: int, profile_ex_data: Dict
|
|
|
|
) -> Optional[int]:
|
2023-02-17 07:02:21 +01:00
|
|
|
profile_ex_data["user"] = aime_id
|
|
|
|
profile_ex_data["version"] = version
|
|
|
|
if "accessCode" in profile_ex_data:
|
|
|
|
profile_ex_data.pop("accessCode")
|
2023-03-09 17:38:58 +01:00
|
|
|
|
2023-02-17 07:02:21 +01:00
|
|
|
sql = insert(profile_ex).values(**profile_ex_data)
|
|
|
|
conflict = sql.on_duplicate_key_update(**profile_ex_data)
|
|
|
|
result = self.execute(conflict)
|
|
|
|
|
|
|
|
if result is None:
|
2023-03-09 17:38:58 +01:00
|
|
|
self.logger.warn(
|
|
|
|
f"put_profile_data_ex: Failed to update! aime_id: {aime_id}"
|
|
|
|
)
|
2023-02-17 07:02:21 +01:00
|
|
|
return None
|
|
|
|
return result.lastrowid
|
2023-03-09 17:38:58 +01:00
|
|
|
|
2023-02-17 07:02:21 +01:00
|
|
|
def get_profile_data_ex(self, aime_id: int, version: int) -> Optional[Row]:
|
2023-03-09 17:38:58 +01:00
|
|
|
sql = select(profile_ex).where(
|
|
|
|
and_(
|
|
|
|
profile_ex.c.user == aime_id,
|
|
|
|
profile_ex.c.version == version,
|
|
|
|
)
|
|
|
|
)
|
2023-02-17 07:02:21 +01:00
|
|
|
|
|
|
|
result = self.execute(sql)
|
2023-03-09 17:38:58 +01:00
|
|
|
if result is None:
|
|
|
|
return None
|
2023-02-17 07:02:21 +01:00
|
|
|
return result.fetchone()
|
2023-03-09 17:38:58 +01:00
|
|
|
|
2023-02-17 07:02:21 +01:00
|
|
|
def put_profile_option(self, aime_id: int, option_data: Dict) -> Optional[int]:
|
|
|
|
option_data["user"] = aime_id
|
|
|
|
|
|
|
|
sql = insert(option).values(**option_data)
|
|
|
|
conflict = sql.on_duplicate_key_update(**option_data)
|
|
|
|
result = self.execute(conflict)
|
|
|
|
|
|
|
|
if result is None:
|
2023-03-09 17:38:58 +01:00
|
|
|
self.logger.warn(
|
|
|
|
f"put_profile_option: Failed to update! aime_id: {aime_id}"
|
|
|
|
)
|
2023-02-17 07:02:21 +01:00
|
|
|
return None
|
|
|
|
return result.lastrowid
|
|
|
|
|
|
|
|
def get_profile_option(self, aime_id: int) -> Optional[Row]:
|
|
|
|
sql = select(option).where(option.c.user == aime_id)
|
|
|
|
|
|
|
|
result = self.execute(sql)
|
2023-03-09 17:38:58 +01:00
|
|
|
if result is None:
|
|
|
|
return None
|
2023-02-17 07:02:21 +01:00
|
|
|
return result.fetchone()
|
2023-03-09 17:38:58 +01:00
|
|
|
|
|
|
|
def put_profile_option_ex(
|
|
|
|
self, aime_id: int, option_ex_data: Dict
|
|
|
|
) -> Optional[int]:
|
2023-02-17 07:02:21 +01:00
|
|
|
option_ex_data["user"] = aime_id
|
2023-03-09 17:38:58 +01:00
|
|
|
|
2023-02-17 07:02:21 +01:00
|
|
|
sql = insert(option_ex).values(**option_ex_data)
|
|
|
|
conflict = sql.on_duplicate_key_update(**option_ex_data)
|
|
|
|
result = self.execute(conflict)
|
|
|
|
|
|
|
|
if result is None:
|
2023-03-09 17:38:58 +01:00
|
|
|
self.logger.warn(
|
|
|
|
f"put_profile_option_ex: Failed to update! aime_id: {aime_id}"
|
|
|
|
)
|
2023-02-17 07:02:21 +01:00
|
|
|
return None
|
|
|
|
return result.lastrowid
|
|
|
|
|
|
|
|
def get_profile_option_ex(self, aime_id: int) -> Optional[Row]:
|
|
|
|
sql = select(option_ex).where(option_ex.c.user == aime_id)
|
|
|
|
|
|
|
|
result = self.execute(sql)
|
2023-03-09 17:38:58 +01:00
|
|
|
if result is None:
|
|
|
|
return None
|
2023-02-17 07:02:21 +01:00
|
|
|
return result.fetchone()
|
|
|
|
|
2023-03-09 17:38:58 +01:00
|
|
|
def put_profile_recent_rating(
|
|
|
|
self, aime_id: int, recent_rating_data: List[Dict]
|
|
|
|
) -> Optional[int]:
|
2023-02-17 07:02:21 +01:00
|
|
|
sql = insert(recent_rating).values(
|
2023-03-09 17:38:58 +01:00
|
|
|
user=aime_id, recentRating=recent_rating_data
|
2023-02-17 07:02:21 +01:00
|
|
|
)
|
2023-03-09 17:38:58 +01:00
|
|
|
conflict = sql.on_duplicate_key_update(recentRating=recent_rating_data)
|
2023-02-17 07:02:21 +01:00
|
|
|
|
|
|
|
result = self.execute(conflict)
|
|
|
|
if result is None:
|
2023-03-09 17:38:58 +01:00
|
|
|
self.logger.warn(
|
|
|
|
f"put_profile_recent_rating: Failed to update! aime_id: {aime_id}"
|
|
|
|
)
|
2023-02-17 07:02:21 +01:00
|
|
|
return None
|
|
|
|
return result.lastrowid
|
2023-03-09 17:38:58 +01:00
|
|
|
|
2023-02-17 07:02:21 +01:00
|
|
|
def get_profile_recent_rating(self, aime_id: int) -> Optional[Row]:
|
|
|
|
sql = select(recent_rating).where(recent_rating.c.user == aime_id)
|
|
|
|
|
|
|
|
result = self.execute(sql)
|
2023-03-09 17:38:58 +01:00
|
|
|
if result is None:
|
|
|
|
return None
|
2023-02-17 07:02:21 +01:00
|
|
|
return result.fetchone()
|
|
|
|
|
|
|
|
def put_profile_activity(self, aime_id: int, activity_data: Dict) -> Optional[int]:
|
|
|
|
# The game just uses "id" but we need to distinguish that from the db column "id"
|
|
|
|
activity_data["user"] = aime_id
|
|
|
|
activity_data["activityId"] = activity_data["id"]
|
|
|
|
activity_data.pop("id")
|
2023-03-09 17:38:58 +01:00
|
|
|
|
2023-02-17 07:02:21 +01:00
|
|
|
sql = insert(activity).values(**activity_data)
|
|
|
|
conflict = sql.on_duplicate_key_update(**activity_data)
|
|
|
|
result = self.execute(conflict)
|
|
|
|
|
|
|
|
if result is None:
|
2023-03-09 17:38:58 +01:00
|
|
|
self.logger.warn(
|
|
|
|
f"put_profile_activity: Failed to update! aime_id: {aime_id}"
|
|
|
|
)
|
2023-02-17 07:02:21 +01:00
|
|
|
return None
|
|
|
|
return result.lastrowid
|
|
|
|
|
|
|
|
def get_profile_activity(self, aime_id: int, kind: int) -> Optional[List[Row]]:
|
2023-03-09 17:38:58 +01:00
|
|
|
sql = select(activity).where(
|
|
|
|
and_(activity.c.user == aime_id, activity.c.kind == kind)
|
|
|
|
)
|
2023-02-17 07:02:21 +01:00
|
|
|
|
|
|
|
result = self.execute(sql)
|
2023-03-09 17:38:58 +01:00
|
|
|
if result is None:
|
|
|
|
return None
|
2023-02-17 07:02:21 +01:00
|
|
|
return result.fetchall()
|
|
|
|
|
|
|
|
def put_profile_charge(self, aime_id: int, charge_data: Dict) -> Optional[int]:
|
|
|
|
charge_data["user"] = aime_id
|
2023-03-09 17:38:58 +01:00
|
|
|
|
2023-02-17 07:02:21 +01:00
|
|
|
sql = insert(charge).values(**charge_data)
|
|
|
|
conflict = sql.on_duplicate_key_update(**charge_data)
|
|
|
|
result = self.execute(conflict)
|
|
|
|
|
|
|
|
if result is None:
|
2023-03-09 17:38:58 +01:00
|
|
|
self.logger.warn(
|
|
|
|
f"put_profile_charge: Failed to update! aime_id: {aime_id}"
|
|
|
|
)
|
2023-02-17 07:02:21 +01:00
|
|
|
return None
|
|
|
|
return result.lastrowid
|
|
|
|
|
|
|
|
def get_profile_charge(self, aime_id: int) -> Optional[List[Row]]:
|
|
|
|
sql = select(charge).where(charge.c.user == aime_id)
|
|
|
|
|
|
|
|
result = self.execute(sql)
|
2023-03-09 17:38:58 +01:00
|
|
|
if result is None:
|
|
|
|
return None
|
2023-02-17 07:02:21 +01:00
|
|
|
return result.fetchall()
|
2023-03-09 17:38:58 +01:00
|
|
|
|
2023-02-17 07:02:21 +01:00
|
|
|
def add_profile_region(self, aime_id: int, region_id: int) -> Optional[int]:
|
|
|
|
pass
|
|
|
|
|
|
|
|
def get_profile_regions(self, aime_id: int) -> Optional[List[Row]]:
|
|
|
|
pass
|
|
|
|
|
|
|
|
def put_profile_emoney(self, aime_id: int, emoney_data: Dict) -> Optional[int]:
|
|
|
|
emoney_data["user"] = aime_id
|
|
|
|
|
|
|
|
sql = insert(emoney).values(**emoney_data)
|
|
|
|
conflict = sql.on_duplicate_key_update(**emoney_data)
|
|
|
|
|
|
|
|
result = self.execute(conflict)
|
2023-03-09 17:38:58 +01:00
|
|
|
if result is None:
|
|
|
|
return None
|
2023-02-17 07:02:21 +01:00
|
|
|
return result.lastrowid
|
2023-03-09 17:38:58 +01:00
|
|
|
|
2023-02-17 07:02:21 +01:00
|
|
|
def get_profile_emoney(self, aime_id: int) -> Optional[List[Row]]:
|
|
|
|
sql = select(emoney).where(emoney.c.user == aime_id)
|
|
|
|
|
|
|
|
result = self.execute(sql)
|
2023-03-09 17:38:58 +01:00
|
|
|
if result is None:
|
|
|
|
return None
|
2023-02-17 07:02:21 +01:00
|
|
|
return result.fetchall()
|
|
|
|
|
2023-03-09 17:38:58 +01:00
|
|
|
def put_profile_overpower(
|
|
|
|
self, aime_id: int, overpower_data: Dict
|
|
|
|
) -> Optional[int]:
|
2023-02-17 07:02:21 +01:00
|
|
|
overpower_data["user"] = aime_id
|
|
|
|
|
|
|
|
sql = insert(overpower).values(**overpower_data)
|
|
|
|
conflict = sql.on_duplicate_key_update(**overpower_data)
|
|
|
|
|
|
|
|
result = self.execute(conflict)
|
2023-03-09 17:38:58 +01:00
|
|
|
if result is None:
|
|
|
|
return None
|
2023-02-17 07:02:21 +01:00
|
|
|
return result.lastrowid
|
2023-03-09 17:38:58 +01:00
|
|
|
|
2023-02-17 07:02:21 +01:00
|
|
|
def get_profile_overpower(self, aime_id: int) -> Optional[List[Row]]:
|
|
|
|
sql = select(overpower).where(overpower.c.user == aime_id)
|
|
|
|
|
|
|
|
result = self.execute(sql)
|
2023-03-09 17:38:58 +01:00
|
|
|
if result is None:
|
|
|
|
return None
|
2023-02-17 07:02:21 +01:00
|
|
|
return result.fetchall()
|