56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
# we assume that every line after # Books
|
|
# starting with * is a book title if file type is old
|
|
# starting with | (and not with | Name or |--) is a book if the file type is new
|
|
|
|
# ARGUMENT HANDLING
|
|
try:
|
|
import argparse
|
|
parser = argparse.ArgumentParser(description='Process file.')
|
|
parser.add_argument(
|
|
'--in_file',
|
|
help='File to process, defaults to ./../README.MD')
|
|
parser.add_argument(
|
|
'--out_file',
|
|
help='File to save to, defaults to ./../README-NEW.MD')
|
|
parser.add_argument(
|
|
'--file_type',
|
|
choices=['old', 'new'],
|
|
help='old if links are displayed in a list, new if in a table')
|
|
parser.add_argument(
|
|
'--sort_by',
|
|
choices = ['rating', 'title', 'author', 'year'],
|
|
help='defaults to rating')
|
|
flags = parser.parse_args()
|
|
except ImportError:
|
|
flags = None
|
|
|
|
|
|
def sort(library, key_to_sort_on, reverse = False):
|
|
new_library = {}
|
|
for key in library:
|
|
books = library[key]
|
|
new_library[key] = sorted(books, key=lambda k: k[key_to_sort_on], reverse=reverse)
|
|
return new_library
|
|
|
|
|
|
def main():
|
|
from read_file import load
|
|
from gooodreads import get_goodread_info
|
|
from write_file import render
|
|
|
|
in_file = flags.in_file or './../README.MD'
|
|
out_file = flags.out_file or './../README-new.md'
|
|
file_type = flags.file_type or 'new'
|
|
sort_by = flags.sort_by or 'rating'
|
|
reverse = True if sort_by == 'rating' else False
|
|
|
|
library = load(in_file, file_type)
|
|
get_goodread_info(library)
|
|
library = sort(library, sort_by, reverse)
|
|
render(in_file, out_file, library)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|
|
|
|
|