from app.main.data_model.games.stratedge.se_competitor import SECompetitor


class CompetitorRepository:

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

    def get_competitor_by_id(self, competitor_id):
        competitor = self.session.query(SECompetitor).filter(SECompetitor.id == competitor_id).all()
        if len(competitor) == 0:
            return None
        return competitor[0]

    def get_competitor_by_game_configuration_id(self, game_configuration_id):
        return self.session.query(SECompetitor).filter(SECompetitor.se_configuration_id == game_configuration_id).all()