From c69a161976c38297db69a40f74e2d3281f702eec Mon Sep 17 00:00:00 2001 From: Roger Hunwicks Date: Sun, 1 Oct 2017 15:07:12 +0200 Subject: [PATCH] Use ChoiceType for PandasDatasource.format - see #3302 --- contrib/connectors/pandas/models.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/contrib/connectors/pandas/models.py b/contrib/connectors/pandas/models.py index d87867e7f33f2..b89eff0b04349 100644 --- 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):