X-Git-Url: http://git.wpitchoune.net/gitweb/?p=radio.git;a=blobdiff_plain;f=mp3tohtml.py;fp=mp3tohtml.py;h=c8631c55b8458665483a3a34b2b347632ed6039b;hp=a683c88eacc23af6168d5a82bc64668ad0580f88;hb=806a30f9e27dfe92748e971bb3d683d7841aab19;hpb=974601b5acfe90049cde855d50715719a04d061d

diff --git a/mp3tohtml.py b/mp3tohtml.py
index a683c88..c8631c5 100755
--- a/mp3tohtml.py
+++ b/mp3tohtml.py
@@ -48,6 +48,14 @@ def copy_header(path):
     for line in header:
         sys.stdout.write(line)
 
+def get_mp3s(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))
+
+    return files
+
 parser = argparse.ArgumentParser(description='Generate an HTML containing information about the MP3 files.')
 parser.add_argument('dir', help='The directory containing the MP3 files')
 
@@ -55,10 +63,7 @@ args = parser.parse_args()
 
 copy_header(sys.argv[0])
 
-files = []
-for root, dirnames, filenames in os.walk(args.dir):
-    for filename in fnmatch.filter(filenames, '*.mp3'):
-        files.append(os.path.join(root, filename))
+files = get_mp3s(args.dir)
 
 for f in files:
     tag = eyeD3.Tag()