From af39f49a09dc7a1a477a382c507954e153de9ca1 Mon Sep 17 00:00:00 2001 From: Willem Renes Date: Tue, 22 Jun 2021 15:03:08 +0200 Subject: reading the list --- booklist.py | 26 +++++++++++++++++++++----- 1 file changed, 21 insertions(+), 5 deletions(-) (limited to 'booklist.py') diff --git a/booklist.py b/booklist.py index c210a23..3ece6f7 100644 --- a/booklist.py +++ b/booklist.py @@ -13,13 +13,18 @@ Parsing/managing a book wishlist. """ import sys +from pprint import pprint +from collections import defaultdict +def read_booklist(file_name="booklist.txt"): -def read_booklist(file="booklist.txt"): + booklist = [] try: - with open(file) as f: - for line in f: - print(line) + with open(file_name, 'r') as file: + for line in file: + author, title = line.rstrip().split(' - ') + booklist.append([author, title]) + except OSError as exc: tb = sys.exc_info()[-1] lineno = tb.tb_lineno @@ -27,10 +32,21 @@ def read_booklist(file="booklist.txt"): print(f'{exc.strerror} at {filename} line {lineno}.') sys.exit(exc.errno) + return booklist + def main(): """ main function, do stuff """ - read_booklist("booklist.txt") + booklist = read_booklist("booklist.txt") + pprint(booklist) + + newbooklist = defaultdict(list) + + for k, v in booklist: + newbooklist[k].append(v) + + pprint(newbooklist) + if __name__ == '__main__': -- cgit v1.2.3-70-g09d2