feat: referal spec on registration

This commit is contained in:
2026-08-03 10:35:26 +07:00
parent 680c15414c
commit 801014c392
8 changed files with 88 additions and 5 deletions

View File

@@ -3,7 +3,7 @@ from db.models.users import User
from repositories.sessions import SessionsRepository
from schemas.login import UserLogin
from schemas.providers import ProvidersType
from schemas.user import UserInfo
from schemas.user import SubscriptionData, UserInfo
async def authorize_user(
@@ -15,9 +15,23 @@ async def authorize_user(
await sessions_repo.create(user_id=user.id, refresh_token_hash=refresh_token_hash, iss=iss)
sub = (
SubscriptionData(
devices=user.subscription.devices,
expires_at=user.subscription.expires_at,
addon_ids=[a.id for a in user.subscription.addons],
)
if user.subscription
else None
)
return UserLogin(
access_token=key_pair.access_token,
refresh_token=key_pair.refresh_token,
user=UserInfo(username=user.username, telegram_id=user.telegram_id),
user=UserInfo(
username=user.username,
telegram_id=user.telegram_id,
referal_code=user.referal_code,
subscription=sub,
),
expires_at=key_pair.expires_at,
)