from app.main.data_model.games.stratedge.se_team import SETeam



class TeamRepository:

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

    def get_teams_by_game_session_id(self, game_session_id):
        teams = self.session.query(SETeam).filter(SETeam.game_session_id == game_session_id).all()
        if len(teams) == 0:
            return None
        return teams