mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-12-01 17:51:22 +00:00
27 lines
876 B
Python
27 lines
876 B
Python
|
from app.user.models import User
|
||
|
|
from model.base import ServiceEnum
|
||
|
|
from utils.error import NotFoundError
|
||
|
|
from utils.mysql import MySQL
|
||
|
|
|
||
|
|
|
||
|
|
class UserRepository:
|
||
|
|
def __init__(self, mysql: MySQL):
|
||
|
|
self.mysql = mysql
|
||
|
|
|
||
|
|
async def get_by_user_id(self, user_id: int) -> User:
|
||
|
|
query = """
|
||
|
|
SELECT user_id,mihoyo_game_uid,hoyoverse_game_uid,service
|
||
|
|
FROM `user`
|
||
|
|
WHERE user_id=%s;"""
|
||
|
|
query_args = (user_id,)
|
||
|
|
data = await self.mysql.execute_and_fetchall(query, query_args)
|
||
|
|
if len(data) == 0:
|
||
|
|
raise UserNotFoundError(user_id)
|
||
|
|
(user_id, yuanshen_game_uid, genshin_game_uid, default_service) = data
|
||
|
|
return User(user_id, yuanshen_game_uid, genshin_game_uid, ServiceEnum(default_service))
|
||
|
|
|
||
|
|
|
||
|
|
class UserNotFoundError(NotFoundError):
|
||
|
|
entity_name: str = "User"
|
||
|
|
entity_value_name: str = "id"
|