from fastapi import APIRouter, Depends, HTTPException, Query, Header, status from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer from sqlalchemy.orm import Session from typing import Optional, List # from database import get_db from app.core.auth import get_current_user from app.modules.users.models import Users # Modelo base de usuario from app.modules.location.models import Locations from app.modules.location.schema import LocationBase, LocationUpdate, LocationResponse, messageResponse from app.modules.location.service import LocationsService router = APIRouter(prefix="/locations", tags=["locations"]) security = HTTPBearer() #================ Create ==================== @router.post("/create", response_model=LocationResponse, status_code=status.HTTP_201_CREATED) async def create_locations( data: LocationBase, db: Session = Depends(get_db), current_user: Users = Depends(get_current_user) ): """Create a locations - Requires authentication""" if not current_user: raise HTTPException(status_code=401, detail="User not found - invalid token") if current_user.tipo_usuario not in ["root", "admin"]: raise HTTPException(status_code=403, detail="You don't have permission to create locations") try: result = LocationsService.create_locations(db=db, data=data) return result except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) #================ Get ==================== @router.get("/", response_model=List[LocationResponse]) def get_locationss( db: Session = Depends(get_db), current_user: Users = Depends(get_current_user), skip: int = Query(0, ge=0, description="Number of records to skip"), limit: int = Query(100, ge=1, le=1000, description="Max number of records to return") ): """Get list with pagination - requires authentication""" if not current_user: raise HTTPException(status_code=401, detail="Authentication required") return LocationsService.get_locations(db=db, skip=skip, limit=limit) #================ Get by ID ==================== @router.get("/{{{entity_name}}_id}", response_model=LocationResponse) def get_locations_by_id( locations_id: int, db: Session = Depends(get_db), current_user: Users = Depends(get_current_user) ): """Get locations by ID - requires authentication""" if not current_user: raise HTTPException(status_code=401, detail="Authentication required") try: result = LocationsService.get_locations(db=db, locations_id=locations_id, current_user=current_user) return result except ValueError as e: raise HTTPException(status_code=404, detail=str(e)) #================ Update ==================== @router.patch("/update/{{{entity_name}}_id}", response_model=LocationResponse) async def update_locations( locations_id: int, data: LocationUpdate , db: Session = Depends(get_db), current_user: Users = Depends(get_current_user) ): """Update existing locations - requires authentication""" if not current_user: raise HTTPException(status_code=401, detail="Authentication required") if current_user.tipo_usuario not in ["root", "admin"]: raise HTTPException(status_code=403, detail="You don't have permission to update locations") try: result = LocationsService.update_locations( db=db, locations_id=locations_id, data=data, current_user=current_user ) return result except ValueError as e: raise HTTPException(status_code=404, detail=str(e)) #================ Delete ==================== @router.delete("/delete/{{{entity_name}}_id}", response_model=messageResponse ,status_code=status.HTTP_200_OK) async def delete_locations( locations_id: int, db: Session = Depends(get_db), current_user: Users = Depends(get_current_user) ): """Delete locations by ID - requires authentication""" if not current_user: raise HTTPException(status_code=401, detail="Authentication required") if current_user.tipo_usuario not in ["root", "admin"]: raise HTTPException(status_code=403, detail="You don't have permission to delete locations") try: result = LocationsService.delete_locations(db=db, locations_id=locations_id, current_user=current_user) return {"message": "locations deleted successfully"} except ValueError as e: raise HTTPException(status_code=404, detail=str(e))