def search(self,
asset_manager_id,
party_ids=None,
party_classes=None,
party_types=None,
country_ids=None):
self.logger.info('Search Parties - Asset Manager: %s',
asset_manager_id)
search_params = {}
# Potentially roll this into a loop through args rather than explicitly named - depends on additional validation
if party_ids:
search_params['party_ids'] = ','.join(party_ids)
if party_classes:
search_params['party_classes'] = ','.join(party_classes)
if party_types:
search_params['party_types'] = ','.join(party_types)
if country_ids:
search_params['country_ids'] = ','.join(country_ids)
url = '%s/parties/%s' % (self.endpoint, asset_manager_id)
response = self.session.get(url, params=search_params)
if response.ok:
parties = [
json_to_party(json_party) for json_party in response.json()
]
self.logger.info('Returned %s Parties.', len(parties))
return parties
else:
self.logger.error(response.text)
response.raise_for_status()
def amend(self, party):
self.logger.info('Amend Party - Asset Manager: %s - Party ID: %s', party.asset_manager_id, party.party_id)
url = '%s/parties/%s/%s' % (self.endpoint, party.asset_manager_id, party.party_id)
response = self.session.put(url, json=party.to_interface())
if response.ok:
party = json_to_party(response.json())
return party
else:
self.logger.error(response.text)
response.raise_for_status()
def retrieve(self, asset_manager_id, party_id, version=None):
self.logger.info('Retrieve Party - Asset Manager: %s - Party ID: %s', asset_manager_id, party_id)
url = '%s/parties/%s/%s' % (self.endpoint, asset_manager_id, party_id)
if version:
url += '?version=%d' % int(version)
response = self.session.get(url)
if response.ok:
return json_to_party(response.json())
else:
self.logger.error(response.text)
response.raise_for_status()
def parties_by_asset_manager(self, asset_manager_id):
self.logger.info('Retrieve Parties by Asset Manager: %s', asset_manager_id)
url = '%s/parties/%s' % (self.endpoint, asset_manager_id)
response = self.session.get(url)
if response.ok:
parties = [json_to_party(json_party) for json_party in response.json()]
self.logger.info('Returned %s Parties.', len(parties))
return parties
else:
self.logger.error(response.text)
response.raise_for_status()
def partial(self, asset_manager_id, party_id, updates):
self.logger.info('Partial Amend Asset - Asset Manager: %s - Party ID: %s', asset_manager_id,
party_id)
url = '%s/parties/%s/%s' % (self.endpoint, asset_manager_id, party_id)
# Setting handler ourselves so we can be sure Decimals work
response = self.session.patch(url, data=json.dumps(updates, default=json_handler), headers=self.json_header)
if response.ok:
party = json_to_party(response.json())
return party
else:
self.logger.error(response.text)
response.raise_for_status()
def test_JsonToParty(self):
party = generate_party()
json_party = party.to_json()
gen_party = json_to_party(json_party)
self.assertEqual(gen_party, party)
