from sqlmodel import Session

from app.models import UserSession


class UserSessionRepo:

    def find_or_create_user_session(self, session: Session, user_session_id: str):
        user_session = session.get(UserSession, user_session_id)
        if not user_session:
            user_session = UserSession(
                id=user_session_id,
            )
            session.add(user_session)
        session.commit()
        return user_session