artemis/titles/pokken/schema/profile.py

from typing import Optional, Dict, List
from sqlalchemy import Table, Column, UniqueConstraint, PrimaryKeyConstraint, and_, case
from sqlalchemy.types import Integer, String, TIMESTAMP, Boolean, JSON
from sqlalchemy.schema import ForeignKey
from sqlalchemy.sql import func, select, update, delete
from sqlalchemy.engine import Row
from sqlalchemy.dialects.mysql import insert

from core.data.schema import BaseData, metadata

# Some more of the repeated fields could probably be their own tables, for now I just did the ones that made sense to me
# Having the profile table be this massive kinda blows for updates but w/e, **kwargs to the rescue
profile = Table(
    'pokken_profile',
    metadata,
    Column('id', Integer, primary_key=True, nullable=False),
    Column('user', ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False, unique=True),
    Column('trainer_name', String(16)), # optional
    Column('home_region_code', Integer),
    Column('home_loc_name', String(255)),
    Column('pref_code', Integer),
    Column('navi_newbie_flag', Boolean),
    Column('navi_enable_flag', Boolean),
    Column('pad_vibrate_flag', Boolean),
    Column('trainer_rank_point', Integer),
    Column('wallet', Integer),
    Column('fight_money', Integer),
    Column('score_point', Integer),
    Column('grade_max_num', Integer),
    Column('extra_counter', Integer), # Optional
    Column('total_play_days', Integer),
    Column('play_date_time', Integer),
    Column('lucky_box_fail_num', Integer),
    Column('event_reward_get_flag', Integer),
    Column('rank_pvp_all', Integer),
    Column('rank_pvp_loc', Integer),
    Column('rank_cpu_all', Integer),
    Column('rank_cpu_loc', Integer),
    Column('rank_event', Integer),
    Column('awake_num', Integer),
    Column('use_support_num', Integer),
    Column('rankmatch_flag', Integer),
    Column('rankmatch_max', Integer), # Optional
    Column('rankmatch_success', Integer), # Optional
    Column('beat_num', Integer), # Optional
    Column('title_text_id', Integer),
    Column('title_plate_id', Integer),
    Column('title_decoration_id', Integer),
    Column('support_pokemon_list', JSON), # Repeated, Integer
    Column('support_set_1', JSON), # Repeated, Integer
    Column('support_set_2', JSON), # Repeated, Integer
    Column('support_set_3', JSON), # Repeated, Integer
    Column('navi_trainer', Integer),
    Column('navi_version_id', Integer),
    Column('aid_skill_list', JSON), # Repeated, Integer
    Column('aid_skill', Integer),
    Column('comment_text_id', Integer),
    Column('comment_word_id', Integer),
    Column('latest_use_pokemon', Integer),
    Column('ex_ko_num', Integer),
    Column('wko_num', Integer),
    Column('timeup_win_num', Integer),
    Column('cool_ko_num', Integer),
    Column('cool_ko_num', Integer),
    Column('perfect_ko_num', Integer),
    Column('record_flag', Integer),
    Column('continue_num', Integer),
    Column('avatar_body', Integer), # Optional
    Column('avatar_gender', Integer), # Optional
    Column('avatar_background', Integer), # Optional
    Column('avatar_head', Integer), # Optional
    Column('avatar_battleglass', Integer), # Optional
    Column('avatar_face0', Integer), # Optional
    Column('avatar_face1', Integer), # Optional
    Column('avatar_face2', Integer), # Optional
    Column('avatar_bodyall', Integer), # Optional
    Column('avatar_wear', Integer), # Optional
    Column('avatar_accessory', Integer), # Optional
    Column('avatar_stamp', Integer), # Optional
    Column('event_state', Integer),
    Column('event_id', Integer),
    Column('sp_bonus_category_id_1', Integer),
    Column('sp_bonus_key_value_1', Integer),
    Column('sp_bonus_category_id_2', Integer),
    Column('sp_bonus_key_value_2', Integer),
    Column('last_play_event_id', Integer), # Optional
    mysql_charset="utf8mb4",
)

