Fix cursor.arraysize - pymssql doesn't have arraysize

sqlobject · Feb 16, 2018 · f08401c · f08401c
1 parent 96a66ec
commit f08401c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/docs/News.rst b/docs/News.rst
@@ -21,6 +21,8 @@ Drivers
 
 * Fix _setAutoCommit for MSSQL.
 
+* Fix cursor.arraysize - pymssql doesn't have arraysize.
+
 Documentation
 -------------
 

diff --git a/sqlobject/inheritance/iteration.py b/sqlobject/inheritance/iteration.py
@@ -11,7 +11,11 @@ def __init__(self, dbconn, rawconn, select, keepConnection=False):
         super(InheritableIteration, self).__init__(dbconn, rawconn, select,
                                                    keepConnection)
         self.lazyColumns = select.ops.get('lazyColumns', False)
-        self.cursor.arraysize = self.defaultArraySize
+        try:
+            self.cursor.arraysize = self.defaultArraySize
+            self.use_arraysize = True
+        except AttributeError:  # pymssql doesn't have arraysize
+            self.use_arraysize = False
         self._results = []
         # Find the index of the childName column
         childNameIdx = None
@@ -24,7 +28,11 @@ def __init__(self, dbconn, rawconn, select, keepConnection=False):
 
     def next(self):
         if not self._results:
-            self._results = list(self.cursor.fetchmany())
+            if self.use_arraysize:
+                _results = self.cursor.fetchmany()
+            else:
+                _results = self.cursor.fetchmany(size=self.defaultArraySize)
+            self._results = list(_results)
             if not self.lazyColumns:
                 self.fetchChildren()
         if not self._results: