import pymssql

Author: grihabor

luigi/contrib/mssqldb.py

@@ -22,7 +22,7 @@
 logger = logging.getLogger('luigi-interface')
 
 try:
-    import _mssql
+    import pymssql
 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 pymssql.MSSQLDatabaseException as e:
             # Error number for table doesn't exist
             if e.number == 208:
                 row = None
@@ -120,7 +120,7 @@ def connect(self):
         """
         Create a SQL Server connection and return a connection object
         """
-        connection = _mssql.connect(user=self.user,
+        connection = pymssql.connect(user=self.user,
                                     password=self.password,
                                     server=self.host,
                                     port=self.port,
@@ -145,7 +145,7 @@ def create_marker_table(self):
                 """
                 .format(marker_table=self.marker_table)
             )
-        except _mssql.MSSQLDatabaseException as e:
+        except pymssql.MSSQLDatabaseException as e:
             # Table already exists code
             if e.number == 2714:
                 pass