from app.main.data_model.games.stratedge.se_strategic_decision import SEStrategicDecision


class StrategicDecisionRepository:

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

    def get_strategic_decision_by_id(self, strategic_decision_id):
        strategic_decision = self.session.query(SEStrategicDecision).\
            filter(SEStrategicDecision.id == strategic_decision_id).all()
        if len(strategic_decision) == 0:
            return None
        return strategic_decision[0]

    def get_strategic_decision_by_game_configuration_id(self, game_configuration_id):
        return self.session.query(SEStrategicDecision). \
            filter(SEStrategicDecision.se_configuration_id == game_configuration_id).all()