from app.main.data_model.games.finedge.fe_portfolio import FEPortfolio


class FEPortfolioRepository:

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

    def get_portfolio_by_user_and_game_session(self, user_id, game_session_id):
        portfolio = self.session.query(FEPortfolio).filter(FEPortfolio.user_id == user_id).\
            filter(FEPortfolio.game_session_id == game_session_id).all()
        if len(portfolio) == 0:
            return None
        return portfolio[0]

    def get_portfolio_by_portfolio_id(self, portfolio_id):
        portfolio = self.session.query(FEPortfolio).filter(FEPortfolio.id == portfolio_id).all()
        if len(portfolio) == 0:
            return None
        return portfolio[0]