def test_get_latest_portfolio_positions(mocker): id_1 = 'MP1' date = dt.date(2019, 2, 18) mock_response = {'results': ( PositionSet.from_dict({ 'id': 'mock1', 'positionDate': '2019-02-18', 'lastUpdateTime': '2019-02-19T12:10:32.401Z', 'positions': [ {'assetId': 'MQA123', 'quantity': 0.3}, {'assetId': 'MQA456', 'quantity': 0.7} ] }), )} expected_response = ( PositionSet('mock1', date, dup.parse('2019-02-19T12:10:32.401Z'), ( Position(assetId='MQA123', quantity=0.3), Position(assetId='MQA456', quantity=0.7) )) ) # mock GsSession mocker.patch.object(GsSession.__class__, 'current', return_value=GsSession.get(Environment.QA, 'client_id', 'secret')) mocker.patch.object(GsSession.current, '_get', return_value=mock_response) # run test response = GsPortfolioApi.get_latest_positions(id_1) GsSession.current._get.assert_called_with('/portfolios/{id}/positions/last?type=close'.format(id=id_1), cls=PositionSet) assert response == expected_response
def from_portfolio_id(portfolio_id: str): response = GsPortfolioApi.get_latest_positions(portfolio_id) positions = response.positions if isinstance(response, PositionSet) else response['positions'] instruments = GsAssetApi.get_instruments_for_positions(positions) ret = Portfolio(instruments) ret.name = portfolio_id return ret
def get_latest_position_set(self, position_type: PositionType = PositionType.CLOSE) -> PositionSet: if self.positioned_entity_type == EntityType.ASSET: response = GsAssetApi.get_latest_positions(self.id, position_type) return PositionSet.from_target(response) if self.positioned_entity_type == EntityType.PORTFOLIO: response = GsPortfolioApi.get_latest_positions(portfolio_id=self.id, position_type=position_type.value) return PositionSet.from_target(response) raise NotImplementedError
def from_portfolio_id(portfolio_id: str, date=None): portfolio = GsPortfolioApi.get_portfolio(portfolio_id) response = GsPortfolioApi.get_positions_for_date(portfolio_id, date) if date else \ GsPortfolioApi.get_latest_positions(portfolio_id) response = response[0] if isinstance(response, tuple) else response positions = response.positions if isinstance( response, PositionSet) else response['positions'] instruments = GsAssetApi.get_instruments_for_positions(positions) ret = Portfolio(instruments, name=portfolio.name) ret.__id = portfolio_id return ret
def load_from_portfolio_id(portfolio_id): positions = GsPortfolioApi.get_latest_positions(portfolio_id) instruments = GsAssetApi.get_instruments_for_positions( positions.positions) return Portfolio(instruments)