58 lines
1.8 KiB
Python
58 lines
1.8 KiB
Python
import asyncio
|
|
import traceback
|
|
|
|
from fastapi import APIRouter, Request
|
|
|
|
from common import *
|
|
from models import DataModel
|
|
from settings import settings
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
def dest_partition(s: str) -> int:
|
|
try:
|
|
return int(s[-18:], 16) % 16
|
|
except:
|
|
return 0
|
|
|
|
|
|
@router.post("/")
|
|
async def point(request: Request, items: DataModel):
|
|
try:
|
|
public_data, _ = restore_field(items.public)
|
|
rdb = request.app.state.redis
|
|
data_factory = request.app.state.data_factory
|
|
output_factory = request.app.state.output_factory
|
|
appid = items.public.get('b01')
|
|
if not appid:
|
|
return {'code': -1, 'msg': '必须要appid'}
|
|
output_factory.name = settings.OUTPUT_NAME.get(appid)
|
|
partition = dest_partition(items.public.get('x02') or '0')
|
|
|
|
if not output_factory.name:
|
|
return {'code': -1, 'msg': '没有匹配到appid'}
|
|
for item in items.data:
|
|
data, sign_data = restore_field(item)
|
|
_ = data.pop('sign')
|
|
properties = data.pop('properties')
|
|
|
|
#
|
|
sign = sign_data.pop('sign')
|
|
sign_properties = sign_data.pop('properties')
|
|
|
|
if not check_sign(sign, settings.SALT.get(appid, ''), sign_data, sign_properties):
|
|
# continue
|
|
return {"code": -1, 'msg': '签名错误'}
|
|
single_data = dict(**public_data, **data, properties=properties)
|
|
check_preset(single_data)
|
|
await asyncio.gather(
|
|
*map(lambda o: asyncio.create_task(o(request, rdb, single_data)), data_factory.handler_link))
|
|
msg = data_factory.format_data(single_data)
|
|
output_factory.send(msg, partition)
|
|
|
|
return {"code": 0, 'msg': 'ok'}
|
|
except Exception as e:
|
|
msg = traceback.format_exc()
|
|
print(msg)
|