From: Jean-Philippe Orsini <jeanfi@gmail.com>
Date: Mon, 1 Feb 2016 14:57:25 +0000 (+0100)
Subject: handle error when reading templates
X-Git-Url: https://git.wpitchoune.net/gitweb/?p=mp3tohtml.git;a=commitdiff_plain;h=09378a563a2e0f94e94fd83c0089edc59b33b77c

handle error when reading templates
---

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')