feat: invite code

This commit is contained in:
2025-03-05 19:34:16 +03:00
parent 6c2698679d
commit 3d9a3cef73
5 changed files with 17 additions and 12 deletions

View File

@@ -175,7 +175,7 @@ class UserService(BaseService):
random.shuffle(code)
return ''.join(code)
async def generate_invite_code(self, user: User) -> GenerateInviteCodeResponse:
async def generate_invite_code(self, user: User, request: GenerateInviteCodeRequest) -> GenerateInviteCodeResponse:
MAX_ATTEMPTS = 5
try:
if not user.is_admin:
@@ -195,7 +195,8 @@ class UserService(BaseService):
new_invite_code = InviteCode(
code=invite_code,
created_at=datetime.datetime.now(),
created_by_id=user.id
created_by_id=user.id,
user_id=request.user_id
)
self.session.add(new_invite_code)
await self.session.commit()