from app.main.data_model.games.stratedge.se_product import SEProduct


class ProductRepository:

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

    def get_product_by_id(self, product_id):
        product = self.session.query(SEProduct).filter(SEProduct.id == product_id).all()
        if len(product) == 0:
            return None
        return product[0]

    def get_product_by_game_configuration_id(self, game_configuration_id):
        return self.session.query(SEProduct).filter(SEProduct.se_configuration_id == game_configuration_id).first()