441 lines
17 KiB
Python
441 lines
17 KiB
Python
|
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.sql import func, select
|
||
|
|
from sqlalchemy.engine import Row
|
||
|
|
from sqlalchemy.dialects.mysql import insert
|
||
|
|
|
||
|
|
from core.data.schema import BaseData, metadata
|
||
|
|
from core.config import CoreConfig
|
||
|
|
|
||
|
|
profile = Table(
|
||
|
|
"idac_profile",
|
||
|
|
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(8)),
|
||
|
|
Column("country", Integer),
|
||
|
|
Column("store", Integer),
|
||
|
|
Column("team_id", Integer, server_default="0"),
|
||
|
|
Column("total_play", Integer, server_default="0"),
|
||
|
|
Column("daily_play", Integer, server_default="0"),
|
||
|
|
Column("day_play", Integer, server_default="0"),
|
||
|
|
Column("mileage", Integer, server_default="0"),
|
||
|
|
Column("asset_version", Integer, server_default="1"),
|
||
|
|
Column("last_play_date", TIMESTAMP, server_default=func.now()),
|
||
|
|
Column("mytitle_id", Integer, server_default="0"),
|
||
|
|
Column("mytitle_efffect_id", Integer, server_default="0"),
|
||
|
|
Column("sticker_id", Integer, server_default="0"),
|
||
|
|
Column("sticker_effect_id", Integer, server_default="0"),
|
||
|
|
Column("papercup_id", Integer, server_default="0"),
|
||
|
|
Column("tachometer_id", Integer, server_default="0"),
|
||
|
|
Column("aura_id", Integer, server_default="0"),
|
||
|
|
Column("aura_color_id", Integer, server_default="0"),
|
||
|
|
Column("aura_line_id", Integer, server_default="0"),
|
||
|
|
Column("bgm_id", Integer, server_default="0"),
|
||
|
|
Column("keyholder_id", Integer, server_default="0"),
|
||
|
|
Column("start_menu_bg_id", Integer, server_default="0"),
|
||
|
|
Column("use_car_id", Integer, server_default="1"),
|
||
|
|
Column("use_style_car_id", Integer, server_default="1"),
|
||
|
|
Column("bothwin_count", Integer, server_default="0"),
|
||
|
|
Column("bothwin_score", Integer, server_default="0"),
|
||
|
|
Column("subcard_count", Integer, server_default="0"),
|
||
|
|
Column("vs_history", Integer, server_default="0"),
|
||
|
|
Column("stamp_key_assign_0", Integer),
|
||
|
|
Column("stamp_key_assign_1", Integer),
|
||
|
|
Column("stamp_key_assign_2", Integer),
|
||
|
|
Column("stamp_key_assign_3", Integer),
|
||
|
|
Column("name_change_category", Integer, server_default="0"),
|
||
|
|
Column("factory_disp", Integer, server_default="0"),
|
||
|
|
Column("create_date", TIMESTAMP, server_default=func.now()),
|
||
|
|
Column("cash", Integer, server_default="0"),
|
||
|
|
Column("dressup_point", Integer, server_default="0"),
|
||
|
|
Column("avatar_point", Integer, server_default="0"),
|
||
|
|
Column("total_cash", Integer, server_default="0"),
|
||
|
|
UniqueConstraint("user", "version", name="idac_profile_uk"),
|
||
|
|
mysql_charset="utf8mb4",
|
||
|
|
)
|
||
|
|
|
||
|
|
# No point setting defaults since the game sends everything on profile creation anyway
|
||
|
|
config = Table(
|
||
|
|
"idac_profile_config",
|
||
|
|
metadata,
|
||
|
|
Column("id", Integer, primary_key=True, nullable=False),
|
||
|
|
Column(
|
||
|
|
"user",
|
||
|
|
ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"),
|
||
|
|
nullable=False,
|
||
|
|
),
|
||
|
|
Column("config_id", Integer),
|
||
|
|
Column("steering_intensity", Integer),
|
||
|
|
Column("transmission_type", Integer),
|
||
|
|
Column("default_viewpoint", Integer),
|
||
|
|
Column("favorite_bgm", Integer),
|
||
|
|
Column("bgm_volume", Integer),
|
||
|
|
Column("se_volume", Integer),
|
||
|
|
Column("master_volume", Integer),
|
||
|
|
Column("store_battle_policy", Integer),
|
||
|
|
Column("battle_onomatope_display", Integer),
|
||
|
|
Column("cornering_guide", Integer),
|
||
|
|
Column("minimap", Integer),
|
||
|
|
Column("line_guide", Integer),
|
||
|
|
Column("ghost", Integer),
|
||
|
|
Column("race_exit", Integer),
|
||
|
|
Column("result_skip", Integer),
|
||
|
|
Column("stamp_select_skip", Integer),
|
||
|
|
UniqueConstraint("user", name="idac_profile_config_uk"),
|
||
|
|
mysql_charset="utf8mb4",
|
||
|
|
)
|
||
|
|
|
||
|
|
# No point setting defaults since the game sends everything on profile creation anyway
|
||
|
|
avatar = Table(
|
||
|
|
"idac_profile_avatar",
|
||
|
|
metadata,
|
||
|
|
Column("id", Integer, primary_key=True, nullable=False),
|
||
|
|
Column(
|
||
|
|
"user",
|
||
|
|
ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"),
|
||
|
|
nullable=False,
|
||
|
|
),
|
||
|
|
Column("sex", Integer),
|
||
|
|
Column("face", Integer),
|
||
|
|
Column("eye", Integer),
|
||
|
|
Column("mouth", Integer),
|
||
|
|
Column("hair", Integer),
|
||
|
|
Column("glasses", Integer),
|
||
|
|
Column("face_accessory", Integer),
|
||
|
|
Column("body", Integer),
|
||
|
|
Column("body_accessory", Integer),
|
||
|
|
Column("behind", Integer),
|
||
|
|
Column("bg", Integer),
|
||
|
|
Column("effect", Integer),
|
||
|
|
Column("special", Integer),
|
||
|
|
UniqueConstraint("user", name="idac_profile_avatar_uk"),
|
||
|
|
mysql_charset="utf8mb4",
|
||
|
|
)
|
||
|
|
|
||
|
|
rank = Table(
|
||
|
|
"idac_profile_rank",
|
||
|
|
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("story_rank_exp", Integer, server_default="0"),
|
||
|
|
Column("story_rank", Integer, server_default="1"),
|
||
|
|
Column("time_trial_rank_exp", Integer, server_default="0"),
|
||
|
|
Column("time_trial_rank", Integer, server_default="1"),
|
||
|
|
Column("online_battle_rank_exp", Integer, server_default="0"),
|
||
|
|
Column("online_battle_rank", Integer, server_default="1"),
|
||
|
|
Column("store_battle_rank_exp", Integer, server_default="0"),
|
||
|
|
Column("store_battle_rank", Integer, server_default="1"),
|
||
|
|
Column("theory_exp", Integer, server_default="0"),
|
||
|
|
Column("theory_rank", Integer, server_default="1"),
|
||
|
|
Column("pride_group_id", Integer, server_default="0"),
|
||
|
|
Column("pride_point", Integer, server_default="0"),
|
||
|
|
Column("grade_exp", Integer, server_default="0"),
|
||
|
|
Column("grade", Integer, server_default="1"),
|
||
|
|
Column("grade_reward_dist", Integer, server_default="0"),
|
||
|
|
Column("story_rank_reward_dist", Integer, server_default="0"),
|
||
|
|
Column("time_trial_rank_reward_dist", Integer, server_default="0"),
|
||
|
|
Column("online_battle_rank_reward_dist", Integer, server_default="0"),
|
||
|
|
Column("store_battle_rank_reward_dist", Integer, server_default="0"),
|
||
|
|
Column("theory_rank_reward_dist", Integer, server_default="0"),
|
||
|
|
Column("max_attained_online_battle_rank", Integer, server_default="1"),
|
||
|
|
Column("max_attained_pride_point", Integer, server_default="0"),
|
||
|
|
Column("is_last_max", Integer, server_default="0"),
|
||
|
|
UniqueConstraint("user", "version", name="idac_profile_rank_uk"),
|
||
|
|
mysql_charset="utf8mb4",
|
||
|
|
)
|
||
|
|
|
||
|
|
stock = Table(
|
||
|
|
"idac_profile_stock",
|
||
|
|
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("mytitle_list", String(1024), server_default=""),
|
||
|
|
Column("mytitle_new_list", String(1024), server_default=""),
|
||
|
|
Column("avatar_face_list", String(255), server_default=""),
|
||
|
|
Column("avatar_face_new_list", String(255), server_default=""),
|
||
|
|
Column("avatar_eye_list", String(255), server_default=""),
|
||
|
|
Column("avatar_eye_new_list", String(255), server_default=""),
|
||
|
|
Column("avatar_hair_list", String(255), server_default=""),
|
||
|
|
Column("avatar_hair_new_list", String(255), server_default=""),
|
||
|
|
Column("avatar_body_list", String(255), server_default=""),
|
||
|
|
Column("avatar_body_new_list", String(255), server_default=""),
|
||
|
|
Column("avatar_mouth_list", String(255), server_default=""),
|
||
|
|
Column("avatar_mouth_new_list", String(255), server_default=""),
|
||
|
|
Column("avatar_glasses_list", String(255), server_default=""),
|
||
|
|
Column("avatar_glasses_new_list", String(255), server_default=""),
|
||
|
|
Column("avatar_face_accessory_list", String(255), server_default=""),
|
||
|
|
Column("avatar_face_accessory_new_list", String(255), server_default=""),
|
||
|
|
Column("avatar_body_accessory_list", String(255), server_default=""),
|
||
|
|
Column("avatar_body_accessory_new_list", String(255), server_default=""),
|
||
|
|
Column("avatar_behind_list", String(255), server_default=""),
|
||
|
|
Column("avatar_behind_new_list", String(255), server_default=""),
|
||
|
|
Column("avatar_bg_list", String(255), server_default=""),
|
||
|
|
Column("avatar_bg_new_list", String(255), server_default=""),
|
||
|
|
Column("avatar_effect_list", String(255), server_default=""),
|
||
|
|
Column("avatar_effect_new_list", String(255), server_default=""),
|
||
|
|
Column("avatar_special_list", String(255), server_default=""),
|
||
|
|
Column("avatar_special_new_list", String(255), server_default=""),
|
||
|
|
Column("stamp_list", String(255), server_default=""),
|
||
|
|
Column("stamp_new_list", String(255), server_default=""),
|
||
|
|
Column("keyholder_list", String(256), server_default=""),
|
||
|
|
Column("keyholder_new_list", String(256), server_default=""),
|
||
|
|
Column("papercup_list", String(255), server_default=""),
|
||
|
|
Column("papercup_new_list", String(255), server_default=""),
|
||
|
|
Column("tachometer_list", String(255), server_default=""),
|
||
|
|
Column("tachometer_new_list", String(255), server_default=""),
|
||
|
|
Column("aura_list", String(255), server_default=""),
|
||
|
|
Column("aura_new_list", String(255), server_default=""),
|
||
|
|
Column("aura_color_list", String(255), server_default=""),
|
||
|
|
Column("aura_color_new_list", String(255), server_default=""),
|
||
|
|
Column("aura_line_list", String(255), server_default=""),
|
||
|
|
Column("aura_line_new_list", String(255), server_default=""),
|
||
|
|
Column("bgm_list", String(255), server_default=""),
|
||
|
|
Column("bgm_new_list", String(255), server_default=""),
|
||
|
|
Column("dx_color_list", String(255), server_default=""),
|
||
|
|
Column("dx_color_new_list", String(255), server_default=""),
|
||
|
|
Column("start_menu_bg_list", String(255), server_default=""),
|
||
|
|
Column("start_menu_bg_new_list", String(255), server_default=""),
|
||
|
|
Column("under_neon_list", String(255), server_default=""),
|
||
|
|
UniqueConstraint("user", "version", name="idac_profile_stock_uk"),
|
||
|
|
mysql_charset="utf8mb4",
|
||
|
|
)
|
||
|
|
|
||
|
|
theory = Table(
|
||
|
|
"idac_profile_theory",
|
||
|
|
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("play_count", Integer, server_default="0"),
|
||
|
|
Column("play_count_multi", Integer, server_default="0"),
|
||
|
|
Column("partner_id", Integer),
|
||
|
|
Column("partner_progress", Integer),
|
||
|
|
Column("partner_progress_score", Integer),
|
||
|
|
Column("practice_start_rank", Integer, server_default="0"),
|
||
|
|
Column("general_flag", Integer, server_default="0"),
|
||
|
|
Column("vs_history", Integer, server_default="0"),
|
||
|
|
Column("vs_history_multi", Integer, server_default="0"),
|
||
|
|
Column("win_count", Integer, server_default="0"),
|
||
|
|
Column("win_count_multi", Integer, server_default="0"),
|
||
|
|
UniqueConstraint("user", "version", name="idac_profile_theory_uk"),
|
||
|
|
mysql_charset="utf8mb4",
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class IDACProfileData(BaseData):
|
||
|
|
def __init__(self, cfg: CoreConfig, conn: Connection) -> None:
|
||
|
|
super().__init__(cfg, conn)
|
||
|
|
self.date_time_format_ext = (
|
||
|
|
"%Y-%m-%d %H:%M:%S.%f" # needs to be lopped off at [:-5]
|
||
|
|
)
|
||
|
|
self.date_time_format_short = "%Y-%m-%d"
|
||
|
|
|
||
|
|
def get_profile(self, aime_id: int, version: int) -> Optional[Row]:
|
||
|
|
sql = select(profile).where(
|
||
|
|
and_(
|
||
|
|
profile.c.user == aime_id,
|
||
|
|
profile.c.version == version,
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
result = self.execute(sql)
|
||
|
|
if result is None:
|
||
|
|
return None
|
||
|
|
return result.fetchone()
|
||
|
|
|
||
|
|
def get_different_random_profiles(
|
||
|
|
self, aime_id: int, version: int, count: int = 9
|
||
|
|
) -> Optional[Row]:
|
||
|
|
sql = (
|
||
|
|
select(profile)
|
||
|
|
.where(
|
||
|
|
and_(
|
||
|
|
profile.c.user != aime_id,
|
||
|
|
profile.c.version == version,
|
||
|
|
)
|
||
|
|
)
|
||
|
|
.order_by(func.rand())
|
||
|
|
.limit(count)
|
||
|
|
)
|
||
|
|
|
||
|
|
result = self.execute(sql)
|
||
|
|
if result is None:
|
||
|
|
return None
|
||
|
|
return result.fetchall()
|
||
|
|
|
||
|
|
def get_profile_config(self, aime_id: int) -> Optional[Row]:
|
||
|
|
sql = select(config).where(
|
||
|
|
and_(
|
||
|
|
config.c.user == aime_id,
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
result = self.execute(sql)
|
||
|
|
if result is None:
|
||
|
|
return None
|
||
|
|
return result.fetchone()
|
||
|
|
|
||
|
|
def get_profile_avatar(self, aime_id: int) -> Optional[Row]:
|
||
|
|
sql = select(avatar).where(
|
||
|
|
and_(
|
||
|
|
avatar.c.user == aime_id,
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
result = self.execute(sql)
|
||
|
|
if result is None:
|
||
|
|
return None
|
||
|
|
return result.fetchone()
|
||
|
|
|
||
|
|
def get_profile_rank(self, aime_id: int, version: int) -> Optional[Row]:
|
||
|
|
sql = select(rank).where(
|
||
|
|
and_(
|
||
|
|
rank.c.user == aime_id,
|
||
|
|
rank.c.version == version,
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
result = self.execute(sql)
|
||
|
|
if result is None:
|
||
|
|
return None
|
||
|
|
return result.fetchone()
|
||
|
|
|
||
|
|
def get_profile_stock(self, aime_id: int, version: int) -> Optional[Row]:
|
||
|
|
sql = select(stock).where(
|
||
|
|
and_(
|
||
|
|
stock.c.user == aime_id,
|
||
|
|
stock.c.version == version,
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
result = self.execute(sql)
|
||
|
|
if result is None:
|
||
|
|
return None
|
||
|
|
return result.fetchone()
|
||
|
|
|
||
|
|
def get_profile_theory(self, aime_id: int, version: int) -> Optional[Row]:
|
||
|
|
sql = select(theory).where(
|
||
|
|
and_(
|
||
|
|
theory.c.user == aime_id,
|
||
|
|
theory.c.version == version,
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
result = self.execute(sql)
|
||
|
|
if result is None:
|
||
|
|
return None
|
||
|
|
return result.fetchone()
|
||
|
|
|
||
|
|
def put_profile(
|
||
|
|
self, aime_id: int, version: int, profile_data: Dict
|
||
|
|
) -> Optional[int]:
|
||
|
|
profile_data["user"] = aime_id
|
||
|
|
profile_data["version"] = version
|
||
|
|
|
||
|
|
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: Failed to update! aime_id: {aime_id}")
|
||
|
|
return None
|
||
|
|
return result.lastrowid
|
||
|
|
|
||
|
|
def put_profile_config(self, aime_id: int, config_data: Dict) -> Optional[int]:
|
||
|
|
config_data["user"] = aime_id
|
||
|
|
|
||
|
|
sql = insert(config).values(**config_data)
|
||
|
|
conflict = sql.on_duplicate_key_update(**config_data)
|
||
|
|
result = self.execute(conflict)
|
||
|
|
|
||
|
|
if result is None:
|
||
|
|
self.logger.warn(
|
||
|
|
f"put_profile_config: Failed to update! aime_id: {aime_id}"
|
||
|
|
)
|
||
|
|
return None
|
||
|
|
return result.lastrowid
|
||
|
|
|
||
|
|
def put_profile_avatar(self, aime_id: int, avatar_data: Dict) -> Optional[int]:
|
||
|
|
avatar_data["user"] = aime_id
|
||
|
|
|
||
|
|
sql = insert(avatar).values(**avatar_data)
|
||
|
|
conflict = sql.on_duplicate_key_update(**avatar_data)
|
||
|
|
result = self.execute(conflict)
|
||
|
|
|
||
|
|
if result is None:
|
||
|
|
self.logger.warn(
|
||
|
|
f"put_profile_avatar: Failed to update! aime_id: {aime_id}"
|
||
|
|
)
|
||
|
|
return None
|
||
|
|
return result.lastrowid
|
||
|
|
|
||
|
|
def put_profile_rank(
|
||
|
|
self, aime_id: int, version: int, rank_data: Dict
|
||
|
|
) -> Optional[int]:
|
||
|
|
rank_data["user"] = aime_id
|
||
|
|
rank_data["version"] = version
|
||
|
|
|
||
|
|
sql = insert(rank).values(**rank_data)
|
||
|
|
conflict = sql.on_duplicate_key_update(**rank_data)
|
||
|
|
result = self.execute(conflict)
|
||
|
|
|
||
|
|
if result is None:
|
||
|
|
self.logger.warn(f"put_profile_rank: Failed to update! aime_id: {aime_id}")
|
||
|
|
return None
|
||
|
|
return result.lastrowid
|
||
|
|
|
||
|
|
def put_profile_stock(
|
||
|
|
self, aime_id: int, version: int, stock_data: Dict
|
||
|
|
) -> Optional[int]:
|
||
|
|
stock_data["user"] = aime_id
|
||
|
|
stock_data["version"] = version
|
||
|
|
|
||
|
|
sql = insert(stock).values(**stock_data)
|
||
|
|
conflict = sql.on_duplicate_key_update(**stock_data)
|
||
|
|
result = self.execute(conflict)
|
||
|
|
|
||
|
|
if result is None:
|
||
|
|
self.logger.warn(f"put_profile_stock: Failed to update! aime_id: {aime_id}")
|
||
|
|
return None
|
||
|
|
return result.lastrowid
|
||
|
|
|
||
|
|
def put_profile_theory(
|
||
|
|
self, aime_id: int, version: int, theory_data: Dict
|
||
|
|
) -> Optional[int]:
|
||
|
|
theory_data["user"] = aime_id
|
||
|
|
theory_data["version"] = version
|
||
|
|
|
||
|
|
sql = insert(theory).values(**theory_data)
|
||
|
|
conflict = sql.on_duplicate_key_update(**theory_data)
|
||
|
|
result = self.execute(conflict)
|
||
|
|
|
||
|
|
if result is None:
|
||
|
|
self.logger.warn(
|
||
|
|
f"put_profile_theory: Failed to update! aime_id: {aime_id}"
|
||
|
|
)
|
||
|
|
return None
|
||
|
|
return result.lastrowid
|