2022-11-04 17:37:08 +08:00
|
|
|
|
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
|
2022-11-18 17:04:00 +08:00
|
|
|
|
from app.configs.global_var import _init
|
2022-11-04 17:37:08 +08:00
|
|
|
|
|
|
|
|
|
import sys
|
2022-11-08 10:02:42 +08:00
|
|
|
|
sys.path.append("/mnt/sdc/algorithm/R-ODY")
|
|
|
|
|
# sys.path.append('E:/alg_demo-master/alg_demo/app/yolov5')
|
2022-11-04 17:37:08 +08:00
|
|
|
|
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()
|
2022-11-18 17:04:00 +08:00
|
|
|
|
_init()
|
2022-11-04 17:37:08 +08:00
|
|
|
|
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'
|
2022-11-08 10:02:42 +08:00
|
|
|
|
server = pywsgi.WSGIServer(('192.168.0.20', 6914), app, handler_class=WebSocketHandler)
|
2022-11-04 17:37:08 +08:00
|
|
|
|
server.serve_forever()
|