tutorial_progress = Table(
    'pokken_tutorial_progress',
    metadata,
    Column('id', Integer, primary_key=True, nullable=False),
    Column('user', ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
    Column('flag', Integer),
    UniqueConstraint('user', 'flag', name='pokken_tutorial_progress_uk'),
    mysql_charset="utf8mb4",
)

rankmatch_progress = Table(
    'pokken_rankmatch_progress',
    metadata,
    Column('id', Integer, primary_key=True, nullable=False),
    Column('user', ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
    Column('progress', Integer),
    UniqueConstraint('user', 'progress', name='pokken_rankmatch_progress_uk'),
    mysql_charset="utf8mb4",
)

achievement_flag = Table(
    'pokken_achievement_flag',
    metadata,
    Column('id', Integer, primary_key=True, nullable=False),
    Column('user', ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
    Column('flag', Integer),
    UniqueConstraint('user', 'flag', name='pokken_achievement_flag_uk'),
    mysql_charset="utf8mb4",
)

event_achievement_flag = Table(
    'pokken_event_achievement_flag',
    metadata,
    Column('id', Integer, primary_key=True, nullable=False),
    Column('user', ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
    Column('flag', Integer),
    UniqueConstraint('user', 'flag', name='pokken_event_achievement_flag_uk'),
    mysql_charset="utf8mb4",
)

event_achievement_param = Table(
    'pokken_event_achievement_param',
    metadata,
    Column('id', Integer, primary_key=True, nullable=False),
    Column('user', ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
    Column('param', Integer),
    UniqueConstraint('user', 'param', name='pokken_event_achievement_param_uk'),
    mysql_charset="utf8mb4",
)

pokemon_data = Table(
    'pokken_pokemon_data',
    metadata,
    Column('id', Integer, primary_key=True, nullable=False),
    Column('user', ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
    Column('char_id', Integer, nullable=False),
    Column('illustration_book_no', Integer, nullable=False),
    Column('pokemon_exp', Integer, nullable=False),
    Column('battle_num_vs_wan', Integer, nullable=False),
    Column('win_vs_wan', Integer, nullable=False),
    Column('battle_num_vs_lan', Integer, nullable=False),
    Column('win_vs_lan', Integer, nullable=False),
    Column('battle_num_vs_cpu', Integer, nullable=False),
    Column('win_cpu', Integer, nullable=False),
    Column('battle_all_num_tutorial', Integer, nullable=False),
    Column('battle_num_tutorial', Integer, nullable=False),
    Column('bp_point_atk', Integer, nullable=False),
    Column('bp_point_res', Integer, nullable=False),
    Column('bp_point_def', Integer, nullable=False),
    Column('bp_point_sp', Integer, nullable=False),
)

class PokkenProfileData(BaseData):
    def create_profile(self, user_id: int) -> Optional[int]:
        sql = insert(profile).values(user = user_id)
        conflict = sql.on_duplicate_key_update(user = user_id)
        
        result = self.execute(conflict)
        if result is None:
            self.logger.error(f"Failed to create pokken profile for user {user_id}!")
            return None
        return result.lastrowid
    
    def set_profile_name(self, user_id: int, new_name: str) -> None:
        sql = update(profile).where(profile.c.user == user_id).values(trainer_name = new_name)
        result = self.execute(sql)
        if result is None:
            self.logger.error(f"Failed to update pokken profile name for user {user_id}!")
    
    def update_profile(self, user_id: int, profile_data: Dict) -> None:
        """
        TODO: Find out what comes in on the SaveUserRequestData protobuf and save it!
        """
        pass

    def get_profile(self, user_id: int) -> Optional[Row]:
        sql = profile.select(profile.c.user == user_id)
        result = self.execute(sql)
        if result is None: return None
        return result.fetchone()

    def put_pokemon_data(self, user_id: int, pokemon_data: Dict) -> Optional[int]:
        pass

    def get_pokemon_data(self, user_id: int, pokemon_id: int) -> Optional[Row]:
        pass

    def get_all_pokemon_data(self, user_id: int) -> Optional[List[Row]]:
        pass