forked from Hay1tsme/artemis
312 lines
13 KiB
Python
312 lines
13 KiB
Python
from typing import Optional, Tuple, List
|
|
from sqlalchemy import Table, Column, UniqueConstraint, PrimaryKeyConstraint, and_, case
|
|
from sqlalchemy.types import Integer, String, BOOLEAN, INTEGER, BIGINT, VARCHAR, TIMESTAMP
|
|
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
|
|
|
|
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="0"),
|
|
Column("own_yui_medal", Integer, server_default="0"),
|
|
Column("setting_title_id", Integer, server_default="20005"),
|
|
Column("my_shop", INTEGER),
|
|
Column("fav_hero", INTEGER, ForeignKey("sao_hero_log_data.id", ondelete="set null", onupdate="cascade")),
|
|
Column("when_register", TIMESTAMP, server_default=func.now()),
|
|
Column("last_login_date", TIMESTAMP),
|
|
Column("last_yui_medal_date", TIMESTAMP),
|
|
Column("last_bonus_yui_medal_date", TIMESTAMP),
|
|
Column("last_comeback_date", TIMESTAMP),
|
|
Column("last_login_bonus_date", TIMESTAMP),
|
|
Column("ad_confirm_date", TIMESTAMP),
|
|
Column("login_ct", INTEGER, server_default="0"),
|
|
mysql_charset="utf8mb4"
|
|
)
|
|
|
|
beginner_mission = Table(
|
|
"sao_player_beginner_mission",
|
|
metadata,
|
|
Column("id", BIGINT, primary_key=True, nullable=False),
|
|
Column("user", INTEGER, ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False, unique=True),
|
|
Column("beginner_mission_id", INTEGER, nullable=False),
|
|
Column("condition_id", INTEGER, nullable=False),
|
|
Column("is_seat", BOOLEAN, nullable=False, server_default="0"),
|
|
Column("achievement_num", INTEGER, nullable=False),
|
|
Column("complete_flag", BOOLEAN, nullable=False, server_default="0"),
|
|
Column("complete_date", TIMESTAMP),
|
|
Column("reward_received_flag", BOOLEAN, nullable=False, server_default="0"),
|
|
Column("reward_received_date", TIMESTAMP),
|
|
UniqueConstraint("user", "condition_id", name="sao_player_beginner_mission_uk"),
|
|
mysql_charset="utf8mb4"
|
|
)
|
|
|
|
resource_card = Table(
|
|
"sao_player_resource_card",
|
|
metadata,
|
|
Column("id", BIGINT, primary_key=True, nullable=False),
|
|
Column("user", INTEGER, ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
|
|
Column("common_reward_type", INTEGER, nullable=False),
|
|
Column("common_reward_id", INTEGER, nullable=False),
|
|
Column("holographic_flag", BOOLEAN, nullable=False, server_default="0"),
|
|
Column("serial", VARCHAR(20), unique=True),
|
|
mysql_charset="utf8mb4"
|
|
)
|
|
|
|
hero_card = Table(
|
|
"sao_player_hero_card",
|
|
metadata,
|
|
Column("id", BIGINT, primary_key=True, nullable=False),
|
|
Column("user", INTEGER, ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
|
|
Column("user_hero_id", INTEGER, ForeignKey("sao_hero_log_data.id", ondelete="cascade", onupdate="cascade"), nullable=False),
|
|
Column("holographic_flag", BOOLEAN, nullable=False, server_default="0"),
|
|
Column("serial", VARCHAR(20), unique=True),
|
|
mysql_charset="utf8mb4"
|
|
)
|
|
|
|
tutorial = Table(
|
|
"sao_player_tutorial",
|
|
metadata,
|
|
Column("id", BIGINT, primary_key=True, nullable=False),
|
|
Column("user", INTEGER, ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
|
|
Column("tutorial_byte", INTEGER, nullable=False),
|
|
UniqueConstraint("user", "tutorial_byte", name="sao_player_tutorial_uk"),
|
|
mysql_charset="utf8mb4"
|
|
)
|
|
|
|
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:
|
|
return result.inserted_primary_key['id']
|
|
self.logger.error("Failed to create SAO profile!")
|
|
|
|
async def set_my_shop(self, user_id: int, store_id: int):
|
|
result = await self.execute(profile.update(profile.c.user == user_id).values(my_shop = store_id))
|
|
if result is None:
|
|
self.logger.error(f"Failed to set my shop for user {user_id} to {store_id}!")
|
|
|
|
async def user_login(self, user_id: int) -> Optional[Row]:
|
|
sql = profile.update(profile.c.user == user_id).values(
|
|
login_ct=profile.c.login_ct + 1,
|
|
last_login_date=func.now()
|
|
)
|
|
result = await self.execute(sql)
|
|
if result:
|
|
return result.last_updated_params()
|
|
self.logger.error(f"Failed to create log in user {user_id}!")
|
|
|
|
async def update_yui_medal_date(self, user_id: int) -> None:
|
|
sql = profile.update(profile.c.user == user_id).values(
|
|
last_yui_medal_date=func.now()
|
|
)
|
|
result = await self.execute(sql)
|
|
if result is None:
|
|
self.logger.error(f"Failed to update user {user_id} yui medal date!")
|
|
|
|
async def add_yui_medals(self, user_id: int, num_medals: int = 1):
|
|
sql = profile.update(profile.c.user == user_id).values(
|
|
own_yui_medal=profile.c.own_yui_medal + num_medals
|
|
)
|
|
result = await self.execute(sql)
|
|
if result is None:
|
|
self.logger.error(f"Failed to give user {user_id} {num_medals} yui medals!")
|
|
|
|
async def add_col(self, user_id: int, num_col: int) -> None:
|
|
sql = profile.update(profile.c.user == user_id).values(
|
|
own_col=profile.c.own_col + num_col
|
|
)
|
|
result = await self.execute(sql)
|
|
if not result:
|
|
self.logger.error(f"Failed to give user {user_id} {num_col} Col!")
|
|
|
|
async def add_vp(self, user_id: int, num_vp: int) -> None:
|
|
sql = profile.update(profile.c.user == user_id).values(
|
|
own_vp=profile.c.own_vp + num_vp
|
|
)
|
|
result = await self.execute(sql)
|
|
if not result:
|
|
self.logger.error(f"Failed to give user {user_id} {num_vp} VP!")
|
|
|
|
async def add_exp(self, user_id: int, xp_ammount: int) -> Optional[int]:
|
|
sql = profile.update(profile.c.user == user_id).values(
|
|
rank_exp=profile.c.rank_exp + xp_ammount
|
|
)
|
|
result = await self.execute(sql)
|
|
if not result:
|
|
self.logger.error(f"Failed to give user {user_id} {xp_ammount} xp!")
|
|
|
|
async def get_exp(self, user_id: int) -> Optional[int]:
|
|
result = await self.execute(select(profile.c.rank_exp).where(profile.c.user==user_id))
|
|
if result:
|
|
row = result.fetchone()
|
|
if row:
|
|
return row['rank_exp']
|
|
return 0
|
|
self.logger.error(f"Failed to query rank xp for user {user_id}")
|
|
|
|
async def set_level(self, user_id: int, level: int):
|
|
sql = profile.update(profile.c.user == user_id).values(
|
|
rank_num=level
|
|
)
|
|
result = await self.execute(sql)
|
|
if result is None:
|
|
self.logger.error(f"Failed to set user {user_id} level to {level}!")
|
|
|
|
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:
|
|
return result.inserted_primary_key['id']
|
|
self.logger.error(f"Failed to insert profile! user: {user_id}")
|
|
|
|
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()
|
|
|
|
async def set_profile_name(self, user_id: int, new_name: str) -> None:
|
|
sql = profile.update(profile.c.user == user_id).values(
|
|
nick_name=new_name
|
|
)
|
|
result = await self.execute(sql)
|
|
if result is None:
|
|
self.logger.error(f"Failed to update nickname {new_name} for user {user_id}")
|
|
|
|
async def add_tutorial_byte(self, user_id: int, tutorial_byte: int) -> None:
|
|
sql = insert(tutorial).values(
|
|
user = user_id,
|
|
tutorial_byte = tutorial_byte
|
|
)
|
|
|
|
conflict = sql.on_duplicate_key_update(tutorial_byte = tutorial_byte)
|
|
result = await self.execute(conflict)
|
|
if result is None:
|
|
self.logger.error(f"Failed to add tutorial byte {tutorial_byte} to user {user_id}")
|
|
|
|
async def get_tutorial_bytes(self, user_id: int) -> Optional[List[Row]]:
|
|
sql = tutorial.select(tutorial.c.user == user_id)
|
|
|
|
result = await self.execute(sql)
|
|
if result is None:
|
|
return None
|
|
return result.fetchall()
|
|
|
|
async def put_hero_card(self, user_id: int, serial: str, user_hero_id: int, is_holo: bool) -> Optional[int]:
|
|
sql = insert(hero_card).values(
|
|
user=user_id,
|
|
user_hero_id=user_hero_id,
|
|
holographic_flag=is_holo,
|
|
serial=serial
|
|
)
|
|
|
|
conflict = sql.on_duplicate_key_update(
|
|
holographic_flag=is_holo,
|
|
serial=serial
|
|
)
|
|
result = await self.execute(conflict)
|
|
if result:
|
|
return result.inserted_primary_key['id']
|
|
self.logger.error(f"Failed to insert card {serial} for user {user_id} as hero {user_hero_id}")
|
|
|
|
async def get_hero_card(self, serial: str) -> Optional[Row]:
|
|
result = await self.execute(hero_card.select(hero_card.c.serial == serial))
|
|
if result is None:
|
|
return None
|
|
return result.fetchone()
|
|
|
|
async def put_resource_card(self, user_id: int, serial: str, reward_type: int, reward_id: int, is_holo: bool) -> Optional[int]:
|
|
sql = insert(resource_card).values(
|
|
user=user_id,
|
|
common_reward_type=reward_type,
|
|
common_reward_id=reward_id,
|
|
holographic_flag=is_holo,
|
|
serial=serial
|
|
)
|
|
|
|
conflict = sql.on_duplicate_key_update(
|
|
holographic_flag=is_holo,
|
|
serial=serial
|
|
)
|
|
result = await self.execute(conflict)
|
|
if result:
|
|
return result.inserted_primary_key['id']
|
|
self.logger.error(f"Failed to insert card {serial} for user {user_id} as resource {reward_id}")
|
|
|
|
async def get_resource_card(self, serial: str) -> Optional[int]:
|
|
result = await self.execute(resource_card.select(resource_card.c.serial == serial))
|
|
if result is None:
|
|
return None
|
|
return result.fetchone()
|
|
|
|
async def update_beginner_mission_date(self, user_id: int) -> None:
|
|
sql = profile.update(profile.c.user == user_id).values(
|
|
ad_confirm_date=func.now()
|
|
)
|
|
result = await self.execute(sql)
|
|
if result is None:
|
|
self.logger.error(f"Failed to update user {user_id} yui medal date!")
|
|
|
|
async def put_beginner_mission(self, user_id: int, beginner_mission_id: int, condition_id: int, is_seat: bool, achievement_num: int) -> Optional[int]:
|
|
pass
|
|
|
|
async def complete_beginner_mission(self, user_id: int, condition_id: int) -> None:
|
|
pass
|
|
|
|
async def reward_beginner_mission(self, user_id: int, condition_id: int) -> None:
|
|
pass
|
|
|
|
async def get_beginner_missions(self, user_id: int) -> Optional[List[Row]]:
|
|
pass
|
|
|
|
async def get_beginner_missions_by_mission_id(self, user_id: int, beginner_mission_id: int) -> Optional[List[Row]]:
|
|
pass
|
|
|
|
async def get_beginner_mission(self, user_id: int, condition_id: int) -> Optional[Row]:
|
|
pass
|
|
|
|
async def set_title(self, user_id: int, title_id: int) -> None:
|
|
result = await self.execute(profile.update(profile.c.user == user_id).values(setting_title_id=title_id))
|
|
if not result:
|
|
self.logger.error(f"Failed to set user {user_id} title to {title_id}")
|