Skip to content

Commit 82af43b

Browse files
authored
fix if custom_cols is None
fixes bug introduced in #40
1 parent cce21fb commit 82af43b

File tree

1 file changed

+10
-8
lines changed

1 file changed

+10
-8
lines changed

drf_renderer_xlsx/renderers.py

+10-8
Original file line numberDiff line numberDiff line change
@@ -163,14 +163,16 @@ def render(self, data, accepted_media_type=None, renderer_context=None):
163163
self.xlsx_header_dict = self._flatten_serializer_keys(
164164
drf_view.get_serializer(), use_labels=use_labels
165165
)
166-
167-
custom_header_dict = {
168-
key: self.custom_cols[key].get('label', None) or key
169-
for key in self.custom_cols.keys()
170-
}
171-
self.combined_header_dict = dict(
172-
list(self.xlsx_header_dict.items()) + list(custom_header_dict.items())
173-
)
166+
if self.custom_cols:
167+
custom_header_dict = {
168+
key: self.custom_cols[key].get('label', None) or key
169+
for key in self.custom_cols.keys()
170+
}
171+
self.combined_header_dict = dict(
172+
list(self.xlsx_header_dict.items()) + list(custom_header_dict.items())
173+
)
174+
else:
175+
self.combined_header_dict = self.xlsx_header_dict
174176

175177
for column_name, column_label in self.combined_header_dict.items():
176178
if column_name == "row_color":

0 commit comments

Comments
 (0)