uv migration

1 files changed, 18 insertions, 31 deletions

diff --git a/app/routes/register.py b/app/routes/register.py
index 779cf73..f0b36ed 100644
--- a/app/routes/register.py
+++ b/app/routes/register.py
@@ -4,9 +4,9 @@ from typing import Optional
 from fastapi import APIRouter, Depends, HTTPException, Response
 from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy.future import select
-from sqlalchemy.orm import selectinload
 
 from app.auth.jwt import create_access_token, create_refresh_token
+from app.models.integrations import UserIntegration
 from app.models.profile import Profile
 from app.models.user import User
 from app.schemas.user import UserCreate, UserRead
@@ -27,7 +27,9 @@ async def register_user(
 
     email: Optional[str] = user.email.strip() if user.email else None
     logger.debug("Normalized email value: {}", email)
-    logger.info("Registration attempt | username={} email={}", user.username, email)
+    logger.info(
+        "Registration attempt | username={} email={}", user.username, email
+    )
 
     if not (
         re.search(r"[A-Za-z]", user.password)
@@ -46,10 +48,13 @@ async def register_user(
             },
         )
 
-    result = await session.execute(select(User).where(User.username == user.username))
+    result = await session.execute(
+        select(User).where(User.username == user.username)
+    )
     if result.scalars().first():
         logger.warning(
-            "Registration failed | username already exists | username={}", user.username
+            "Registration failed | username already exists | username={}",
+            user.username,
         )
         raise HTTPException(
             status_code=400,
@@ -69,34 +74,17 @@ async def register_user(
 
     hashed_password = hash_password(user.password)
 
-    new_user = User(username=user.username, email=email, password=hashed_password)
-    session.add(new_user)
-    await session.flush()
-
-    new_profile = Profile(user_id=new_user.id)
-    session.add(new_profile)
-
-    from app.models.integrations import UserIntegration
-
-    new_integrations = UserIntegration(user_id=new_user.id)
-    session.add(new_integrations)
-
-    await session.commit()
-
-    result = await session.execute(
-        select(User)
-        .options(
-            selectinload(User.profile),
-            selectinload(User.integrations),
-        )
-        .where(User.id == new_user.id)
+    new_user = User(
+        username=user.username,
+        email=email,
+        password=hashed_password,
+        profile=Profile(),
+        integrations=UserIntegration(),
     )
-    new_user = result.scalars().first()
 
-    result = await session.execute(
-        select(User).options(selectinload(User.profile)).where(User.id == new_user.id)
-    )
-    new_user = result.scalars().first()
+    session.add(new_user)
+    await session.commit()
+    await session.refresh(new_user, ["profile", "integrations"])
 
     logger.success(
         "User successfully registered | id={} username={} email={}",
@@ -104,7 +92,6 @@ async def register_user(
         new_user.username,
         new_user.email,
     )
-
     access_token = create_access_token(
         {"sub": str(new_user.id), "token_version": new_user.token_version}
     )