Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ dependencies = [
"psycopg2-binary==2.9.10",
"sqlalchemy==2.0.43",
"sshtunnel==0.4.0",
"singer-sdk[faker]~=0.49.0",
"singer-sdk[faker,sql]~=0.49.0",
]

[project.urls]
Expand Down Expand Up @@ -132,3 +132,6 @@ convention = "google"

[tool.pytest.ini_options]
addopts = '--durations=10'
filterwarnings = [
"error",
]
4 changes: 2 additions & 2 deletions tap_postgres/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,10 +17,10 @@
import sqlalchemy as sa
import sqlalchemy.types
from psycopg2 import extras
from singer_sdk import SQLConnector, SQLStream
from singer_sdk.connectors.sql import SQLToJSONSchema
from singer_sdk.helpers._state import increment_state
from singer_sdk.helpers._typing import TypeConformanceLevel
from singer_sdk.sql import SQLConnector, SQLStream
from singer_sdk.sql.connector import SQLToJSONSchema
from sqlalchemy.dialects import postgresql

if t.TYPE_CHECKING:
Expand Down
5 changes: 3 additions & 2 deletions tap_postgres/tap.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,9 +12,10 @@
from typing import TYPE_CHECKING, Any, cast

import paramiko
from singer_sdk import SQLStream, SQLTap, Stream
from singer_sdk import Stream
from singer_sdk import typing as th # JSON schema typing helpers
from singer_sdk.singerlib import Catalog, Metadata, Schema
from singer_sdk.sql import SQLStream, SQLTap
from sqlalchemy.engine import URL
from sqlalchemy.engine.url import make_url
from sshtunnel import SSHTunnelForwarder
Expand Down Expand Up @@ -499,7 +500,7 @@ def ssh_tunnel_connect(self, *, ssh_config: dict[str, Any], url: URL) -> URL:
self.ssh_tunnel: SSHTunnelForwarder = SSHTunnelForwarder(
ssh_address_or_host=(ssh_config["host"], ssh_config["port"]),
ssh_username=ssh_config["username"],
ssh_private_key=self.guess_key_type(ssh_config["private_key"]),
ssh_pkey=self.guess_key_type(ssh_config["private_key"]),
ssh_private_key_password=ssh_config.get("private_key_password"),
remote_bind_address=(url.host, url.port),
)
Expand Down
7 changes: 5 additions & 2 deletions uv.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.