miguelgrinberg.python-socketio/examples/server/wsgi/app.py

# set async_mode to 'threading', 'eventlet', 'gevent' or 'gevent_uwsgi' to
# force a mode else, the best mode is selected automatically from what's
# installed
async_mode = None

# set instrument to `True` to accept connections from the official Socket.IO
# Admin UI hosted at https://admin.socket.io
instrument = True
admin_login = {
    "username": "admin",
    "password": "python",  # change this to a strong secret for production use!
}

from flask import Flask, render_template

import socketio

sio = socketio.Server(
    async_mode=async_mode,
    cors_allowed_origins=None
    if not instrument
    else [
        "http://localhost:5000",
        "https://admin.socket.io",  # edit the allowed origins if necessary
    ],
)
if instrument:
    sio.instrument(auth=admin_login)

app = Flask(__name__)
app.wsgi_app = socketio.WSGIApp(sio, app.wsgi_app)
app.config["SECRET_KEY"] = "secret!"
thread = None


def background_thread():
    """Example of how to send server generated events to clients."""
    count = 0
    while True:
        sio.sleep(10)
        count += 1
        sio.emit("my_response", {"data": "Server generated event"})


@app.route("/")
def index():
    global thread
    if thread is None:
        thread = sio.start_background_task(background_thread)
    return render_template("index.html")


@sio.event
def my_event(sid, message):
    sio.emit("my_response", {"data": message["data"]}, room=sid)


@sio.event
def my_broadcast_event(sid, message):
    sio.emit("my_response", {"data": message["data"]})


@sio.event
def join(sid, message):
    sio.enter_room(sid, message["room"])
    sio.emit("my_response", {"data": "Entered room: " + message["room"]}, room=sid)


@sio.event
def leave(sid, message):
    sio.leave_room(sid, message["room"])
    sio.emit("my_response", {"data": "Left room: " + message["room"]}, room=sid)


@sio.event
def close_room(sid, message):
    sio.emit(
        "my_response",
        {"data": "Room " + message["room"] + " is closing."},
        room=message["room"],
    )
    sio.close_room(message["room"])


@sio.event
def my_room_event(sid, message):
    sio.emit("my_response", {"data": message["data"]}, room=message["room"])


@sio.event
def disconnect_request(sid):
    sio.disconnect(sid)


@sio.event
def connect(sid, environ):
    sio.emit("my_response", {"data": "Connected", "count": 0}, room=sid)


@sio.event
def disconnect(sid, reason):
    print("Client disconnected, reason:", reason)


if __name__ == "__main__":
    if instrument:
        print("The server is instrumented for remote administration.")
        print(
            "Use the official Socket.IO Admin UI at https://admin.socket.io "
            "with the following connection details:"
        )
        print(" - Server URL: http://localhost:5000")
        print(" - Username:", admin_login["username"])
        print(" - Password:", admin_login["password"])
        print("")
    if sio.async_mode == "threading":
        # deploy with Werkzeug
        app.run(threaded=True)
    elif sio.async_mode == "eventlet":
        # deploy with eventlet
        import eventlet
        import eventlet.wsgi

        eventlet.wsgi.server(eventlet.listen(("", 5000)), app)
    elif sio.async_mode == "gevent":
        # deploy with gevent
        from gevent import pywsgi

        try:
            from geventwebsocket.handler import WebSocketHandler

            websocket = True
        except ImportError:
            websocket = False
        if websocket:
            pywsgi.WSGIServer(
                ("", 5000), app, handler_class=WebSocketHandler
            ).serve_forever()
        else:
            pywsgi.WSGIServer(("", 5000), app).serve_forever()
    elif sio.async_mode == "gevent_uwsgi":
        print("Start the application through the uwsgi server. Example:")
        print(
            "uwsgi --http :5000 --gevent 1000 --http-websockets --master "
            "--wsgi-file app.py --callable app"
        )
    else:
        print("Unknown async_mode: " + sio.async_mode)