Skip to content

Commit

Permalink
add sqlalchemy_url property
Browse files Browse the repository at this point in the history
  • Loading branch information
rawwar committed May 12, 2024
1 parent 0698252 commit 449cd44
Showing 1 changed file with 13 additions and 0 deletions.
13 changes: 13 additions & 0 deletions airflow/providers/microsoft/mssql/hooks/mssql.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@
from typing import Any

import pymssql
from sqlalchemy.engine import URL

from airflow.providers.common.sql.hooks.sql import DbApiHook

Expand Down Expand Up @@ -71,6 +72,18 @@ def sqlalchemy_scheme(self) -> str:
raise RuntimeError("sqlalchemy_scheme in connection extra should not contain : or / characters")
return self._sqlalchemy_scheme or extra_scheme or self.DEFAULT_SQLALCHEMY_SCHEME

@property
def sqlalchemy_url(self) -> URL:
conn = self.get_connection(self.conn_name_attr)
return URL.create(
drivername=self.DEFAULT_SQLALCHEMY_SCHEME,
username=conn.login,
password=conn.password,
host=conn.host,
port=conn.port,
database=self.schema or conn.schema,
)

def get_uri(self) -> str:
from urllib.parse import parse_qs, urlencode, urlsplit, urlunsplit

Expand Down

0 comments on commit 449cd44

Please sign in to comment.