Merge pull request #6 from dfds-data/feat/column-names

Feat/column names

denodoclient/__init__.py

@@ -9,4 +9,4 @@
 
 __author__ = """Martin Morset"""
 __email__ = "[email protected]"
-__version__ = "__version__ = '0.2.0'"
+__version__ = "0.3.0"

denodoclient/dataframes/denododataframeclient.py

@@ -14,10 +14,10 @@ def to_dataframe(self, chunksize: int = None):
             return self._chunkwise_rows_to_dataframe(chunksize)
 
     def _all_rows_to_dataframe(self) -> pd.DataFrame:
-        return pd.DataFrame((tuple(t) for t in self._cursor.fetchall()))
+        return pd.DataFrame((tuple(t) for t in self._cursor.fetchall()), columns=self.columns)
 
     def _chunkwise_rows_to_dataframe(self, chunksize) -> Iterable[pd.DataFrame]:
         rows = self._cursor.fetchmany(chunksize)
         while rows:
-            yield pd.DataFrame(tuple(t) for t in rows)
+            yield pd.DataFrame((tuple(t) for t in rows), columns=self.columns)
             rows = self._cursor.fetchmany(chunksize)

denodoclient/denodoclient.py

@@ -88,3 +88,7 @@ def query(self, query: VqlQuery):
     @property
     def cursor(self):
         return self._cursor
+
+    @property
+    def columns(self):
+        return [description[0] for description in self._cursor.description]

setup.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.2.0
+current_version = 0.3.0
 commit = True
 tag = True
 

setup.py

@@ -47,6 +47,6 @@
     test_suite="tests",
     tests_require=test_requirements,
     url="https://github.com/dfds-data/denodoclient",
-    version="0.2.0",
+    version="0.3.0",
     zip_safe=False,
 )