Fix database name

spotify · Sep 13, 2023 · 7a3c834 · 7a3c834
1 parent ef8a227
commit 7a3c834
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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