lib/Mind-Expanding-Books
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

utils: Set book category in book object before storing.

Browse Source
This commit is contained in:
Vishnu KS
2019-10-13 20:15:41 +05:30
parent 63350a60a3
commit 49763eecd2
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
utils/housekeep.py
View File

@@ -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) new_library[key] = sorted(books, key=lambda k: k[key_to_sort_on], reverse=reverse)
return new_library 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(): def main():
from read_file import load from read_file import load
@@ -66,6 +73,6 @@ def main():
render(in_file, out_file, library) render(in_file, out_file, library)
if store_json: if store_json:
with open("out.json", "w") as f: 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__': if __name__ == '__main__':
main() main()