artemis/titles/sao/schema/profile.py

from typing import Optional

from core.data.schema import BaseData, metadata
from sqlalchemy import Column, Table
from sqlalchemy.dialects.mysql import insert
from sqlalchemy.engine import Row
from sqlalchemy.schema import ForeignKey
from sqlalchemy.types import Integer, String

profile = Table(
    "sao_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("user_type", Integer, server_default="1"),
    Column("nick_name", String(16), server_default="PLAYER"),
    Column("rank_num", Integer, server_default="1"),
    Column("rank_exp", Integer, server_default="0"),
    Column("own_col", Integer, server_default="0"),
    Column("own_vp", Integer, server_default="300"),
    Column("own_yui_medal", Integer, server_default="0"),
    Column("setting_title_id", Integer, server_default="20005"),
)


class SaoProfileData(BaseData):
    async 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 = await self.execute(conflict)
        if result is None:
            self.logger.error(f"Failed to create SAO profile for user {user_id}!")
            return None
        return result.lastrowid

    async def put_profile(
        self,
        user_id: int,
        user_type: int,
        nick_name: str,
        rank_num: int,
        rank_exp: int,
        own_col: int,
        own_vp: int,
        own_yui_medal: int,
        setting_title_id: int,
    ) -> Optional[int]:
        sql = insert(profile).values(
            user=user_id,
            user_type=user_type,
            nick_name=nick_name,
            rank_num=rank_num,
            rank_exp=rank_exp,
            own_col=own_col,
            own_vp=own_vp,
            own_yui_medal=own_yui_medal,
            setting_title_id=setting_title_id,
        )

        conflict = sql.on_duplicate_key_update(
            rank_num=rank_num,
            rank_exp=rank_exp,
            own_col=own_col,
            own_vp=own_vp,
            own_yui_medal=own_yui_medal,
            setting_title_id=setting_title_id,
        )

        result = await self.execute(conflict)
        if result is None:
            self.logger.error(f"{__name__} failed to insert profile! user: {user_id}")
            return None

        return result.lastrowid

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