Merge branch 'master' of wpitchoune.net:/srv/git/mp3tohtml

[mp3tohtml.git] / mp3tohtml / mp3tohtml.py

diff --git a/mp3tohtml/mp3tohtml.py b/mp3tohtml/mp3tohtml.py
index db32bdc..49052bb 100755 (executable)
--- a/mp3tohtml/mp3tohtml.py
+++ b/mp3tohtml/mp3tohtml.py
@@ -1,16 +1,9 @@
 #!/usr/bin/python2
 
-# Generate an HTML page containing information about MP3s in a
-# directory.
-#
-# Usage: mp3tohtml.py <mp3_dir>
-#
-# Last version of this tool can be get from the GIT repository:
-# http://git.wpitchoune.net/radio.git
-#
-# (c) 2016 Jean-Philippe Orsini  <jeanfi@gmail.com>
-
-import argparse;
+# http://wpitchoune.net/mp3tohtml
+# Released under the terms of the GPLv2 license.
+# (c) 2016 Jean-Philippe Orsini <jeanfi@gmail.com>
+
 import eyeD3
 import fnmatch
 import glob
@@ -43,74 +36,78 @@ 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)
+        header.close()
+    except IOError as e:
+        sys.stderr.write("Failed to read header file "
+                         + header_path
+                         + ": " + e.strerror + "\n")
+        raise
 
 def copy_footer(path):
     footer_path = path + "/footer.tpl"
-    footer = open(footer_path, "r")
-    for line in footer:
-        sys.stdout.write(line)
-
-parser = argparse.ArgumentParser(description='Generate an HTML containing information about the MP3 files.')
-parser.add_argument('dir', help='The directory containing the MP3 files')
-parser.add_argument("--config", help="The directory containing the configuration")
-
-args = parser.parse_args()
-
-if (args.config is None):
-    cfg_dir = os.path.dirname(sys.argv[0])
-else:
-    cfg_dir = args.config
-
-copy_header(cfg_dir)
-
-files = []
-for root, dirnames, filenames in os.walk(args.dir):
-    for filename in fnmatch.filter(filenames, '*.mp3'):
-        files.append(os.path.join(root, filename))
-
-for f in files:
-    tag = eyeD3.Tag()
-
-    tag.link(f)
-
-    print("\t<tr>")
-
-    print("\t\t<td>" + tag.getArtist().encode("UTF8") + "</td>\n")
-    print("\t\t<td>" + tag.getAlbum().encode("UTF-8") + "</td>")
-    print("\t\t<td>" + tag.getTitle().encode("UTF-8") + "</td>")
-
-    comments = tag.getComments()
-    strComments = ""
-    for c in comments:
-        strComments += c.comment
-
-    p = re.compile("URL: (.*)\r\nComments: (.*)\r\nCurators?: (.*)\r\nCopyright: (.*)")
-    m = p.match(strComments)
-
-    if m is None:
-        url = ""
-        curator = ""
-        copyright = ""
-    else:
-        if m.group(2) == "http://freemusicarchive.org/":
-            strComments = ""
-            url = "<a href=\"" + m.group(1) + "\">FMA</a>"
+    try:
+        footer = open(footer_path, "r")
+        for line in footer:
+            sys.stdout.write(line)
+        footer.close()
+    except IOError as e:
+        sys.stderr.write("Failed to read footer file "
+                         + footer_path
+                         + ": " + e.strerror + "\n")
+        raise
+
+def generate(mp3_dir, cfg_dir):
+    copy_header(cfg_dir)
+
+    files = []
+    for root, dirnames, filenames in os.walk(mp3_dir):
+        for filename in fnmatch.filter(filenames, '*.mp3'):
+            files.append(os.path.join(root, filename))
+
+    for f in files:
+        tag = eyeD3.Tag()
+
+        tag.link(f)
+
+        print("\t<tr>")
+
+        print("\t\t<td>" + tag.getArtist().encode("UTF8") + "</td>\n")
+        print("\t\t<td>" + tag.getAlbum().encode("UTF-8") + "</td>")
+        print("\t\t<td>" + tag.getTitle().encode("UTF-8") + "</td>")
+
+        comments = tag.getComments()
+        strComments = ""
+        for c in comments:
+            strComments += c.comment
+
+        p = re.compile("URL: (.*)\r\nComments: (.*)\r\nCurators?: (.*)\r\nCopyright: (.*)")
+        m = p.match(strComments)
+
+        if m is None:
+            url = ""
+            curator = ""
+            copyright = ""
         else:
-            strComments =  m.group(2)
-            url = "<a href=\"" + m.group(1) + "\">source</a>"
+            if m.group(2) == "http://freemusicarchive.org/":
+                strComments = ""
+                url = "<a href=\"" + m.group(1) + "\">FMA</a>"
+            else:
+                strComments =  m.group(2)
+                url = "<a href=\"" + m.group(1) + "\">source</a>"
 
-        curator = m.group(3)
+            curator = m.group(3)
 
-        copyright = fma_copyright_to_html(f, m.group(4))
+            copyright = fma_copyright_to_html(f, m.group(4))
 
-    print("\t\t<td>" + url + "</td>")
-    print("\t\t<td>" + curator + "</td>")
-    print("\t\t<td>" + copyright + "</td>")
-    print("\t\t<td>" + strComments + "</td>")
+        print("\t\t<td>" + url + "</td>")
+        print("\t\t<td>" + curator + "</td>")
+        print("\t\t<td>" + copyright + "</td>")
+        print("\t\t<td>" + strComments + "</td>")
 
-    print("\t</tr>")
+        print("\t</tr>")
 
-copy_footer(cfg_dir)
+    copy_footer(cfg_dir)