Fix export csv missing dict keys

This commit is contained in:
JonnyWong16 2020-08-03 16:28:57 -07:00
commit 58292067f0
No known key found for this signature in database
GPG key ID: B1F1F9807184697A

View file

@ -924,9 +924,9 @@ def export(section_id=None, rating_key=None, file_format='json'):
elif file_format == 'csv':
flatten_result = helpers.flatten_dict(result)
flatten_attrs = helpers.flatten_dict(attrs)
flatten_attrs = set().union(*flatten_result)
with open(filepath, 'w', encoding='utf-8', newline='') as outfile:
writer = csv.DictWriter(outfile, sorted(flatten_attrs[0].keys()))
writer = csv.DictWriter(outfile, sorted(flatten_attrs))
writer.writeheader()
writer.writerows(flatten_result)