algorithms.py 942 B

12345678910111213141516171819202122232425262728293031
  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 AttendeesRecommendationRequest, AttendeesRecommendationResponse
  6. router = APIRouter()
  7. @router.post('/graph-coloring', response_model=GraphColoringResponse)
  8. async def get_graph_coloring_result(graph: GraphColoringRequest):
  9. is_solvable, colored = await get_graph_coloring(graph.graph)
  10. solution = {
  11. 'is_solvable': is_solvable,
  12. 'colored': colored
  13. }
  14. return solution
  15. @router.post('/attendees-recommendation', response_model=AttendeesRecommendationResponse)
  16. async def get_recommended(meeting_info: AttendeesRecommendationRequest):
  17. return {
  18. 'code': 200,
  19. 'message': 'Test',
  20. 'data': {
  21. 'userIdList': []
  22. }
  23. }