Skip to content

Commit d8c187f

Browse files
committed
drop python 2.7 support (fixes #48)
1 parent 5ed4e90 commit d8c187f

File tree

3 files changed

+6
-14
lines changed

3 files changed

+6
-14
lines changed

rest_pandas/renderers.py

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -2,16 +2,8 @@
22
from rest_framework import status
33
from tempfile import mkstemp
44
from pandas import DataFrame
5-
6-
try:
7-
# Python 2 (uses str)
8-
from StringIO import StringIO
9-
except ImportError:
10-
# Python 3 (Python 2 equivalent uses unicode)
11-
from io import StringIO
12-
5+
from io import StringIO, BytesIO
136
import os
14-
from io import BytesIO
157

168

179
RESPONSE_ERROR = (
@@ -189,7 +181,7 @@ def render_dataframe(self, data, name, *args, **kwargs):
189181
if kwargs.get('orient') == 'records-index':
190182
kwargs['orient'] = 'records'
191183
data.reset_index(inplace=True)
192-
return super(PandasJSONRenderer, self).render_dataframe(
184+
return super().render_dataframe(
193185
data, name, *args, **kwargs
194186
)
195187

rest_pandas/serializers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ def transform_dataframe(self, dataframe):
4343

4444
@property
4545
def data(self):
46-
data = super(serializers.ListSerializer, self).data
46+
data = super().data
4747
if isinstance(data, DataFrame) or data:
4848
dataframe = self.get_dataframe(data)
4949
return self.transform_dataframe(dataframe)
@@ -53,7 +53,7 @@ def data(self):
5353
def to_representation(self, data):
5454
if isinstance(data, DataFrame):
5555
return data
56-
return super(PandasSerializer, self).to_representation(data)
56+
return super().to_representation(data)
5757

5858
@property
5959
def model_serializer(self):

rest_pandas/views.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,7 @@ class PandasView(PandasViewBase, ListAPIView):
126126
"""
127127

128128
def list(self, request, *args, **kwargs):
129-
response = super(PandasView, self).list(request, *args, **kwargs)
129+
response = super().list(request, *args, **kwargs)
130130
return self.update_pandas_headers(response)
131131

132132

@@ -135,5 +135,5 @@ class PandasViewSet(PandasViewBase, ListModelMixin, GenericViewSet):
135135
Pandas-capable model ViewSet (list only)
136136
"""
137137
def list(self, request, *args, **kwargs):
138-
response = super(PandasViewSet, self).list(request, *args, **kwargs)
138+
response = super().list(request, *args, **kwargs)
139139
return self.update_pandas_headers(response)

0 commit comments

Comments
 (0)