134 lines
3.6 KiB
Python
134 lines
3.6 KiB
Python
import datetime
|
||
import importlib
|
||
import json
|
||
import logging
|
||
from typing import List, Dict, Union
|
||
|
||
from flask import Flask, request
|
||
from flask.helpers import get_env
|
||
from flask_sockets import Sockets
|
||
from app.configs.global_var import _init
|
||
|
||
import sys
|
||
#sys.path.append("/mnt/sdc/algorithm/AICheck-MaskRCNN")
|
||
sys.path.append('E:/alg_demo-master/alg_demo/app/yolov5')
|
||
from app.core.common_utils import logger
|
||
from app.core.err_handler import page_not_found, method_not_allowed, exception_500, exception_400
|
||
|
||
from app.exts import db, migrate, ma
|
||
from app.utils.redis_config import redis_client
|
||
from app.utils.websocket_tool import manager
|
||
|
||
|
||
def register_extensions(app):
|
||
db.init_app(app)
|
||
migrate.init_app(app, db)
|
||
ma.init_app(app)
|
||
|
||
|
||
def register_blueprints(app):
|
||
from app import controller
|
||
for name in controller.__all__:
|
||
bp = getattr(importlib.import_module(f'app.controller.{name}'), 'bp', None)
|
||
if bp is not None:
|
||
app.register_blueprint(
|
||
bp, url_prefix=f"/api{bp.url_prefix if bp.url_prefix else ''}")
|
||
|
||
|
||
def register_error_handler(app):
|
||
app.register_error_handler(404, page_not_found)
|
||
app.register_error_handler(405, method_not_allowed)
|
||
app.register_error_handler(400, exception_400)
|
||
# app.register_error_handler(500, exception_500)
|
||
|
||
|
||
def register_cmds(app):
|
||
pass
|
||
|
||
|
||
def register_log(app):
|
||
logging.root.handlers = []
|
||
logging.basicConfig(format='%(asctime)s : %(levelname)s : %(message)s', level=logging.INFO, filename='ex.log')
|
||
|
||
# set up logging to console
|
||
console = logging.StreamHandler()
|
||
console.setLevel(logging.ERROR)
|
||
# set a format which is simpler for console use
|
||
formatter = logging.Formatter('%(asctime)s : %(levelname)s : %(message)s')
|
||
console.setFormatter(formatter)
|
||
logging.getLogger("").addHandler(console)
|
||
# logging.debug('debug')
|
||
# logging.info('info')
|
||
# logging.warning('warning')
|
||
# logging.error('error')
|
||
# logging.exception('exp')
|
||
|
||
|
||
def register_redis(app):
|
||
redis_client.init_redis_connect()
|
||
|
||
|
||
# socketio = SocketIO()
|
||
|
||
|
||
def create_app():
|
||
app = Flask(__name__, instance_relative_config=True)
|
||
app.config.from_object('app.configs.{}'.format(get_env()))
|
||
app.debug = False
|
||
# app.json_encoder = CustomJSONEncoder
|
||
with app.app_context():
|
||
register_extensions(app)
|
||
register_blueprints(app)
|
||
register_error_handler(app)
|
||
db.create_all()
|
||
_init()
|
||
register_cmds(app)
|
||
register_log(app)
|
||
register_redis(app)
|
||
|
||
@app.before_request
|
||
def log_request_info():
|
||
logger = logging.getLogger('werkzeug')
|
||
if request.method in ['POST', 'PUT']:
|
||
logger.info('Request Body: %s', request.get_data())
|
||
|
||
return app
|
||
|
||
|
||
app = create_app()
|
||
sockets = Sockets(app)
|
||
|
||
|
||
@sockets.route('/echo/<id>')
|
||
def echo_socket(ws, id: str):
|
||
# 保存ws对象，根据
|
||
manager.connect(ws=ws, id=id)
|
||
logger.info('------进入websocket')
|
||
while not ws.closed:
|
||
msg = ws.receive()
|
||
print(msg)
|
||
res = {'code': 1, 'type': 'init', 'msg': '建立连接成功', 'data': '1'}
|
||
try:
|
||
ws.send(json.dumps(res), callable(success())) # 发送数据
|
||
except:
|
||
manager.disconnect(ws=ws, id=id)
|
||
# ws.send(now) # 发送数据
|
||
|
||
|
||
def success():
|
||
logger.info('-----------回调消息成功------------')
|
||
|
||
|
||
@app.route('/')
|
||
def hello_world():
|
||
return 'Hello World!'
|
||
|
||
|
||
if __name__ == '__main__':
|
||
from gevent import pywsgi
|
||
from geventwebsocket.handler import WebSocketHandler
|
||
|
||
#8080 6913 '192.168.0.20'
|
||
server = pywsgi.WSGIServer(('192.168.2.118', 6914), app, handler_class=WebSocketHandler)
|
||
server.serve_forever()
|