Changed query to return and added fetch_all generator

query() now returns a cursor for you to work with, while fetch_all now has the behavior that query() used to have
eyeonus · Aug 5, 2014 · aa04c40 · aa04c40
1 parent 78edc32
commit aa04c40
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tradedb.py b/tradedb.py
@@ -129,5 +129,8 @@ def query(self, sql):
         conn = pypyodbc.connect(self.path)
         cur = conn.cursor()
         cur.execute(sql)
-        for row in cur:
-            yield row
+        return cur
+
+    def fetch_all(self, sql):
+        for row in self.query(sql):
+            yield row