Use ChoiceType for PandasDatasource.format - see apache#3302

kimetrica · Oct 1, 2017 · c69a161 · c69a161
1 parent ce7ff69
commit c69a161
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/contrib/connectors/pandas/models.py b/contrib/connectors/pandas/models.py
@@ -266,7 +266,12 @@ def data(self):
 
     @property
     def pandas_read_method(self):
-        return getattr(pd, 'read_{obj.format.code}'.format(obj=self))
+        try:
+            # The format is a Choice object
+            format = self.format.code
+        except AttributeError:
+            format = self.format
+        return getattr(pd, 'read_{format}'.format(format=format))
 
     @property
     def pandas_read_parameters(self):