fix apply_catalog to make rep_key work dynamically (#30)

hotgluexyz · Sep 24, 2024 · a4fe02d · a4fe02d
1 parent 5ac2aec
commit a4fe02d
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/tap_stripe/streams.py b/tap_stripe/streams.py
@@ -797,7 +797,16 @@ def replication_key(self):
         th.Property("total_usage", th.IntegerType),
         th.Property("exchange_rate", th.NumberType),
     ).to_dict()
-
+
+    def apply_catalog(self, catalog) -> None:
+        self._tap_input_catalog = catalog
+        catalog_entry = catalog.get_stream(self.name)
+        if catalog_entry:
+            self.primary_keys = catalog_entry.key_properties
+            if catalog_entry.replication_method:
+                self.forced_replication_method = catalog_entry.replication_method
+
+
 class ChargesStream(stripeStream):
 
     name = "charges"