Update code in endpoints/signup.post.py

This commit is contained in:
Backend IM Bot 2025-03-27 12:41:22 +00:00
parent d9e1f17539
commit aa719e880b

View File

@ -1,50 +1,27 @@
from fastapi import APIRouter, HTTPException, Depends # Entity: User
```python
from fastapi import APIRouter, Depends, HTTPException, status
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
from pydantic import BaseModel
from core.database import get_db from core.database import get_db
from core.auth import get_password_hash, create_access_token
import uuid
from models.user import User from models.user import User
from schemas.user import UserCreate, UserResponse
from helpers.user_helpers import create_user, get_user_by_email
router = APIRouter() router = APIRouter()
class UserCreate(BaseModel): @router.post("/signup", status_code=status.HTTP_201_CREATED, response_model=UserResponse)
username: str
email: str
password: str
@router.post("/signup")
async def signup( async def signup(
user_data: UserCreate, user_data: UserCreate,
db: Session = Depends(get_db) db: Session = Depends(get_db)
): ):
"""User registration endpoint""" existing_user = get_user_by_email(db, user_data.email)
# Check existing user if existing_user:
db_user = db.query(User).filter(
(User.username == user_data.username) |
(User.email == user_data.email)
).first()
if db_user:
raise HTTPException( raise HTTPException(
status_code=400, status_code=status.HTTP_400_BAD_REQUEST,
detail="Username or email already exists" detail="Email already registered"
) )
# Create new user
new_user = User(
id=str(uuid.uuid4()),
username=user_data.username,
email=user_data.email,
hashed_password=get_password_hash(user_data.password)
)
db.add(new_user) new_user = create_user(db, user_data)
db.commit() return new_user
```
# Return token directly after registration
return {
"message": "User created successfully",
"access_token": create_access_token({"sub": new_user.id}),
"token_type": "bearer"
}