add back games, conform them to new title dispatch

titles/mai2/schema/profile.py

@@ -0,0 +1,402 @@
+from core.data.schema import BaseData, metadata
+from titles.mai2.const import Mai2Constants
+
+from typing import Optional, Dict, List
+from sqlalchemy import Table, Column, UniqueConstraint, PrimaryKeyConstraint, and_
+from sqlalchemy.types import Integer, String, TIMESTAMP, Boolean, JSON, BigInteger
+from sqlalchemy.schema import ForeignKey
+from sqlalchemy.sql import func, select
+from sqlalchemy.engine import Row
+from sqlalchemy.dialects.mysql import insert
+from datetime import datetime
+
+detail = Table(
+    "mai2_profile_detail",
+    metadata,
+    Column("id", Integer, primary_key=True, nullable=False),
+    Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
+    Column("version", Integer, nullable=False),
+    Column("userName", String(25)),
+    Column("isNetMember", Integer),
+    Column("iconId", Integer),
+    Column("plateId", Integer),
+    Column("titleId", Integer),
+    Column("partnerId", Integer),
+    Column("frameId", Integer),
+    Column("selectMapId", Integer),
+    Column("totalAwake", Integer),
+    Column("gradeRating", Integer),
+    Column("musicRating", Integer),
+    Column("playerRating", Integer),
+    Column("highestRating", Integer),
+    Column("gradeRank", Integer),
+    Column("classRank", Integer),
+    Column("courseRank", Integer),
+    Column("charaSlot", JSON),
+    Column("charaLockSlot", JSON),
+    Column("contentBit", BigInteger),
+    Column("playCount", Integer),
+    Column("eventWatchedDate", String(25)),
+    Column("lastGameId", String(25)),
+    Column("lastRomVersion", String(25)),
+    Column("lastDataVersion", String(25)),
+    Column("lastLoginDate", String(25)),
+    Column("lastPairLoginDate", String(25)), # new with uni+
+    Column("lastPlayDate", String(25)),
+    Column("lastTrialPlayDate", String(25)), # new with uni+
+    Column("lastPlayCredit", Integer),
+    Column("lastPlayMode", Integer),
+    Column("lastPlaceId", Integer),
+    Column("lastPlaceName", String(25)),
+    Column("lastAllNetId", Integer),
+    Column("lastRegionId", Integer),
+    Column("lastRegionName", String(25)),
+    Column("lastClientId", String(25)),
+    Column("lastCountryCode", String(25)),
+    Column("lastSelectEMoney", Integer),
+    Column("lastSelectTicket", Integer),
+    Column("lastSelectCourse", Integer),
+    Column("lastCountCourse", Integer),
+    Column("firstGameId", String(25)),
+    Column("firstRomVersion", String(25)),
+    Column("firstDataVersion", String(25)),
+    Column("firstPlayDate", String(25)),
+    Column("compatibleCmVersion", String(25)),
+    Column("dailyBonusDate", String(25)),
+    Column("dailyCourseBonusDate", String(25)),
+    Column("playVsCount", Integer),
+    Column("playSyncCount", Integer),
+    Column("winCount", Integer),
+    Column("helpCount", Integer),
+    Column("comboCount", Integer),
+    Column("totalDeluxscore", BigInteger),
+    Column("totalBasicDeluxscore", BigInteger),
+    Column("totalAdvancedDeluxscore", BigInteger),
+    Column("totalExpertDeluxscore", BigInteger),
+    Column("totalMasterDeluxscore", BigInteger),
+    Column("totalReMasterDeluxscore", BigInteger),
+    Column("totalSync", Integer),
+    Column("totalBasicSync", Integer),
+    Column("totalAdvancedSync", Integer),
+    Column("totalExpertSync", Integer),
+    Column("totalMasterSync", Integer),
+    Column("totalReMasterSync", Integer),
+    Column("totalAchievement", BigInteger),
+    Column("totalBasicAchievement", BigInteger),
+    Column("totalAdvancedAchievement", BigInteger),
+    Column("totalExpertAchievement", BigInteger),
+    Column("totalMasterAchievement", BigInteger),
+    Column("totalReMasterAchievement", BigInteger),
+    Column("playerOldRating", BigInteger),
+    Column("playerNewRating", BigInteger),
+    Column("dateTime", BigInteger),
+    Column("banState", Integer), # new with uni+
+    UniqueConstraint("user", "version", name="mai2_profile_detail_uk"),
+    mysql_charset='utf8mb4'
+)
+
+ghost = Table(
+    "mai2_profile_ghost",
+    metadata,
+    Column("id", Integer, primary_key=True, nullable=False),
+    Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
+    Column("version_int", Integer, nullable=False),
+    Column("name", String(25)),
+    Column("iconId", Integer),
+    Column("plateId", Integer),
+    Column("titleId", Integer),
+    Column("rate", Integer),
+    Column("udemaeRate", Integer),
+    Column("courseRank", Integer),
+    Column("classRank", Integer),
+    Column("classValue", Integer),
+    Column("playDatetime", String(25)),
+    Column("shopId", Integer),
+    Column("regionCode", Integer),
+    Column("typeId", Integer),
+    Column("musicId", Integer),
+    Column("difficulty", Integer),
+    Column("version", Integer),
+    Column("resultBitList", JSON),
+    Column("resultNum", Integer),
+    Column("achievement", Integer),
+    UniqueConstraint("user", "version", "musicId", "difficulty", name="mai2_profile_ghost_uk"),
+    mysql_charset='utf8mb4'
+)
+
+extend = Table(
+    "mai2_profile_extend",
+    metadata,
+    Column("id", Integer, primary_key=True, nullable=False),
+    Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
+    Column("version", Integer, nullable=False),
+    Column("selectMusicId", Integer),
+    Column("selectDifficultyId", Integer),
+    Column("categoryIndex", Integer),
+    Column("musicIndex", Integer),
+    Column("extraFlag", Integer),
+    Column("selectScoreType", Integer),
+    Column("extendContentBit", BigInteger),
+    Column("isPhotoAgree", Boolean),
+    Column("isGotoCodeRead", Boolean),
+    Column("selectResultDetails", Boolean),
+    Column("sortCategorySetting", Integer),
+    Column("sortMusicSetting", Integer),
+    Column("selectedCardList", JSON),
+    Column("encountMapNpcList", JSON),
+    UniqueConstraint("user", "version", name="mai2_profile_extend_uk"),
+    mysql_charset='utf8mb4'
+)
+
+option = Table(
+    "mai2_profile_option",
+    metadata,
+    Column("id", Integer, primary_key=True, nullable=False),
+    Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
+    Column("version", Integer, nullable=False),
+    Column("selectMusicId", Integer),
+    Column("optionKind", Integer),
+    Column("noteSpeed", Integer),
+    Column("slideSpeed", Integer),
+    Column("touchSpeed", Integer),
+    Column("tapDesign", Integer),
+    Column("holdDesign", Integer),
+    Column("slideDesign", Integer),
+    Column("starType", Integer),
+    Column("outlineDesign", Integer),
+    Column("noteSize", Integer),
+    Column("slideSize", Integer),
+    Column("touchSize", Integer),
+    Column("starRotate", Integer),
+    Column("dispCenter", Integer),
+    Column("dispChain", Integer),
+    Column("dispRate", Integer),
+    Column("dispBar", Integer),
+    Column("touchEffect", Integer),
+    Column("submonitorAnimation", Integer),
+    Column("submonitorAchive", Integer),
+    Column("submonitorAppeal", Integer),
+    Column("matching", Integer),
+    Column("trackSkip", Integer),
+    Column("brightness", Integer),
+    Column("mirrorMode", Integer),
+    Column("dispJudge", Integer),
+    Column("dispJudgePos", Integer),
+    Column("dispJudgeTouchPos", Integer),
+    Column("adjustTiming", Integer),
+    Column("judgeTiming", Integer),
+    Column("ansVolume", Integer),
+    Column("tapHoldVolume", Integer),
+    Column("criticalSe", Integer),
+    Column("breakSe", Integer),
+    Column("breakVolume", Integer),
+    Column("exSe", Integer),
+    Column("exVolume", Integer),
+    Column("slideSe", Integer),
+    Column("slideVolume", Integer),
+    Column("touchHoldVolume", Integer),
+    Column("damageSeVolume", Integer),
+    Column("headPhoneVolume", Integer),
+    Column("sortTab", Integer),
+    Column("sortMusic", Integer),
+    UniqueConstraint("user", "version", name="mai2_profile_option_uk"),
+    mysql_charset='utf8mb4'
+)
+
+rating = Table(
+    "mai2_profile_rating",
+    metadata,
+    Column("id", Integer, primary_key=True, nullable=False),
+    Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
+    Column("version", Integer, nullable=False),
+    Column("rating", Integer),
+    Column("ratingList", JSON),
+    Column("newRatingList", JSON),
+    Column("nextRatingList", JSON),
+    Column("nextNewRatingList", JSON),
+    Column("udemae", JSON),
+    UniqueConstraint("user", "version", name="mai2_profile_rating_uk"),
+    mysql_charset='utf8mb4'
+)
+
+region = Table(
+    "mai2_profile_region",
+    metadata,
+    Column("id", Integer, primary_key=True, nullable=False),
+    Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
+    Column("regionId", Integer),
+    Column("playCount", Integer, server_default="1"),
+    Column("created", String(25)),
+    UniqueConstraint("user", "regionId", name="mai2_profile_region_uk"),
+    mysql_charset='utf8mb4'
+)
+
+activity = Table(
+    "mai2_profile_activity",
+    metadata,
+    Column("id", Integer, primary_key=True, nullable=False),
+    Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
+    Column("kind", Integer, nullable=False),
+    Column("activityId", Integer, nullable=False),
+    Column("param1", Integer, nullable=False),
+    Column("param2", Integer, nullable=False),
+    Column("param3", Integer, nullable=False),
+    Column("param4", Integer, nullable=False),
+    Column("sortNumber", Integer, nullable=False),
+    UniqueConstraint("user", "kind", "activityId", name="mai2_profile_activity_uk"),
+    mysql_charset='utf8mb4'
+)
+
+class Mai2ProfileData(BaseData):
+    def put_profile_detail(self, user_id: int, version: int, detail_data: Dict) -> Optional[Row]:
+        detail_data["user"] = user_id
+        detail_data["version"] = version
+        sql = insert(detail).values(**detail_data)
+
+        conflict = sql.on_duplicate_key_update(**detail_data)
+
+        result = self.execute(conflict)
+        if result is None:
+            self.logger.warn(f"put_profile: Failed to create profile! user_id {user_id}")
+            return None
+        return result.lastrowid
+
+    def get_profile_detail(self, user_id: int, version: int) -> Optional[Row]:
+        sql = select(detail).where(and_(detail.c.user == user_id, detail.c.version == version))
+
+        result = self.execute(sql)
+        if result is None:return None
+        return result.fetchone()
+
+    def put_profile_ghost(self, user_id: int, version: int, ghost_data: Dict) -> Optional[int]:
+        ghost_data["user"] = user_id
+        ghost_data["version_int"] = version
+
+        sql = insert(ghost).values(**ghost_data)
+        conflict = sql.on_duplicate_key_update(**ghost_data)
+
+        result = self.execute(conflict)
+        if result is None:
+            self.logger.warn(f"put_profile_ghost: failed to update! {user_id}")
+            return None
+        return result.lastrowid
+
+    def get_profile_ghost(self, user_id: int, version: int) -> Optional[Row]:
+        sql = select(ghost).where(and_(ghost.c.user == user_id, ghost.c.version_int == version))
+
+        result = self.execute(sql)
+        if result is None:return None
+        return result.fetchone()
+
+    def put_profile_extend(self, user_id: int, version: int, extend_data: Dict) -> Optional[int]:
+        extend_data["user"] = user_id
+        extend_data["version"] = version
+
+        sql = insert(extend).values(**extend_data)
+        conflict = sql.on_duplicate_key_update(**extend_data)
+
+        result = self.execute(conflict)
+        if result is None:
+            self.logger.warn(f"put_profile_extend: failed to update! {user_id}")
+            return None
+        return result.lastrowid
+
+    def get_profile_extend(self, user_id: int, version: int) -> Optional[Row]:
+        sql = select(extend).where(and_(extend.c.user == user_id, extend.c.version == version))
+
+        result = self.execute(sql)
+        if result is None:return None
+        return result.fetchone()
+
+    def put_profile_option(self, user_id: int, version: int, option_data: Dict) -> Optional[int]:
+        option_data["user"] = user_id
+        option_data["version"] = version
+
+        sql = insert(option).values(**option_data)
+        conflict = sql.on_duplicate_key_update(**option_data)
+
+        result = self.execute(conflict)
+        if result is None:
+            self.logger.warn(f"put_profile_option: failed to update! {user_id}")
+            return None
+        return result.lastrowid
+
+    def get_profile_option(self, user_id: int, version: int) -> Optional[Row]:
+        sql = select(option).where(and_(option.c.user == user_id, option.c.version == version))
+
+        result = self.execute(sql)
+        if result is None:return None
+        return result.fetchone()
+    
+    def put_profile_rating(self, user_id: int, version: int, rating_data: Dict) -> Optional[int]:
+        rating_data["user"] = user_id
+        rating_data["version"] = version
+
+        sql = insert(rating).values(**rating_data)
+        conflict = sql.on_duplicate_key_update(**rating_data)
+
+        result = self.execute(conflict)
+        if result is None:
+            self.logger.warn(f"put_profile_rating: failed to update! {user_id}")
+            return None
+        return result.lastrowid
+
+    def get_profile_rating(self, user_id: int, version: int) -> Optional[Row]:
+        sql = select(rating).where(and_(rating.c.user == user_id, rating.c.version == version))
+
+        result = self.execute(sql)
+        if result is None:return None
+        return result.fetchone()
+
+    def put_profile_region(self, user_id: int, region_id: int) -> Optional[int]:
+        sql = insert(region).values(
+            user = user_id,
+            regionId = region_id,
+            created = datetime.strftime(datetime.now(), Mai2Constants.DATE_TIME_FORMAT)
+        )
+        
+        conflict = sql.on_duplicate_key_update(
+            playCount = region.c.playCount + 1
+        )
+
+        result = self.execute(conflict)
+        if result is None:
+            self.logger.warn(f"put_region: failed to update! {user_id}")
+            return None
+        return result.lastrowid
+
+    def get_regions(self, user_id: int) -> Optional[List[Dict]]:
+        sql = select(region).where(region.c.user == user_id)
+
+        result = self.execute(sql)
+        if result is None:return None
+        return result.fetchall()
+    
+    def put_profile_activity(self, user_id: int, activity_data: Dict) -> Optional[int]:
+        if "id" in activity_data:
+            activity_data["activityId"] = activity_data["id"]
+            activity_data.pop("id")
+        
+        activity_data["user"] = user_id
+        
+        sql = insert(activity).values(**activity_data)
+
+        conflict = sql.on_duplicate_key_update(**activity_data)
+
+        result = self.execute(conflict)
+        if result is None:
+            self.logger.warn(f"put_profile_activity: failed to update! user_id: {user_id}")
+            return None
+        return result.lastrowid
+
+    def get_profile_activity(self, user_id: int, kind: int = None) -> Optional[Row]:
+        sql = activity.select(
+            and_(
+                activity.c.user == user_id, 
+                (activity.c.kind == kind) if kind is not None else True,
+            )
+        )
+
+        result = self.execute(sql)
+        if result is None:return None
+        return result.fetchone()