41 lines
939 B
Python
41 lines
939 B
Python
import uvicorn
|
|
|
|
import uiautomator2 as u2
|
|
from app import *
|
|
from api.base import *
|
|
|
|
@app.get("/health", response_model=Response)
|
|
def health():
|
|
return Response()
|
|
|
|
|
|
@app.post("/device/connect", response_model=Response)
|
|
def connect_device(item: Device):
|
|
try:
|
|
d = u2.connect(item.serial)
|
|
except Exception as e:
|
|
return Response(code=500, msg=str(e))
|
|
print(d.info)
|
|
print(d.device_info)
|
|
devices[item.serial] = d
|
|
return Response(data=d.device_info)
|
|
|
|
|
|
@app.post("/device/size", response_model=Response)
|
|
def get_device_size(item: Device):
|
|
if item.serial not in devices:
|
|
return Response(code=500, msg="设备 {} 不存在".format(item.serial))
|
|
|
|
d = devices[item.serial]
|
|
width, height = d.window_size()
|
|
return Response(data={
|
|
"width": width,
|
|
"heighth": height,
|
|
})
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
uvicorn.run('main:app', host='0.0.0.0', port=8000, reload=True)
|