algorithms.py 978 B

1234567891011121314151617181920212223242526272829303132
  1. from fastapi import APIRouter
  2. from loguru import logger
  3. from app.controllers.algorithms.graph_coloring import get_graph_coloring
  4. from app.models.domain.algorithms import GraphColoringRequest, GraphColoringResponse
  5. from app.models.domain.algorithms import (
  6. AttendeesRecommendationRequest,
  7. AttendeesRecommendationResponse,
  8. )
  9. router = APIRouter()
  10. @router.get("/status")
  11. async def get_living_condition():
  12. return {"message": "alive"}
  13. @router.post("/graph-coloring", response_model=GraphColoringResponse)
  14. async def get_graph_coloring_result(graph: GraphColoringRequest):
  15. is_solvable, colored = await get_graph_coloring(graph.graph)
  16. solution = {"is_solvable": is_solvable, "colored": colored}
  17. return solution
  18. @router.post(
  19. "/attendees-recommendation", response_model=AttendeesRecommendationResponse
  20. )
  21. async def get_recommended(meeting_info: AttendeesRecommendationRequest):
  22. return {"code": 200, "message": "Test", "data": {"userIdList": []}}