From 111357934eb17bfa24c23fadbf1bfc92d45bfa88 Mon Sep 17 00:00:00 2001 From: blacktwin Date: Thu, 21 Apr 2022 14:05:57 -0400 Subject: [PATCH] more clean up after removing jsonpickle --- fun/playlist_manager.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/fun/playlist_manager.py b/fun/playlist_manager.py index 881d8c0..933fb42 100644 --- a/fun/playlist_manager.py +++ b/fun/playlist_manager.py @@ -968,20 +968,20 @@ if __name__ == "__main__": output_file = '{}-{}-Playlist.{}'.format(user, title, opts.export) if opts.export == 'json': with open(output_file, 'w') as fp: - json.dump(json_dump, fp, indent=4, sort_keys=True) + json.dump(pl_dict, fp, indent=4, sort_keys=True) elif opts.export == 'csv': columns = [] data_list = [] - for rows in json_dump['items']: + for rows in pl_dict['items']: flat_data = flatten(rows) - columns += list(flat_data) - data_list.append(flat_data) + columns += list(rows) + data_list.append(rows) with open(output_file, 'w', encoding='UTF-8', newline='') as data_file: columns = sorted(list(set(columns))) writer = csv.DictWriter(data_file, fieldnames=columns) writer.writeheader() - for data in data_list: + for data in pl_dict['items']: writer.writerow(data)