From 58292067f01663bdbc5435bded559f4814841c07 Mon Sep 17 00:00:00 2001 From: JonnyWong16 <9099342+JonnyWong16@users.noreply.github.com> Date: Mon, 3 Aug 2020 16:28:57 -0700 Subject: [PATCH] Fix export csv missing dict keys --- plexpy/exporter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/plexpy/exporter.py b/plexpy/exporter.py index 8445c12b..25b147dc 100644 --- a/plexpy/exporter.py +++ b/plexpy/exporter.py @@ -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)