mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-11-30 17:15:08 +08:00
69 lines
1.8 KiB
Python
69 lines
1.8 KiB
Python
import os
|
|
import importlib
|
|
|
|
from flask import Flask, request
|
|
|
|
from little_shit import SkipException, get_filters_dir
|
|
from filter import apply_filters
|
|
|
|
app = Flask(__name__)
|
|
|
|
|
|
@app.route('/qq/', methods=['POST'])
|
|
def _handle_qq_message():
|
|
ctx_msg = request.json
|
|
ctx_msg['via'] = 'qq'
|
|
return _main(ctx_msg)
|
|
|
|
|
|
@app.route('/wx/', methods=['POST'])
|
|
def _handle_wx_message():
|
|
ctx_msg = request.json
|
|
ctx_msg['via'] = 'wx'
|
|
return _main(ctx_msg)
|
|
|
|
|
|
def _main(ctx_msg: dict):
|
|
_preprocess_ctx_msg(ctx_msg)
|
|
try:
|
|
if ctx_msg.get('post_type') != 'receive_message':
|
|
raise SkipException
|
|
if not apply_filters(ctx_msg):
|
|
raise SkipException
|
|
print(ctx_msg)
|
|
except SkipException:
|
|
# Skip this message
|
|
pass
|
|
|
|
return '', 204
|
|
|
|
|
|
def _preprocess_ctx_msg(ctx_msg: dict):
|
|
if 'group_uid' in ctx_msg:
|
|
ctx_msg['group_uid'] = str(ctx_msg['group_uid'])
|
|
if 'sender_uid' in ctx_msg:
|
|
ctx_msg['sender_uid'] = str(ctx_msg['sender_uid'])
|
|
if 'sender_id' in ctx_msg:
|
|
ctx_msg['sender_id'] = str(ctx_msg['sender_id'])
|
|
if 'discuss_id' in ctx_msg:
|
|
ctx_msg['discuss_id'] = str(ctx_msg['discuss_id'])
|
|
if 'group_id' in ctx_msg:
|
|
ctx_msg['group_id'] = str(ctx_msg['group_id'])
|
|
if 'id' in ctx_msg:
|
|
ctx_msg['id'] = str(ctx_msg['id'])
|
|
|
|
|
|
def _load_filters():
|
|
filter_mod_files = filter(
|
|
lambda filename: filename.endswith('.py') and not filename.startswith('_'),
|
|
os.listdir(get_filters_dir())
|
|
)
|
|
command_mods = [os.path.splitext(file)[0] for file in filter_mod_files]
|
|
for mod_name in command_mods:
|
|
importlib.import_module('filters.' + mod_name)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
_load_filters()
|
|
app.run(host=os.environ.get('HOST', '0.0.0.0'), port=os.environ.get('PORT', '8080'))
|