From 09378a563a2e0f94e94fd83c0089edc59b33b77c Mon Sep 17 00:00:00 2001 From: Jean-Philippe Orsini Date: Mon, 1 Feb 2016 15:57:25 +0100 Subject: [PATCH] handle error when reading templates --- mp3tohtml/mp3tohtml.py | 27 +++++++++++++++++++++------ 1 file changed, 21 insertions(+), 6 deletions(-) diff --git a/mp3tohtml/mp3tohtml.py b/mp3tohtml/mp3tohtml.py index ee51dee..4a9d8a3 100755 --- a/mp3tohtml/mp3tohtml.py +++ b/mp3tohtml/mp3tohtml.py @@ -50,15 +50,30 @@ def fma_copyright_to_html(path, copyright): def copy_header(path): header_path = path + "/header.tpl" - header = open(header_path, "r") - for line in header: - sys.stdout.write(line) + try: + header = open(header_path, "r") + for line in header: + sys.stdout.write(line) + except IOError as err: + sys.stderr.write("Failed to read header file " + + header_path + + ": " + err.strerror + "\n") + sys.exit(1) + header.close() def copy_footer(path): footer_path = path + "/footer.tpl" - footer = open(footer_path, "r") - for line in footer: - sys.stdout.write(line) + try: + footer = open(footer_path, "r") + for line in footer: + sys.stdout.write(line) + except IOError as err: + sys.stderr.write("Failed to read footer file " + + footer_path + + ": " + err.strerror + "\n") + sys.exit(1) + footer.close() + parser = argparse.ArgumentParser(description='Generate an HTML containing information about the MP3 files.') parser.add_argument('dir', help='The directory containing the MP3 files') -- 2.7.4