CIA/e-voting-system/backend/routes/elections.py

"""
Routes pour les élections et les candidats.
"""

from fastapi import APIRouter, HTTPException, status, Depends
from sqlalchemy.orm import Session
from .. import schemas, services
from ..dependencies import get_db, get_current_voter
from ..models import Voter

router = APIRouter(prefix="/api/elections", tags=["elections"])


@router.get("/active", response_model=schemas.ElectionResponse)
def get_active_election(db: Session = Depends(get_db)):
    """Récupérer l'élection active en cours"""
    
    election = services.ElectionService.get_active_election(db)
    
    if not election:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail="No active election"
        )
    
    return election


@router.get("/completed")
def get_completed_elections(db: Session = Depends(get_db)):
    """Récupérer tous les votes passés/terminés"""
    
    from datetime import datetime
    elections = db.query(services.models.Election).filter(
        services.models.Election.end_date < datetime.utcnow(),
        services.models.Election.results_published == True
    ).all()
    
    return elections


@router.get("/upcoming")
def get_upcoming_elections(db: Session = Depends(get_db)):
    """Récupérer tous les votes à venir"""
    
    from datetime import datetime
    elections = db.query(services.models.Election).filter(
        services.models.Election.start_date > datetime.utcnow()
    ).all()
    
    return elections


@router.get("/active/results")
def get_active_election_results(db: Session = Depends(get_db)):
    """Récupérer les résultats de l'élection active"""
    
    election = services.ElectionService.get_active_election(db)
    
    if not election:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail="No active election"
        )
    
    results = services.VoteService.get_election_results(db, election.id)
    
    return results


@router.get("/{election_id}", response_model=schemas.ElectionResponse)
def get_election(election_id: int, db: Session = Depends(get_db)):
    """Récupérer une élection par son ID"""
    
    election = services.ElectionService.get_election(db, election_id)
    
    if not election:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail="Election not found"
        )
    
    return election


@router.get("/{election_id}/results", response_model=schemas.ElectionResultResponse)
def get_election_results(
    election_id: int,
    db: Session = Depends(get_db)
):
    """
    Récupérer les résultats d'une élection.
    Disponible après la fermeture du scrutin.
    """
    
    election = services.ElectionService.get_election(db, election_id)
    
    if not election:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail="Election not found"
        )
    
    if not election.results_published:
        raise HTTPException(
            status_code=status.HTTP_403_FORBIDDEN,
            detail="Results not yet published"
        )
    
    results = services.VoteService.get_election_results(db, election_id)
    
    return schemas.ElectionResultResponse(
        election_id=election.id,
        election_name=election.name,
        total_votes=sum(r.vote_count for r in results),
        results=results
    )


@router.post("/{election_id}/publish-results")
def publish_results(
    election_id: int,
    db: Session = Depends(get_db)
):
    """
    Publier les résultats d'une élection (admin only).
    À utiliser après la fermeture du scrutin.
    """
    
    election = services.ElectionService.get_election(db, election_id)
    
    if not election:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail="Election not found"
        )
    
    # Marquer les résultats comme publiés
    election.results_published = True
    db.commit()
    
    return {
        "message": "Results published successfully",
        "election_id": election.id,
        "election_name": election.name
    }


@router.get("/completed", response_model=list[schemas.ElectionResponse])
def get_completed_elections(db: Session = Depends(get_db)):
    """Récupérer toutes les élections terminées (archives)"""
    from datetime import datetime
    from .. import models
    
    completed = db.query(models.Election).filter(
        models.Election.end_date < datetime.utcnow(),
        models.Election.results_published == True
    ).all()
    
    return completed


@router.get("/upcoming", response_model=list[schemas.ElectionResponse])
def get_upcoming_elections(db: Session = Depends(get_db)):
    """Récupérer toutes les élections futures"""
    from datetime import datetime
    from .. import models
    
    upcoming = db.query(models.Election).filter(
        models.Election.start_date > datetime.utcnow()
    ).all()
    
    return upcoming