from app.main.data_model.games.stratedge.se_configuration import StratEdgeConfiguration


class StratEdgeConfigurationRepository:

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

    def get_stratedge_configuration(self, course_id):
        stratedge_configuration = self.session.query(StratEdgeConfiguration).filter(
            StratEdgeConfiguration.course_id == course_id).all()
        if len(stratedge_configuration) == 0:
            return None
        return stratedge_configuration[0]

    def get_stratedge_configuration_by_id(self, game_configuration_id):
        stratedge_configuration = self.session.query(StratEdgeConfiguration).filter(
            StratEdgeConfiguration.id == game_configuration_id).first()
        return stratedge_configuration