test_server.py

import os
import sys
from ruamel.yaml import YAML
yaml = YAML()
from flask import Flask
from flask_socketio import SocketIO
from server import SECRET_KEY, setup, set_globals
ROOT = os.path.abspath(os.path.dirname(__file__))


server = sys.argv[-1] if len(sys.argv) > 1 else "broker.hivemq.com"
topic = "mqtt_viewer_test"
test_settings = ("""
mqtt_settings:
  port: 1883
  server: %s


graphs:
  Oven:
    'Oven Temp':
      topic: "__TOPIC__"
      json_template: '{{ oven_temp }}'
    'Meat Temp':
      topic: "__TOPIC__"
      json_template: '{{ meat_temp }}'
      
""" % server).replace("__TOPIC__", topic)


if __name__ == '__main__':
    settings = yaml.load(test_settings)
    if os.path.isfile("./Oven.db"):
        os.remove("./Oven.db")
    app = Flask(__name__, static_url_path='', static_folder='static')
    app.secret_key = SECRET_KEY
    app.config['SESSION_TYPE'] = 'memcache'
    app.config['ADMIN_USERNAME'] = "admin"
    app.config['ADMIN_PASSWORD'] = "admin123"

    socketio = SocketIO(app, async_mode='gevent', logger=True, engineio_logger=True)
    setup(app, socketio, **settings)
    socketio.run(app, host="0.0.0.0",
                 port=3000, debug=True, use_reloader=False)