Fixed users not being in a default organization at account creation

Author: kevinjnev

backend/app/api/routes/users.py

@@ -152,6 +152,15 @@ def register_user(session: SessionDep, user_in: UserRegister) -> Any:
         )
     user_create = UserCreate.model_validate(user_in)
     user = crud.create_user(session=session, user_create=user_create)
+    
+    # Assign user to default organization
+    default_org = crud.get_default_organization(session=session)
+    if default_org:
+        user.organization_id = default_org.id
+        session.add(user)
+        session.commit()
+        session.refresh(user)
+    
     return user
 
 

backend/app/crud.py

@@ -92,6 +92,12 @@ def get_organization(
     return session.get(Organization, organization_id)
 
 
+def get_default_organization(*, session: Session) -> Organization | None:
+    """Get the default organization (typically 'Default Organization')"""
+    statement = select(Organization).where(Organization.name == "Default Organization")
+    return session.exec(statement).first()
+
+
 def update_organization(
     *,
     session: Session,