diff --git a/endpoints/signup.post.py b/endpoints/signup.post.py index 77e4d09..65c4c13 100644 --- a/endpoints/signup.post.py +++ b/endpoints/signup.post.py @@ -1,50 +1,35 @@ -from fastapi import APIRouter, HTTPException, Depends +# Entity: User + +from fastapi import APIRouter, Depends, HTTPException, status 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 -from models.user import User +from core.models.user import User +from core.schemas.user import UserCreate, UserSchema +from core.security import get_password_hash router = APIRouter() -class UserCreate(BaseModel): - username: str - email: str - password: str - -@router.post("/signup") +@router.post("/signup", status_code=status.HTTP_201_CREATED, response_model=UserSchema) async def signup( user_data: UserCreate, db: Session = Depends(get_db) ): - """User registration endpoint""" - # Check existing user - db_user = db.query(User).filter( - (User.username == user_data.username) | - (User.email == user_data.email) - ).first() - + db_user = db.query(User).filter(User.email == user_data.email).first() if db_user: raise HTTPException( - status_code=400, - detail="Username or email already exists" + status_code=status.HTTP_400_BAD_REQUEST, + detail="Email already registered" ) - - # Create new user - new_user = User( - id=str(uuid.uuid4()), - username=user_data.username, + + hashed_password = get_password_hash(user_data.password) + db_user = User( email=user_data.email, - hashed_password=get_password_hash(user_data.password) + full_name=user_data.full_name, + hashed_password=hashed_password ) - db.add(new_user) + db.add(db_user) db.commit() - - # Return token directly after registration - return { - "message": "User created successfully", - "access_token": create_access_token({"sub": new_user.id}), - "token_type": "bearer" - } + db.refresh(db_user) + + return db_user \ No newline at end of file