Skip to content

Commit

Permalink
rename property
Browse files Browse the repository at this point in the history
  • Loading branch information
rawwar committed May 12, 2024
1 parent 5136181 commit e45a58d
Showing 1 changed file with 14 additions and 8 deletions.
22 changes: 14 additions & 8 deletions airflow/providers/microsoft/mssql/hooks/mssql.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,12 @@ def __init__(
self._sqlalchemy_scheme = sqlalchemy_scheme

@cached_property
def conn(self) -> Connection:
def connection(self) -> Connection:
"""
Get the airflow connection object.
:return: The connection object.
"""
return self.get_connection(getattr(self, self.conn_name_attr))

@property
Expand All @@ -68,7 +73,7 @@ def connection_extra_lower(self) -> dict:
This is used internally for case-insensitive access of mssql params.
"""
return {k.lower(): v for k, v in self.conn.extra_dejson.items()}
return {k.lower(): v for k, v in self.connection.extra_dejson.items()}

@property
def sqlalchemy_scheme(self) -> str:
Expand Down Expand Up @@ -101,12 +106,13 @@ def get_sqlalchemy_connection(

def get_conn(self) -> pymssql.connect:
"""Return ``pymssql`` connection object."""
conn = pymssql.connect(
server=self.conn.host,
user=self.conn.login,
password=self.conn.password,
database=self.schema or self.conn.schema,
port=self.conn.port,
conn = self.connection
return pymssql.connect(
server=conn.host,
user=conn.login,
password=conn.password,
database=self.schema or conn.schema,
port=conn.port,
)
return conn

Expand Down

0 comments on commit e45a58d

Please sign in to comment.