use columns if given

keboola · Apr 12, 2021 · fc95ff8 · fc95ff8
1 parent d7a0897
commit fc95ff8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/kbcstorage/tables.py b/kbcstorage/tables.py
@@ -419,7 +419,8 @@ def export_to_file(self, table_id, path_name, limit=None,
 
         with open(local_file, mode='rb') as in_file, \
                 open(destination_file, mode='wb') as out_file:
-            columns = table_detail['columns']
+            if columns is None:
+                columns = table_detail['columns']
             columns = ['"{}"'.format(col) for col in columns]
             header = ",".join(columns) + '\n'
             out_file.write(header.encode('utf-8'))