import app.main.util.helpers as helpers
from app.main.data_model.common.user import User


class UserRepository:

    def __init__(self, session):
        self.session = session

    def get_user_with_email(self, email):
        user = self.session.query(User).filter(User.email == email).first()
        return user

    def get_user_by_id(self, user_id):
        user = self.session.query(User).filter(User.id == user_id).first()
        return user

    def get_users(self):
        users = self.session.query(User).all()
        if len(users) == 0:
            return []
        return users

    def get_participant_users(self):
        users = self.session.query(User).filter(User.role_id == helpers.PARTICIPANT_ROLE_ID).all()
        if len(users) == 0:
            return []
        return users