From 49763eecd2dfdead1e55da8ac5821dfb0a9516ac Mon Sep 17 00:00:00 2001 From: Vishnu KS Date: Sun, 13 Oct 2019 20:15:41 +0530 Subject: [PATCH] utils: Set book category in book object before storing. --- utils/housekeep.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/utils/housekeep.py b/utils/housekeep.py index 4b1a9ba..abe8424 100644 --- a/utils/housekeep.py +++ b/utils/housekeep.py @@ -46,6 +46,13 @@ def sort(library, key_to_sort_on, reverse = False): new_library[key] = sorted(books, key=lambda k: k[key_to_sort_on], reverse=reverse) return new_library +def format_library(library): + formated_library = [] + for category in library: + for book in library[category]: + book["category"] = category[len("## "):] + formated_library.append(book) + return formated_library def main(): from read_file import load @@ -66,6 +73,6 @@ def main(): render(in_file, out_file, library) if store_json: with open("out.json", "w") as f: - f.write(simplejson.dumps(library, indent=4, sort_keys=True)) + f.write(simplejson.dumps(format_library(library), indent=4, sort_keys=True)) if __name__ == '__main__': main()