mirror of
http://git.xinwangdao.com/cnnc-embedded-parts-detect/detect-gui.git
synced 2025-06-24 13:14:11 +08:00
62 lines
2.6 KiB
Python
62 lines
2.6 KiB
Python
|
import json
|
|||
|
|
|
|||
|
|
from starlette.requests import Request
|
|||
|
|
from starlette.responses import JSONResponse
|
|||
|
|
|
|||
|
|
from core.edge_util import camel_to_snake
|
|||
|
|
from core.config import settings
|
|||
|
|
|
|||
|
|
|
|||
|
|
def setup_routes(context, app):
|
|||
|
|
@app.get("/")
|
|||
|
|
async def system_info():
|
|||
|
|
info = context.get_component("_database").system_info()
|
|||
|
|
return {"code": 0, "message": "success", "result": info}
|
|||
|
|
|
|||
|
|
@app.get("/components")
|
|||
|
|
async def components():
|
|||
|
|
return {"code": 0, "message": "success", "result": list(context.components.keys())}
|
|||
|
|
|
|||
|
|
@app.middleware("http")
|
|||
|
|
async def add_process_time_header(request: Request, call_next):
|
|||
|
|
if request.url.path.startswith("/service"):
|
|||
|
|
req_body = await request.body()
|
|||
|
|
req_body = req_body.decode("utf-8")
|
|||
|
|
if req_body == '':
|
|||
|
|
req_body = "{}"
|
|||
|
|
req = json.loads(req_body)
|
|||
|
|
req.update(request.query_params)
|
|||
|
|
method = req.get('method')
|
|||
|
|
component = req.get('component')
|
|||
|
|
method = camel_to_snake(method)
|
|||
|
|
if req.get("method") is None:
|
|||
|
|
response = {"code": -1, "type": "service", "message": "没有method参数！"}
|
|||
|
|
else:
|
|||
|
|
if component == "" or component is None:
|
|||
|
|
executor = context
|
|||
|
|
else:
|
|||
|
|
executor = context.get_component(component)
|
|||
|
|
if executor:
|
|||
|
|
try:
|
|||
|
|
kwargs = {key: value for key, value in req.items() if key not in ["method", "component"]}
|
|||
|
|
result = await executor.execute(method, **kwargs)
|
|||
|
|
response = {"code": 0, "type": "service", "message": "success", "result": result}
|
|||
|
|
except Exception as e:
|
|||
|
|
response = {"code": -1, "type": "service", "message": str(e)}
|
|||
|
|
else:
|
|||
|
|
response = {"code": -2, "type": "service", "message": f"Executor {component} not found"}
|
|||
|
|
return JSONResponse(response)
|
|||
|
|
elif request.url.path.startswith("/task"):
|
|||
|
|
req_body = await request.body()
|
|||
|
|
req_body = req_body.decode("utf-8")
|
|||
|
|
if req_body == '':
|
|||
|
|
req_body = "{}"
|
|||
|
|
req = json.loads(req_body)
|
|||
|
|
try:
|
|||
|
|
result = await context.execute_task(**req)
|
|||
|
|
response = {"code": 0, "type": "task", "message": "success", "result": result}
|
|||
|
|
except Exception as e:
|
|||
|
|
response = {"code": -1, "type": "task", "message": str(e)}
|
|||
|
|
return JSONResponse(response)
|
|||
|
|
else:
|
|||
|
|
return await call_next(request)
|