def _delete(cls, entity, parameters=None, ids={}): # pragma: no cover
if isinstance(entity, integer_types) or isinstance(
entity, float_types):
ids['id'] = entity
response = cls.REST_CLIENT.delete(cls.get_base_uri(
cls.endpoint_single(), **ids),
headers=make_headers(),
params=parameters)
elif isinstance(entity, BaseAbstractEntity):
entity_id = getattr(entity, 'id', None)
ids['id'] = entity_id
response = cls.REST_CLIENT.delete(cls.get_base_uri(
cls.endpoint_single(), **ids),
headers=make_headers(),
params=parameters)
else:
response = cls.REST_CLIENT.delete(cls.get_base_uri(
cls.endpoint_single(), **ids),
headers=make_headers(),
params=parameters)
response = cls.REST_CLIENT.handle_response(response)
if response is None:
return entity
if cls.entity_class() is None or not issubclass(
cls.entity_class(), BaseAbstractEntity):
return response # pragma: no cover
else:
try:
return cls.entity_class().from_dict(response.json())
except ValueError:
return response # pragma: no cover
def _partial_update(cls,
entity,
parameters=None,
ids={}): # pragma: no cover
entity_id = getattr(entity, 'id', None)
if entity_id is not None:
ids['id'] = entity_id
if isinstance(entity, BaseAbstractEntity):
response = cls.REST_CLIENT.patch(cls.get_base_uri(
cls.endpoint_single(), **ids),
json=entity.get_json_data(),
headers=make_headers(),
params=parameters)
else:
response = cls.REST_CLIENT.patch(cls.get_base_uri(
cls.endpoint_single(), **ids),
data=json.dumps(entity),
headers=make_headers(),
params=parameters)
response = cls.REST_CLIENT.handle_response(response)
if cls.entity_class() is None or not issubclass(
cls.entity_class(), BaseAbstractEntity):
return response # pragma: no cover
else:
return cls.entity_class().from_dict(response.json())
def _retreive_one(cls,
entity,
parameters=None,
ids={}): # pragma: no cover
if isinstance(entity, (int, float)):
ids["id"] = entity
response = cls.REST_CLIENT.get(cls.get_base_uri(
cls.endpoint_single(), **ids),
headers=make_headers(),
params=parameters)
elif entity is not None and issubclass(entity, BaseAbstractEntity):
entity_id = getattr(entity, "id", None)
ids["id"] = entity_id
response = cls.REST_CLIENT.get(cls.get_base_uri(
cls.endpoint_single(), **ids),
headers=make_headers(),
params=parameters)
else:
response = cls.REST_CLIENT.get(cls.get_base_uri(
cls.endpoint_single(), **ids),
headers=make_headers(),
params=parameters)
response = cls.REST_CLIENT.handle_response(response)
if cls.entity_class() is None or not issubclass(
cls.entity_class(), BaseAbstractEntity):
return response # pragma: no cover
else:
return cls.entity_class().from_dict(response.json())
def reset_mappings(cls, parameters={}):
ids = {'id': 'reset'}
response = cls.REST_CLIENT.post(cls.get_base_uri(
cls.endpoint_single(), **ids),
headers=make_headers(),
params=parameters)
return cls.REST_CLIENT.handle_response(response)
def delete_mapping_by_metadata(cls, metadata, parameters={}):
response = cls.REST_CLIENT.post(cls.get_base_uri(
cls.endpoint_delete_by_metadata()),
headers=make_headers(),
params=parameters,
json=metadata)
return cls.REST_CLIENT.handle_response(response)
def delete_mapping(cls, mapping_id, parameters={}):
mapping_id = cls.get_entity_id(mapping_id, Mapping)
ids = {'id': mapping_id}
response = cls.REST_CLIENT.delete(cls.get_base_uri(
cls.endpoint_single(), **ids),
headers=make_headers(),
params=parameters)
return cls.REST_CLIENT.handle_response(response)
def create_mapping(cls, mapping, parameters={}):
cls.validate_is_entity(mapping, Mapping)
response = cls.REST_CLIENT.post(cls.get_base_uri(cls.endpoint()),
json=mapping.get_json_data(),
headers=make_headers(),
params=parameters)
response = cls.REST_CLIENT.handle_response(response)
return MappingResponse.from_dict(response.json())
def get_unmatched_requests_near_misses(cls, parameters={}):
response = cls.REST_CLIENT.get(
cls.get_base_uri(cls.endpoint_request_unmatched_near_misses()),
headers=make_headers(),
params=parameters
)
response = cls.REST_CLIENT.handle_response(response)
return NearMissMatchResponse.from_dict(response.json())
def get_request(cls, request_id, parameters={}):
ids = {'id': request_id}
response = cls.REST_CLIENT.get(
cls.get_base_uri(cls.endpoint_single(), **ids),
headers=make_headers(),
params=parameters
)
response = cls.REST_CLIENT.handle_response(response)
return RequestResponse.from_dict(response.json())
def retrieve_mapping(cls, mapping_id, parameters={}):
mapping_id = cls.get_entity_id(mapping_id, Mapping)
ids = {'id': mapping_id}
response = cls.REST_CLIENT.get(cls.get_base_uri(
cls.endpoint_single(), **ids),
headers=make_headers(),
params=parameters)
response = cls.REST_CLIENT.handle_response(response)
return Mapping.from_dict(response.json())
def get_unmatched_requests(cls, parameters={}):
response = cls.REST_CLIENT.get(
cls.get_base_uri(cls.endpoint_requests_unmatched()),
headers=make_headers(),
params=parameters
)
response = cls.REST_CLIENT.handle_response(response)
return RequestResponseFindResponse.from_dict(response.json())
def get_all_received_requests(cls, limit=None, since=None, parameters={}):
if limit is not None and limit > 0:
parameters['limit'] = limit
if since is not None:
parameters['since'] = since
response = cls.REST_CLIENT.get(cls.endpoint(), params=parameters, headers=make_headers())
response = cls.REST_CLIENT.handle_response(response)
return RequestResponseAll.from_dict(response.json())
def update_mapping(cls, mapping, parameters={}):
cls.validate_is_entity(mapping, Mapping)
mapping_id = cls.get_entity_id(mapping, Mapping)
ids = {'id': mapping_id}
response = cls.REST_CLIENT.put(cls.get_base_uri(
cls.endpoint_single(), **ids),
json=mapping.get_json_data(),
headers=make_headers(),
params=parameters)
response = cls.REST_CLIENT.handle_response(response)
return Mapping.from_dict(response.json())
def get_matching_requests(cls, request, parameters={}):
cls.validate_is_entity(request, NearMissMatchPatternRequest)
ids = {'id': 'find'}
response = cls.REST_CLIENT.post(
cls.get_base_uri(cls.endpoint_single(), **ids),
json=request.get_json_data(),
headers=make_headers(),
params=parameters
)
response = cls.REST_CLIENT.handle_response(response)
return RequestResponseFindResponse.from_dict(response.json())
def _create(cls, entity, parameters=None, ids={}): # pragma: no cover
if isinstance(entity, BaseAbstractEntity):
response = cls.REST_CLIENT.post(cls.get_base_uri(
cls.endpoint(), **ids),
json=entity.get_json_data(),
headers=make_headers(),
params=parameters)
else:
response = cls.REST_CLIENT.post(cls.get_base_uri(
cls.endpoint(), **ids),
data=json.dumps(entity),
headers=make_headers(),
params=parameters)
response = cls.REST_CLIENT.handle_response(response)
if cls.entity_class() is None or not issubclass(
cls.entity_class(), BaseAbstractEntity):
return response # pragma: no cover
else:
return cls.entity_class().from_dict(response.json())
def _retreive_all(cls, parameters=None, ids={}): # pragma: no cover
response = cls.REST_CLIENT.get(cls.get_base_uri(cls.endpoint(), **ids),
headers=make_headers(),
params=parameters)
response = cls.REST_CLIENT.handle_response(response)
if cls.entity_class() is None or not issubclass(
cls.entity_class(), BaseAbstractEntity):
return response # pragma: no cover
else:
response_json = response.json()
if isinstance(response_json, array_types):
results = []
for r in response_json:
if isinstance(r, dict):
results.append(cls.entity_class().from_dict(r))
return results
else:
return cls.entity_class().from_dict(response.json())
def reset_request_journal(cls, parameters={}):
ids = {"id": "reset"}
response = cls.REST_CLIENT.post(cls.get_base_uri(cls.endpoint_single(), **ids), headers=make_headers(), params=parameters)
return cls.REST_CLIENT.handle_response(response)
def retrieve_all_mappings(cls, parameters={}):
response = cls.REST_CLIENT.get(cls.get_base_uri(cls.endpoint()),
headers=make_headers(),
params=parameters)
response = cls.REST_CLIENT.handle_response(response)
return AllMappings.from_dict(response.json())
def delete_all_mappings(cls, parameters={}):
response = cls.REST_CLIENT.delete(cls.get_base_uri(cls.endpoint()),
headers=make_headers(),
params=parameters)
return cls.REST_CLIENT.handle_response(response)
def reset_all_scenarios(cls, parameters={}):
response = cls.REST_CLIENT.post(cls.get_base_uri(cls.endpoint()),
headers=make_headers(),
params=parameters)
return cls.REST_CLIENT.handle_response(response)
