0DMRMaster version 0.4

api/dmrapi.py

@@ -0,0 +1,100 @@
+import uvicorn
+
+from datetime import datetime
+from fastapi import FastAPI
+from fastapi.staticfiles import StaticFiles
+
+from dmrtools.dispatcher import Dispatcher
+
+
+class DMRApiHelper:
+    CALLS_MAX = 50
+
+    dispatcher: Dispatcher|None = None
+
+    @classmethod
+    def get_peers(cls) -> list:
+        if cls.dispatcher is None:
+            return list()
+
+        peers = sorted(cls.dispatcher.peer_keeper.peers,
+                       key=lambda p: p.connect_time)
+
+        peers_info = [{
+            "name": p.name,
+            "peer_id": p.peer_id,
+            "addr": p.addr_str,
+            "status": p.status.name,
+            "connect_time": p.connect_time,
+            "active_time": p.active_time,
+            "units": list(p.units.keys()),
+            "config": p.config} for p in peers]
+
+        return peers_info
+
+    @classmethod
+    def get_calls(cls) -> list:
+        if cls.dispatcher is None:
+            return list()
+
+        calls = sorted(cls.dispatcher.call_keeper.calls_log,
+                       key=lambda c: c.start_time, reverse=True)
+
+        # get first CALLS_MAX if more than
+        if len(calls) > cls.CALLS_MAX:
+            calls = calls[:cls.CALLS_MAX]
+
+        calls_info = [{
+            "call_id": c.call_id,
+            "dir": f"{c.src_id}->{c.dst_hr}",
+            "src_id": c.src_id,
+            "dst_id": c.dst_id,
+            "peer_id": c.peer_id,
+            "call_type": c.call_type.name,
+            "start_time": c.start_time,
+            "last_packet_time": c.last_packet_time,
+            "is_ended": c.is_ended,
+            "end_time": c.end_time,
+            "broadcast": c.route_to is None,
+            "route_to": (list(peer.name for peer in c.route_to)
+                         if c.route_to is not None else []),
+            "time": f"{c.time:.1f}s"} for c in calls]
+
+        return calls_info
+
+
+# Define FastAPI app
+app = FastAPI()
+
+
+# Mount the "dashboard" directory at /dashboard
+app.mount("/dashboard", StaticFiles(directory="dashboard"), name="dashboard")
+
+
+@app.get("/api/dashboard")
+async def get_dashboard():
+    return {"success": True,
+            "peers": DMRApiHelper.get_peers(),
+            "calls": DMRApiHelper.get_calls()}
+
+
+@app.get("/api/peers")
+async def get_peers():
+    return {"success": True, "peers": DMRApiHelper.get_peers()}
+
+
+@app.get("/api/calls")
+async def get_calls():
+    return {"success": True, "calls": DMRApiHelper.get_calls()}
+
+
+async def start_api(host: str, port: int, dispatcher: Dispatcher) -> None:
+    DMRApiHelper.dispatcher = dispatcher
+    config = uvicorn.Config(app, host=host, port=port, log_level="error",
+                            loop="asyncio")
+    server = uvicorn.Server(config)
+
+    try:
+        await server.serve()
+    except Exception as e:
+        print(f"start_api: Exception {e}")