feat: add postgresql data_source db_version

datalens-tech · Dec 8, 2023 · dfba0d8 · dfba0d8
1 parent b92e1df
commit dfba0d8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/lib/dl_connector_postgresql/dl_connector_postgresql/core/postgresql/data_source.py b/lib/dl_connector_postgresql/dl_connector_postgresql/core/postgresql/data_source.py
@@ -38,6 +38,9 @@ class PostgreSQLDataSourceMixin(BaseSQLDataSource):
     def is_compatible_with_type(cls, source_type: DataSourceType) -> bool:
         return source_type in (SOURCE_TYPE_PG_TABLE, SOURCE_TYPE_PG_SUBSELECT)
 
+    def get_connect_args(self) -> dict:
+        return dict(super().get_connect_args(), server_version=self.db_version)
+
 
 class PostgreSQLDataSource(PostgreSQLDataSourceMixin, StandardSchemaSQLDataSource):
     """PG table"""

diff --git a/...dl_connector_postgresql/dl_connector_postgresql/core/postgresql_base/adapters_postgres.py b/...dl_connector_postgresql/dl_connector_postgresql/core/postgresql_base/adapters_postgres.py
@@ -41,10 +41,11 @@ class PostgresAdapter(BasePostgresAdapter, BaseClassicAdapter[PostgresConnTarget
     }
 
     def get_connect_args(self) -> dict:
-        return {
-            "sslmode": "require" if self._target_dto.ssl_enable else "prefer",
-            "sslrootcert": self.get_ssl_cert_path(self._target_dto.ssl_ca),
-        }
+        return dict(
+            super().get_connect_args(),
+            sslmode="require" if self._target_dto.ssl_enable else "prefer",
+            sslrootcert=self.get_ssl_cert_path(self._target_dto.ssl_ca),
+        )
 
     def get_engine_kwargs(self) -> dict:
         result: Dict = {}