from starlette.formparsers import MultiPartParser as StartletteMultiPartParser from fastapi import APIRouter, Request, Response, Depends from fastapi.responses import FileResponse from key import v1 as key_v1 from hub import methods, Global import importlib router = APIRouter() config_dict = { 'v5': { # --- 关于渣包车 --- 2001: 'v5.hs2000.code_2001', # 获取全部渣包车状态接口 2002: 'v5.hs2000.code_2002', # 指定渣包车点火操作接口 2003: 'v5.hs2000.code_2003', # 指定渣包车熄火操作接口 2004: 'v5.hs2000.code_2004', # 指定渣包车建立远程操作权限（或是切换）接口 # 2005: 'v5.hs2000.code_2005', # 创建并执行自动驾驶任务【已停用】 2101: 'v5.hs2000.code_2101', # 新增渣包车接口 2102: 'v5.hs2000.code_2102', # 修改渣包车接口 2103: 'v5.hs2000.code_2103', # 删除渣包车接口 # --- 关于关于任务 --- 3001: 'v5.hs3000.code_3001', # 任务列表数据获取接口（分页） # 3002: 'v5.hs3000.code_3002', # 任务暂停接口【已停用】 3003: 'v5.hs3000.code_3003', # 任务取消消接口 3004: 'v5.hs3000.code_3004', # 任务创建并执行接口 3005: 'v5.hs3000.code_3005', # 获取指定任务信息 # --- 关于场地数据 --- 4001: 'v5.hs4000.code_4001', # 获取全部渣包状态数据接口 # 4002: 'v5.hs4000.code_4002', # todo 获取接渣口状态数据 # 4003: 'v5.hs4000.code_4003', # todo 获取倒渣口状态数据 # --- 关于告警 --- 5001: 'v5.hs5000.code_5001', # 获取告警数据列表接口 } } methods_dict = {} # {: {: }} def _get_method_by_code(code, tag): """ 通过code获取method（不调用不加载） """ try: # --- check --- if tag not in methods_dict: methods_dict[tag] = {} # --- check --- if code in methods_dict.get(tag): return methods_dict.get(tag).get(code) # --- get method --- method_path = config_dict.get(tag).get(code) tag, file_name, method_name = method_path.split('.') script = importlib.import_module(f"api.{tag}.{file_name}") method = getattr(script, method_name) # --- set method --- methods_dict[tag][code] = method return method except Exception as exception: methods.debug_log('api._get_method_by_code.69', f"#exception: {exception.__class__.__name__}") methods.debug_log('api._get_method_by_code.69', f"#traceback: {methods.trace_log()}") @router.post('/{tag}/api') async def post(tag: str, request: Request, response: Response, user: dict = Depends(key_v1.login_required)): """""" # --- check --- if not user.get('skip_is'): response.headers['authorization'] = key_v1.get_token_by_user(user) # --- get sources --- sources = await request.json() # tag = sources.get('tag') code = int(sources.get('code')) page = sources.get('page', 1) size = sources.get('size', 10) ban_keys = ['code'] # --- set global args --- sources = {key: val for key, val in sources.items() if key not in ban_keys} sources['g_user_id'] = user.get('uid') sources['g_user_name'] = user.get('username') sources['g_user_pass'] = user.get('password') sources['g_role_id'] = user.get('role_id') # --- call action --- try: run_at = methods.now_ts() method = _get_method_by_code(code=code, tag=tag) result = await method(**sources) methods.debug_log(f"api.post.101", f"#{tag}/{code}: use time {round(methods.now_ts() - run_at, 2)}s") # --- check --- if code in [ # 8201, # 访客记录，列表 ] and result.__class__.__name__ == 'dict': return dict( code=result.get('code'), data=result.get('data', [])[(page - 1) * size: page * size], page=page, size=size, total=len(result.get('data', [])), ) elif result.__class__.__name__ in ['FileResponse', 'dict']: return result else: return dict(data=result, type=result.__class__.__name__) except Exception as exception: methods.debug_log('api.post.121', f"#tag: {tag}, #code: {code}") methods.debug_log('api.post.121', f"#exception: {exception.__class__.__name__}") methods.debug_log('api.post.121', f"#traceback: {methods.trace_log()}") return dict(code=-1, data=[], reason=f"{exception.__class__.__name__}", detail=f"{methods.trace_log()}") @router.put('/actions') async def upload(request: Request, user: dict = Depends(key_v1.login_required)): try: # --- get sources --- sources = dict() parser = StartletteMultiPartParser(request.headers, request.stream()) params = await parser.parse() # sources['params'] = params # --- set sources --- fromdata列表处理 for key, value in params.multi_items(): # --- check list --- if key[-4:] == 'list' and key not in sources: sources[key] = list() if key[-4:] == 'list': sources[key].append(value) # --- make list --- else: if key in sources: sources[key] = [sources[key]] sources[key].append(value) else: sources[key] = value # --- clean --- tag = sources.get('tag', 'v1') code = int(sources.get('code')) del sources['tag'] del sources['code'] # --- set sources --- sources['g_user_id'] = user.get('uid') sources['g_user_name'] = user.get('username') sources['g_user_pass'] = user.get('password') sources['g_role_id'] = user.get('role_id') # --- set files --- files = dict() for key, value in sources.items(): if value.__class__.__name__ == 'UploadFile': files[key] = value for key, value in files.items(): del sources[key] sources['files'] = files # --- call action --- return await _get_method_by_code(code=code, tag=tag)(**sources) except Exception as exception: methods.debug_log('api.upload', f"m-299: exception | {exception.__class__.__name__}") methods.debug_log('api.upload', f"m-299: traceback | {methods.trace_log()}") return dict(code=-1, reason=f"{exception.__class__.__name__}", detail=f"{methods.trace_log()}") @router.get('/api') async def download(request: Request, response: Response, user: dict = Depends(key_v1.login_required)): try: parser = StartletteMultiPartParser(request.headers, request.stream()) params = await parser.parse() instance_id = params.get('instance_id') report = Global.mdb.get_one_by_id('', instance_id) report_path = report.get('report') if report else '' if not report_path: return '' return FileResponse(report_path) except Exception as exception: methods.debug_log('api.download.201', f"#exception: {exception.__class__.__name__}") methods.debug_log('api.download.201', f"#traceback: {methods.trace_log()}") return dict(code=-1, reason=f"{exception.__class__.__name__}", detail=f"{methods.trace_log()}")