# note: https://responder.kennethreitz.org/en/latest/search.html?q=resp.content
from hub import methods
import traceback
import requests
import importlib
def generate_app():
""""""
# --- define app --- see: https://responder.kennethreitz.org/en/latest/tour.html#cors
import responder
app = responder.API(
# cors=True,
# allow_methods=['*'],
# allow_credentials=True,
)
async def coroutine_method(method, _params):
"""协程方法"""
# run_at = time.time()
result = await method(**_params)
# methods.debug_log(f"app.coroutine_method", f"use time {round(time.time() - run_at, 2)}s")
return result
def foreground_method(method, _params):
"""等待返回"""
# run_at = time.time()
result = method(**_params)
# methods.debug_log(f"app.foreground_method", f"use time {round(time.time() - run_at, 2)}s")
return result
@app.background.task
def background_method(method, _params):
"""不等返回"""
# run_at = time.time()
result = method(**_params)
# methods.debug_log(f"app.background_method", f"use time {round(time.time() - run_at, 2)}s")
return result
async def run_method(code, _params):
config_dict = {
7000: ('v1', 'f7000', 'IsAsync'),
70000: ('v1', 'f70000', 'IsBack'),
}
if code not in config_dict:
return dict(code=2, detail=f"{code} not found!")
else:
script_name, method_name, method_type = config_dict[code]
script = importlib.import_module(f"api.httpapi.{script_name}")
method = getattr(script, method_name)
del _params['code']
if method_type == 'IsAsync':
return await coroutine_method(method, _params)
elif method_type == 'IsBack':
return foreground_method(method, _params)
elif method_type == 'NoBack':
return background_method(method, _params)
@app.route('/httpapi')
async def httpapi(request, response):
try:
# --- fill ---
if request.params:
params = dict(request.params.items())
else:
params = await request.media()
# --- debug ---
methods.debug_log('app.httpapi92', f"#params: {params}")
# --- check ---
if 'code' not in params:
response.media = dict(code=-1, reason="error98", detail=f"参数缺失")
else:
code = params.get('code')
result = await run_method(code, params)
methods.debug_log('app.httpapi102', f"#result: {result}")
if result.__class__.__name__ == 'bytes':
response.content = result
elif result.__class__.__name__ == 'str':
response.text = result
elif result.__class__.__name__ == 'dict':
response.media = result
elif result.__class__.__name__ == 'Future':
response.media = dict(code=0, detail=f"{result} is running.")
elif not result:
response.media = dict(code=0, detail=f"return is null.")
else:
response.media = dict(code=0, detail=f"return type is {result.__class__.__name__}.")
except Exception as exception:
methods.debug_log("app.httpapi118", f"#exception: {exception.__class__.__name__}")
methods.debug_log("app.httpapi118", f"#traceback: {traceback.format_exc()}")
response.media = dict(code=-1, reason=exception.__class__.__name__, detail=f"{methods.trace_log()}")
@app.route('/wsapi', websocket=True)
async def wsapi(client):
"""
let wsuri = `${location.protocol === 'https' ? 'wss' : 'ws'}://${location.host}:8801/wsapi`;
connect_json_text = {
line_id: 连接id，暂为token
}
)
send_json_text_1 = {
face_uuid: 匹配到的人脸id
face_name: 人脸名称
input_face_b64: 抓拍图片（b64格式字符串）
}
"""
try:
await client.accept()
while True:
# --- check ---
data = await client.receive_text() # 消息接受方法 receive_{text/json/bytes}
if not data:
methods.debug_log('app.wsapi140', f"#reason: error140")
await client.send_json({'code': -1, 'reason': 'error140'})
break
# --- debug ---
# methods.debug_log('app.wsapi145', f"#data: {data} | {type(data)}")
# --- check ---
if not methods.is_json(data):
methods.debug_log('app.wsapi133', f"#reason: error146")
await client.send_json({'code': -1, 'reason': 'error146'})
break
# --- check ---
data = methods.json_loads(data)
if not data.get('code'):
methods.debug_log('app.wsapi144', f"#reason: error157")
await client.send_json({'code': -1, 'reason': 'error157'})
break
# --- debug ---
if data.get('code') not in [2008]:
methods.debug_log('app.wsapi160', f"#data: {data}")
# --- check 2000 ---
code = data.get('code')
if code == 2000:
# --- check ---
if 'acct' not in data or 'pw' not in data:
methods.debug_log('app.wsapi166', f"#reason: error164")
await client.send_json({'code': -1, 'reason': 'error164'})
break
# --- check ---
username = data.get('acct')
password = data.get('pw')
url = 'http://58.34.98.13:8099/api/ar/user/login'
data = {
"acct": username,
"pw": password,
"method": "1" # 客户端类型 1 安卓 2 PC端
}
methods.debug_log('app.wsapi179', f"#url: {url}")
response = requests.post(url=url, json=data)
data = response.json()
if data.get('msg') and data.get('msg') == 'success':
data['code'] = 4000
await client.send_json(data)
else:
methods.debug_log('app.wsapi186', f"#reason: error186")
await client.send_json({'code': -1, 'reason': 'error186'})
# --- save ---
# methods.debug_log('app.wsapi179', f"{dir(client)}")
# methods.debug_log('app.wsapi179', f"#id: {id(client)} | {type(id(client))}")
methods.debug_log('app.wsapi179', f"{type(client)} | {methods.now_ts()} | {id(client)}")
LineManage.line_dict[username] = client, methods.now_ts(), id(client)
# --- check 2008 ---
if code == 2008:
# --- 判断该连接是否是已保存的链接 ---
is_ok = False
for key in list(LineManage.line_dict.keys()):
ws, ts, _id = LineManage.line_dict.get(key)
if _id == id(client):
LineManage.line_dict[username] = client, methods.now_ts(), id(client)
is_ok = True
await client.send_json({'code': 4008, 'result': is_ok})
except Exception as exception:
if exception.__class__.__name__ == 'WebSocketDisconnect':
await client.close()
else:
methods.debug_log("app.wsapi163", f"#exception: {exception.__class__.__name__}")
methods.debug_log("app.wsapi163", f"#traceback: {traceback.format_exc()}")
# --- return ---
return app