diff --git a/tap_hubspot/streams.py b/tap_hubspot/streams.py index 96db20c..7b4d03e 100644 --- a/tap_hubspot/streams.py +++ b/tap_hubspot/streams.py @@ -98,6 +98,11 @@ def get_child_context(self, record: dict, context: Optional[dict]) -> dict: """Return a context dictionary for child streams.""" return {"archived": record["archived"], "company_id": record["id"]} + def post_process(self, row: dict, context: Optional[dict]) -> dict: + """As needed, append or transform raw data to match expected structure. + Returns row""" + return row + class DealsStream(HubspotStream): """Define custom stream.""" @@ -128,6 +133,11 @@ def get_child_context(self, record: dict, context: Optional[dict]) -> dict: "deal_id": record["id"], } + def post_process(self, row: dict, context: Optional[dict]) -> dict: + """As needed, append or transform raw data to match expected structure. + Returns row""" + return row + class ContactsStream(HubspotStream): """Define custom stream.""" @@ -155,6 +165,11 @@ def get_child_context(self, record: dict, context: Optional[dict]) -> dict: """Return a context dictionary for child streams.""" return {"archived": record["archived"], "contact_id": record["id"]} + def post_process(self, row: dict, context: Optional[dict]) -> dict: + """As needed, append or transform raw data to match expected structure. + Returns row""" + return row + class PropertiesStream(HubspotStream): """Define custom stream."""