from app.main.data_model.games.finedge.fe_indicator import FEIndicator


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

    def get_indicator_by_id(self, indicator_id):
        indicator = self.session.query(FEIndicator).filter(FEIndicator.id == indicator_id).all()
        if len(indicator) == 0:
            return None
        return indicator[0]

    def get_indicators(self):
        indicators = self.session.query(FEIndicator).all()
        if len(indicators) == 0:
            return None
        return indicators