base.py

from flask_socketio import Namespace, disconnect
from flask import current_app as app

from extensions import db


class BaseSocket(Namespace):
    
    def on_connect(self):
        pass

    def on_disconnect(self):
        disconnect()
        app.logger.info("socket disconnected")
        db.session.remove() # this is important