31 lines
645 B
Python
31 lines
645 B
Python
from typing import Protocol
|
|
|
|
from app.models.user import User
|
|
|
|
|
|
class IUserRepository(Protocol):
|
|
async def get_by_telegram_id(self, telegram_id: int) -> User | None: ...
|
|
|
|
async def create(
|
|
self,
|
|
*,
|
|
telegram_id: int,
|
|
username: str | None,
|
|
first_name: str | None,
|
|
referrer_id: int | None,
|
|
) -> User: ...
|
|
|
|
async def update_profile(
|
|
self,
|
|
user: User,
|
|
*,
|
|
username: str | None,
|
|
first_name: str | None,
|
|
) -> User: ...
|
|
|
|
async def set_referrer_if_unset(
|
|
self,
|
|
user: User,
|
|
referrer_internal_id: int,
|
|
) -> None: ...
|