diff --git a/handler/__init__.py b/handler/__init__.py deleted file mode 100644 index e69de29..0000000 diff --git a/handler/http_server.py b/handler/http_server.py deleted file mode 100644 index c2fccb1..0000000 --- a/handler/http_server.py +++ /dev/null @@ -1,48 +0,0 @@ -import json - -from flask import Flask, request, Response -from handler.common import MESSAGE_QUEUE, MessagePayload -import logging -# 不要日志,当它不存在 -log = logging.getLogger('werkzeug') -log.setLevel(logging.ERROR) - -app = Flask(__name__) -app.config['JSON_AS_ASCII'] = False - - -@app.post("/message") -def message_from_mitmproxy(): - payload = MessagePayload(request.data, request.headers.get("X-MITM-TS", "")) - MESSAGE_QUEUE.put(payload) - return Response(status=204) - - -@app.post("/user_info") -def user_info_from_mitmproxy(): - try: - user_info = json.loads(request.data) - except json.JSONDecodeError: - return Response(status=403) - print(user_info) - if "user" not in user_info: - # 这个请求有问题 - return Response(status=403) - user = user_info['user'] - #有用的信息 - { - # 抖音加密的用户id,也就是url上的id字符串 - "sec_user_id": user.get('sec_uid', ""), - # 用户真实的数字id - "user_id": user.get('uid', 0), - # 开播状态,1是开播了 - "live_status": user.get('live_status', 0), - # 和西瓜视频一样,每次开播,room_id都会变化,需要动态拿取 - "room_id": user.get('room_id', 0), - "nickname": user.get('nickname', ""), - # 多平台粉丝数,包含西瓜视频等关联字节公司下的账号粉丝总数 - "mp_fans_count": user.get('mplatform_followers_count', 0), - # 近期加的归属地 - "ip_location": user.get('ip_location', ""), - } - return Response(status=204)