fix: Exclude information_schema and pg_catalog from discovery

MeltanoLabs · Dec 3, 2024 · 6c83b46 · 6c83b46
1 parent ff0f5f9
commit 6c83b46
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/tap_postgres/tap.py b/tap_postgres/tap.py
@@ -42,6 +42,11 @@ class TapPostgres(SQLTap):
     name = "tap-postgres"
     default_stream_class = PostgresStream
 
+    exclude_schemas = (
+        "information_schema",
+        "pg_catalog",
+    )
+
     def __init__(
         self,
         *args,
@@ -547,7 +552,11 @@ def catalog_dict(self) -> dict:
             return self.input_catalog.to_dict()
 
         result: dict[str, list[dict]] = {"streams": []}
-        result["streams"].extend(self.connector.discover_catalog_entries())
+        result["streams"].extend(
+            self.connector.discover_catalog_entries(
+                exclude_schemas=self.exclude_schemas,
+            )
+        )
 
         self._catalog_dict: dict = result
         return self._catalog_dict