add primary key for insight streams (#155)

MeltanoLabs · Feb 1, 2024 · b244997 · b244997
1 parent 4233158
commit b244997
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tap_facebook/streams/ad_insights.py b/tap_facebook/streams/ad_insights.py
@@ -63,6 +63,10 @@ def __init__(self, *args, **kwargs) -> None:  # noqa: ANN002, ANN003
         kwargs["name"] = f"{self.name}_{self._report_definition['name']}"
         super().__init__(*args, **kwargs)
 
+    @property
+    def primary_keys(self) -> list[str] | None:
+        return ["date_start", "account_id", "ad_id"] + self._report_definition["breakdowns"]
+
     @staticmethod
     def _get_datatype(field: str) -> th.Type | None:
         d_type = AdsInsights._field_types[field]  # noqa: SLF001