refractoring

mock_backend/mock_api.py

@@ -0,0 +1,66 @@
+from fastapi import FastAPI, HTTPException
+from mock_backend.mock_backend import announcement_system, EndpointGroup
+from typing import List
+import uvicorn
+
+app = FastAPI()
+
+@app.get("/groups", response_model=List[EndpointGroup])
+def get_groups():
+    return announcement_system.get_endpoint_groups()
+
+@app.post("/groups", response_model=EndpointGroup)
+def create_group(group: EndpointGroup):
+    try:
+        return announcement_system.add_endpoint_group(group)
+    except ValueError as e:
+        raise HTTPException(status_code=400, detail=str(e))
+
+@app.put("/groups/{group_id}", response_model=EndpointGroup)
+def update_group(group_id: int, updated_group: EndpointGroup):
+    try:
+        return announcement_system.update_endpoint_group(group_id, updated_group)
+    except ValueError as e:
+        raise HTTPException(status_code=404, detail=str(e))
+
+@app.delete("/groups/{group_id}")
+def delete_group(group_id: int):
+    try:
+        announcement_system.delete_endpoint_group(group_id)
+        return {"message": "Group deleted successfully"}
+    except ValueError as e:
+        raise HTTPException(status_code=404, detail=str(e))
+
+@app.post("/announcements")
+def start_announcement(text: str, group_id: int):
+    group = announcement_system.get_endpoint_group(group_id)
+    if not group:
+        raise HTTPException(status_code=404, detail=f"Group with ID {group_id} not found")
+    
+    try:
+        announcement_system.start_announcement(text, group)
+        return {"message": "Announcement started successfully"}
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=str(e))
+
+@app.get("/announcements/status")
+def get_announcement_status():
+    process = announcement_system.current_process
+    return {
+        "state": process.current_state.value,
+        "progress": process.progress,
+        "error": process.error,
+        "details": {
+            "text": process.details.text,
+            "languages": process.details.languages,
+            "group": process.details.group,
+            "start_time": process.details.start_time
+        }
+    }
+
+@app.get("/endpoints")
+def get_available_endpoints():
+    return announcement_system.available_endpoints
+
+if __name__ == "__main__":
+    uvicorn.run('mock_backend.mock_api:app', host="0.0.0.0", port=7999, reload=True)