from flask import session, current_app
from common.models.user import Tenant


# Definition of Trigger Handlers
def set_tenant_session_data(sender, user, **kwargs):
    current_app.logger.debug(f"Setting tenant session data for user {user.id}")
    tenant = Tenant.query.filter_by(id=user.tenant_id).first()
    session['tenant'] = tenant.to_dict()
    session['default_language'] = tenant.default_language
    session['default_embedding_model'] = tenant.embedding_model
    session['default_llm_model'] = tenant.llm_model


def clear_tenant_session_data(sender, user, **kwargs):
    session.pop('tenant', None)
    session.pop('default_language', None)
    session.pop('default_embedding_model', None)
    session.pop('default_llm_model', None)