Update code in endpoints/signup.post.py

This commit is contained in:
Backend IM Bot 2025-03-24 21:08:42 +00:00
parent b0e4ef2c91
commit 6b9f1083da

View File

@ -1,50 +1,41 @@
from fastapi import APIRouter, HTTPException, Depends from fastapi import APIRouter, HTTPException
from sqlalchemy.orm import Session
from pydantic import BaseModel
from core.database import get_db
from core.auth import get_password_hash, create_access_token
import uuid import uuid
from models.user import User
# In-memory storage
users = []
posts = []
router = APIRouter() router = APIRouter()
class UserCreate(BaseModel):
username: str
email: str
password: str
@router.post("/signup") @router.post("/signup")
async def signup( async def signup_user(
user_data: UserCreate, username: str,
db: Session = Depends(get_db) email: str,
password: str
): ):
"""User registration endpoint""" """Create new user account"""
# Check existing user if any(u["username"] == username for u in users):
db_user = db.query(User).filter( raise HTTPException(status_code=400, detail="Username already exists")
(User.username == user_data.username) |
(User.email == user_data.email) if any(u["email"] == email for u in users):
).first() raise HTTPException(status_code=400, detail="Email already registered")
if db_user:
raise HTTPException(
status_code=400,
detail="Username or email already exists"
)
# Create new user user_id = str(uuid.uuid4())
new_user = User( user = {
id=str(uuid.uuid4()), "id": user_id,
username=user_data.username, "username": username,
email=user_data.email, "email": email,
hashed_password=get_password_hash(user_data.password) "password": password,
) "disabled": False
}
db.add(new_user) users.append(user)
db.commit()
# Return token directly after registration
return { return {
"message": "User created successfully", "message": "User created successfully",
"access_token": create_access_token({"sub": new_user.id}), "user_id": user_id,
"token_type": "bearer" "username": username,
} "next_steps": [
"Verify your email",
"Complete profile setup"
]
}