Initial commit: SIBU 2.0 MISSION

backend/app/api/taxis/__init__.py

@@ -0,0 +1,145 @@
+from fastapi import APIRouter, Depends, Query, HTTPException, UploadFile, File, Form
+from sqlmodel import Session, select
+from typing import Optional
+import os
+import shutil
+from uuid import uuid4
+from app.core.database import get_session
+from app.models.taxi import Taxi
+from app.api.deps import get_current_admin
+
+router = APIRouter(prefix="/api/taxis", tags=["taxis"])
+
+UPLOAD_DIR = "uploads"
+
+
+@router.get("")
+async def get_taxis(
+    corregimiento: Optional[str] = Query(None),
+    shift: Optional[str] = Query(None),
+    english_speaking: Optional[bool] = Query(None),
+    is_active: Optional[bool] = Query(None),
+    session: Session = Depends(get_session)
+):
+    """Get all taxis with optional filters."""
+    statement = select(Taxi)
+    
+    if is_active is not None:
+        statement = statement.where(Taxi.is_active == is_active)
+    
+    if corregimiento:
+        statement = statement.where(Taxi.corregimiento.contains(corregimiento))
+    
+    if shift:
+        statement = statement.where(Taxi.shift == shift)
+    
+    if english_speaking is not None:
+        statement = statement.where(Taxi.english_speaking == english_speaking)
+        
+    taxis = session.exec(statement).all()
+    return taxis
+
+
+@router.post("")
+async def create_taxi(
+    owner_name: str = Form(...),
+    phone_number: str = Form(...),
+    license_plate: str = Form(...),
+    corregimiento: str = Form(...),
+    shift: str = Form(...),
+    cooperative: Optional[str] = Form(None),
+    rating: float = Form(5.0),
+    english_speaking: bool = Form(False),
+    is_active: bool = Form(True),
+    image: Optional[UploadFile] = File(None),
+    session: Session = Depends(get_session),
+    _: bool = Depends(get_current_admin)
+):
+    """Create a new taxi entry (Admin only)."""
+    image_url = None
+    if image:
+        ext = os.path.splitext(image.filename)[1]
+        filename = f"{uuid4()}{ext}"
+        path = os.path.join(UPLOAD_DIR, "profiles", filename)
+        with open(path, "wb") as buffer:
+            shutil.copyfileobj(image.file, buffer)
+        image_url = f"/uploads/profiles/{filename}"
+    
+    taxi = Taxi(
+        owner_name=owner_name,
+        phone_number=phone_number,
+        license_plate=license_plate,
+        cooperative=cooperative,
+        corregimiento=corregimiento,
+        shift=shift,
+        rating=rating,
+        english_speaking=english_speaking,
+        image_url=image_url,
+        is_active=is_active
+    )
+    session.add(taxi)
+    session.commit()
+    session.refresh(taxi)
+    return taxi
+
+
+@router.put("/{taxi_id}")
+async def update_taxi(
+    taxi_id: str,
+    owner_name: str = Form(...),
+    phone_number: str = Form(...),
+    license_plate: str = Form(...),
+    corregimiento: str = Form(...),
+    shift: str = Form(...),
+    cooperative: Optional[str] = Form(None),
+    rating: float = Form(5.0),
+    english_speaking: bool = Form(False),
+    is_active: bool = Form(True),
+    image: Optional[UploadFile] = File(None),
+    session: Session = Depends(get_session),
+    _: bool = Depends(get_current_admin)
+):
+    """Update a taxi entry (Admin only)."""
+    db_taxi = session.get(Taxi, taxi_id)
+    if not db_taxi:
+        raise HTTPException(status_code=404, detail="Taxi not found")
+    
+    # Update fields
+    db_taxi.owner_name = owner_name
+    db_taxi.phone_number = phone_number
+    db_taxi.license_plate = license_plate
+    db_taxi.corregimiento = corregimiento
+    db_taxi.shift = shift
+    db_taxi.cooperative = cooperative
+    db_taxi.rating = rating
+    db_taxi.english_speaking = english_speaking
+    db_taxi.is_active = is_active
+    
+    # Handle image upload
+    if image:
+        ext = os.path.splitext(image.filename)[1]
+        filename = f"{uuid4()}{ext}"
+        path = os.path.join(UPLOAD_DIR, "profiles", filename)
+        with open(path, "wb") as buffer:
+            shutil.copyfileobj(image.file, buffer)
+        db_taxi.image_url = f"/uploads/profiles/{filename}"
+    
+    session.add(db_taxi)
+    session.commit()
+    session.refresh(db_taxi)
+    return db_taxi
+
+
+@router.delete("/{taxi_id}")
+async def delete_taxi(
+    taxi_id: str,
+    session: Session = Depends(get_session),
+    _: bool = Depends(get_current_admin)
+):
+    """Delete a taxi entry (Admin only)."""
+    db_taxi = session.get(Taxi, taxi_id)
+    if not db_taxi:
+        raise HTTPException(status_code=404, detail="Taxi not found")
+    session.delete(db_taxi)
+    session.commit()
+    return {"ok": True}