Refactoring code for modular approach
This commit is contained in:
70
app/api/v1/auth.py
Normal file
70
app/api/v1/auth.py
Normal file
@@ -0,0 +1,70 @@
|
||||
from fastapi import APIRouter, Depends, HTTPException
|
||||
from sqlalchemy.orm import Session
|
||||
from app.core.database import get_db
|
||||
from app.api.deps import get_current_user
|
||||
from models.api_models import UserRegister, UserLogin, UserResponse, LoginResponse
|
||||
from models.database_models import User
|
||||
from middleware.auth import AuthManager
|
||||
import logging
|
||||
|
||||
router = APIRouter()
|
||||
auth_manager = AuthManager()
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
# Authentication Routes
|
||||
@router.post("/register", response_model=UserResponse)
|
||||
def register_user(user_data: UserRegister, db: Session = Depends(get_db)):
|
||||
"""Register a new user"""
|
||||
# Check if email already exists
|
||||
existing_email = db.query(User).filter(User.email == user_data.email).first()
|
||||
if existing_email:
|
||||
raise HTTPException(status_code=400, detail="Email already registered")
|
||||
|
||||
# Check if username already exists
|
||||
existing_username = db.query(User).filter(User.username == user_data.username).first()
|
||||
if existing_username:
|
||||
raise HTTPException(status_code=400, detail="Username already taken")
|
||||
|
||||
# Hash password and create user
|
||||
hashed_password = auth_manager.hash_password(user_data.password)
|
||||
new_user = User(
|
||||
email=user_data.email,
|
||||
username=user_data.username,
|
||||
hashed_password=hashed_password,
|
||||
role="user",
|
||||
is_active=True
|
||||
)
|
||||
|
||||
db.add(new_user)
|
||||
db.commit()
|
||||
db.refresh(new_user)
|
||||
|
||||
logger.info(f"New user registered: {new_user.username}")
|
||||
return new_user
|
||||
|
||||
|
||||
@router.post("/login", response_model=LoginResponse)
|
||||
def login_user(user_credentials: UserLogin, db: Session = Depends(get_db)):
|
||||
"""Login user and return JWT token"""
|
||||
user = auth_manager.authenticate_user(db, user_credentials.username, user_credentials.password)
|
||||
if not user:
|
||||
raise HTTPException(status_code=401, detail="Incorrect username or password")
|
||||
|
||||
# Create access token
|
||||
token_data = auth_manager.create_access_token(user)
|
||||
|
||||
logger.info(f"User logged in: {user.username}")
|
||||
|
||||
return LoginResponse(
|
||||
access_token=token_data["access_token"],
|
||||
token_type=token_data["token_type"],
|
||||
expires_in=token_data["expires_in"],
|
||||
user=UserResponse.model_validate(user)
|
||||
)
|
||||
|
||||
|
||||
@router.get("/me", response_model=UserResponse)
|
||||
def get_current_user_info(current_user: User = Depends(get_current_user)):
|
||||
"""Get current user information"""
|
||||
return UserResponse.model_validate(current_user)
|
||||
Reference in New Issue
Block a user