From d5e13a9d2279cb52540f1919a6167e2cd971bf1c Mon Sep 17 00:00:00 2001 From: Ayush Shah Date: Wed, 8 Dec 2021 21:36:13 +0530 Subject: [PATCH] pymssql and pyodbc support added (#1625) --- ingestion/src/metadata/ingestion/source/mssql.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/ingestion/src/metadata/ingestion/source/mssql.py b/ingestion/src/metadata/ingestion/source/mssql.py index 1b3b931cf80..5c79f8d1500 100644 --- a/ingestion/src/metadata/ingestion/source/mssql.py +++ b/ingestion/src/metadata/ingestion/source/mssql.py @@ -19,8 +19,16 @@ class MssqlConfig(SQLConnectionConfig): host_port = "localhost:1433" scheme = "mssql+pytds" service_type = "MSSQL" + use_pymssql: bool = False + use_pyodbc: bool = False + uri_string: str = "" def get_connection_url(self): + if self.use_pyodbc: + self.scheme = "mssql+pyodbc" + return f"{self.scheme}://{self.uri_string}" + elif self.use_pymssql: + self.scheme = "mssql+pymssql" return super().get_connection_url()