From ef9952a66ec0b8f5bd15f658fb9834dacfd7787f Mon Sep 17 00:00:00 2001 From: Gregory Borodin Date: Fri, 22 Sep 2023 23:07:11 +0200 Subject: [PATCH] Fix pymssql import (#3252) * `import pymssql` * Revert "`import pymssql`" This reverts commit 7881915b125286994c7bfaa4bd54b0909f57f7ca. * Fix exception name --- luigi/contrib/mssqldb.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/luigi/contrib/mssqldb.py b/luigi/contrib/mssqldb.py index 90e30bd2e2..57c0570673 100644 --- a/luigi/contrib/mssqldb.py +++ b/luigi/contrib/mssqldb.py @@ -22,7 +22,7 @@ logger = logging.getLogger('luigi-interface') try: - import _mssql + from pymssql import _mssql except ImportError: logger.warning("Loading MSSQL module without the python package pymssql. \ This will crash at runtime if SQL Server functionality is used.") @@ -107,7 +107,7 @@ def exists(self, connection=None): WHERE update_id = %s """.format(marker_table=self.marker_table), (self.update_id,)) - except _mssql.MSSQLDatabaseException as e: + except _mssql.MssqlDatabaseException as e: # Error number for table doesn't exist if e.number == 208: row = None @@ -145,7 +145,7 @@ def create_marker_table(self): """ .format(marker_table=self.marker_table) ) - except _mssql.MSSQLDatabaseException as e: + except _mssql.MssqlDatabaseException as e: # Table already exists code if e.number == 2714: pass