116 lines
4.5 KiB
Python
116 lines
4.5 KiB
Python
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)) |