class ConnectionEvents(object): def __init__(self): self.pre_request = Event() self.request = Event() self.pre_response = Event() self.response = Event() self.request_data = Event() self.response_data = Event() def attach(self, recorder_session): self.pre_request.handle(recorder_session.pre_request) self.request.handle(recorder_session.request) self.pre_response.handle(recorder_session.pre_response) self.response.handle(recorder_session.response) self.request_data.handle(recorder_session.request_data) self.response_data.handle(recorder_session.response_data) def clear(self): self.pre_request.clear() self.request.clear() self.pre_response.clear() self.response.clear() self.request_data.clear() self.response_data.clear()