def select(self, model_name, raw_filters=None):
"""
Select data from a certain type e.g. blocks or spiketrains from the G-Node REST API
and convert the results to a list of model objects.
:param model_name: The name of the model as string.
:type model_name: str
:param raw_filters: Filters as defined by the G-Node REST API.
:type raw_filters: dict
:returns: A list of (filtered) results.
:rtype: list
"""
results = []
raw_filters = {} if raw_filters is None else raw_filters
location = Model.get_location(model_name)
url = urlparse.urljoin(self.location, location)
headers = {}
future = self.__session.get(url, headers=headers, params=raw_filters)
response = future.result()
self.raise_for_status(response)
raw_results = convert.json_to_collections(response.content, as_list=True)
for obj in raw_results:
results.append(convert.collections_to_model(obj))
return results
def set(self, entity, avoid_collisions=False):
"""
Update or create an entity on the G-Node REST API. If an etag/guid is provided by the entity it
will be included in the header with 'If-match' if avoid_collisions is True.
:param entity: The entity to persist.
:type entity: Model
:param avoid_collisions: Try to avoid collisions (lost update problem)
:type avoid_collisions: bool
:returns: The updated entity.
:rtype: Model
:raises: RuntimeError If the changes collide with remote changes of the entity.
"""
if hasattr(entity, "location") and entity.location is not None:
method = 'put'
url = urlparse.urljoin(self.location, entity.location)
else:
method = 'post'
url = urlparse.urljoin(self.location, Model.get_location(entity.model))
data = convert.model_to_json_response(entity)
headers = {'Content-Type': 'application/json'}
if avoid_collisions and entity.guid is not None:
headers['If-match'] = entity.guid
future = getattr(self.__session, method)(url, data=data, headers=headers)
response = future.result()
if response.status_code == 304:
result = entity
else:
self.raise_for_status(response)
result = convert.collections_to_model(convert.json_to_collections(response.content))
return result
