from app.main.data_model.games.stratedge.se_market import SEMarket


class MarketRepository:

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

    def get_product_by_id(self, market_id):
        market = self.session.query(SEMarket).filter(SEMarket.id == market_id).all()
        if len(market) == 0:
            return None
        return market[0]

    def get_markets_by_game_configuration_id(self, game_configuration_id):
        return self.session.query(SEMarket).filter(SEMarket.se_configuration_id == game_configuration_id).all()