chore: Flush logger hanlders at exit

MeltanoLabs · Nov 26, 2024 · def1ae1 · def1ae1
1 parent 3fc2911
commit def1ae1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tap_postgres/tap.py b/tap_postgres/tap.py
@@ -504,6 +504,8 @@ def ssh_tunnel_connect(self, *, ssh_config: dict[str, Any], url: URL) -> URL:
     def clean_up(self) -> None:
         """Stop the SSH Tunnel."""
         self.logger.info("Shutting down SSH Tunnel")
+        for handler in self.logger.handlers:
+            handler.close()
         self.ssh_tunnel.stop()
 
     def catch_signal(self, signum, frame) -> None: