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)