Make black happy

MeltanoLabs · Dec 22, 2023 · a4c17e1 · a4c17e1
1 parent 29d5982
commit a4c17e1
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 21 deletions.
diff --git a/tap_postgres/client.py b/tap_postgres/client.py
@@ -474,11 +474,13 @@ def consume(self, message) -> dict | None:
         elif message_payload["action"] in delete_actions:
             for column in message_payload["identity"]:
                 row.update({column["name"]: column["value"]})
-            row.update({
-                "_sdc_deleted_at": datetime.datetime.utcnow().strftime(
-                    r"%Y-%m-%dT%H:%M:%SZ"
-                )
-            })
+            row.update(
+                {
+                    "_sdc_deleted_at": datetime.datetime.utcnow().strftime(
+                        r"%Y-%m-%dT%H:%M:%SZ"
+                    )
+                }
+            )
             row.update({"_sdc_lsn": message.data_start})
         elif message_payload["action"] in truncate_actions:
             self.logger.debug(

diff --git a/tap_postgres/tap.py b/tap_postgres/tap.py
@@ -541,24 +541,32 @@ def catalog(self) -> Catalog:  # noqa: C901
                     new_stream.schema.required = None
                 if "_sdc_deleted_at" not in new_stream.schema.properties:  # type: ignore[operator]
                     stream_modified = True
-                    new_stream.schema.properties.update({  # type: ignore[union-attr]
-                        "_sdc_deleted_at": Schema(type=["string", "null"])
-                    })
-                    new_stream.metadata.update({
-                        ("properties", "_sdc_deleted_at"): Metadata(
-                            Metadata.InclusionType.AVAILABLE, True, None
-                        )
-                    })
+                    new_stream.schema.properties.update(
+                        {  # type: ignore[union-attr]
+                            "_sdc_deleted_at": Schema(type=["string", "null"])
+                        }
+                    )
+                    new_stream.metadata.update(
+                        {
+                            ("properties", "_sdc_deleted_at"): Metadata(
+                                Metadata.InclusionType.AVAILABLE, True, None
+                            )
+                        }
+                    )
                 if "_sdc_lsn" not in new_stream.schema.properties:  # type: ignore[operator]
                     stream_modified = True
-                    new_stream.schema.properties.update({  # type: ignore[union-attr]
-                        "_sdc_lsn": Schema(type=["integer", "null"])
-                    })
-                    new_stream.metadata.update({
-                        ("properties", "_sdc_lsn"): Metadata(
-                            Metadata.InclusionType.AVAILABLE, True, None
-                        )
-                    })
+                    new_stream.schema.properties.update(
+                        {  # type: ignore[union-attr]
+                            "_sdc_lsn": Schema(type=["integer", "null"])
+                        }
+                    )
+                    new_stream.metadata.update(
+                        {
+                            ("properties", "_sdc_lsn"): Metadata(
+                                Metadata.InclusionType.AVAILABLE, True, None
+                            )
+                        }
+                    )
             if stream_modified:
                 modified_streams.append(new_stream.tap_stream_id)
             new_catalog.add_stream(new_stream)