export_tsv raises NotImplementedError if necessary

druid-io · Feb 13, 2014 · ba18670 · ba18670
1 parent d8bc3a7
commit ba18670
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,2 +1,4 @@
 *.pyc
 test.py
+pydruid/bard.py
+.idea/
diff --git a/pydruid/client.py b/pydruid/client.py
@@ -80,6 +80,8 @@ def export_tsv(self, dest_path):
             header = self.result[0]['event'].keys()
             header.append('timestamp')
             header.append('version')
+        else:
+            raise NotImplementedError('TSV export not implemented for query type: {0}'.format(self.query_type))
 
         tsv_file.writerow(header)
 
@@ -123,9 +125,7 @@ def export_pandas(self):
                         for v in self.result]
                 nres = [dict(v) for v in nres]
             else:
-                raise NotImplementedError(
-                    'Pandas export not implemented for query type: {0}'
-                    .format(self.query_type))
+                raise NotImplementedError('Pandas export not implemented for query type: {0}'.format(self.query_type))
 
             df = pandas.DataFrame(nres)
             return df