forked from Hay1tsme/artemis
sao: backport changes from diana
This commit is contained in:
@ -1,13 +1,12 @@
|
||||
from typing import Optional, Dict, List
|
||||
from typing import Optional, Tuple, List
|
||||
from sqlalchemy import Table, Column, UniqueConstraint, PrimaryKeyConstraint, and_, case
|
||||
from sqlalchemy.types import Integer, String, TIMESTAMP, Boolean, JSON
|
||||
from sqlalchemy.types import Integer, String, BOOLEAN, INTEGER, BIGINT, VARCHAR, TIMESTAMP
|
||||
from sqlalchemy.schema import ForeignKey
|
||||
from sqlalchemy.sql import func, select, update, delete
|
||||
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 ..const import SaoConstants
|
||||
|
||||
profile = Table(
|
||||
"sao_profile",
|
||||
@ -24,9 +23,70 @@ profile = Table(
|
||||
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_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):
|
||||
@ -35,10 +95,81 @@ class SaoProfileData(BaseData):
|
||||
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 create SAO profile for user {user_id}!")
|
||||
return None
|
||||
return result.lastrowid
|
||||
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(
|
||||
@ -63,17 +194,118 @@ class SaoProfileData(BaseData):
|
||||
)
|
||||
|
||||
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
|
||||
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()
|
||||
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}")
|
||||
|
||||
Reference in New Issue
Block a user