Changeset 8255535

Timestamp:

06/28/10 19:28:01 (3 years ago)

Author:

Aurélien Bompard <aurelien@…>

Branches:

master

Children:

a5c46b2

Parents:

8491ac9

git-author:

Aurélien Bompard <aurelien@…> (06/28/10 19:28:01)

git-committer:

Aurélien Bompard <aurelien@…> (06/28/10 19:28:01)

Message:

make-songs-list: always resize covers to save space

File:

• make-songs-list.py (modified) (8 diffs)

make-songs-list.py

@@ -18 +18 @@
 import optparse
 import atexit
+import re
 from pprint import pprint
+
+import PIL
 
 from reportlab.platypus import BaseDocTemplate, Paragraph, Spacer, Image, Table, TableStyle, Frame, PageTemplate
@@ -27 +30 @@
 
 
-
 defaultPageSize = (defaultPageSize[1], defaultPageSize[0])
 PAGE_HEIGHT=defaultPageSize[1]
 PAGE_WIDTH=defaultPageSize[0]
-MARGIN = 8
+MARGIN = 10
 COLSEP = 5
 
 songs = []
 
-def get_songs():
+def get_songs(cover_size, dpi):
     songs_dirs = ["/usr/share/performous/songs",
                   "/usr/local/share/games/ultrastar/songs",
@@ -45 +47 @@
                   "~/.fretsonfire/songs",
                  ]
+    available_songs = []
     for songs_dir in songs_dirs:
         songs_dir = os.path.expanduser(songs_dir)
@@ -53 +56 @@
                 if not cur_file.lower().endswith(".txt"):
                     continue
-                filepath = os.path.join(root, cur_file)
-                song = process_file(filepath)
-                if song:
-                    songs.append(song)
+                available_songs.append(os.path.join(root, cur_file))
+    for index, available_song in enumerate(available_songs):
+        song = process_file(available_song, cover_size, dpi)
+        if song:
+            songs.append(song)
+        sys.stdout.write("\rAdding song %d/%d"
+                         % (index+1, len(available_songs)))
+        sys.stdout.flush()
+    print
+    print "Building PDF..."
     songs.sort(cmp=lambda x,y: cmp(x["artist"],y["artist"]))
 
-def process_file(filepath):
+def process_file(filepath, cover_size, dpi):
     song = {}
     file_h = open(filepath, "r")
@@ -72 +81 @@
         if tag not in song:
             continue
-        song[tag] = os.path.realpath(os.path.join(
+        song["image"] = os.path.realpath(os.path.join(
                         os.path.dirname(filepath), song[tag]))
-        if song[tag].count("  "):
-            tmpimg, tmpimg_path = tempfile.mkstemp(
-                                    prefix="mksonglist-", suffix=".jpg")
-            print >>sys.stderr, "Warning on image '%s': " % song[tag] \
-                 +"the filename contains two consecutive spaces, " \
-                 +"ReportLab does not like that. Using a workaround."
-            origimg = open(song[tag])
-            os.write(tmpimg, origimg.read())
-            origimg.close()
-            atexit.register(os.remove, tmpimg_path)
-            song[tag] = tmpimg_path
-        if not os.path.exists(song[tag]):
-            del song[tag]
+        if not os.path.exists(song["image"]):
+            del song["image"]
+            continue
+        # Resize the image
+        tmpimg, tmpimg_path = tempfile.mkstemp(
+                                prefix="mksonglist-", suffix=".jpg")
+        img = PIL.Image.open(song["image"])
+        img.thumbnail((cover_size * dpi, cover_size * dpi), PIL.Image.ANTIALIAS)
+        img.save(os.fdopen(tmpimg, "w"), "JPEG")
+        atexit.register(os.remove, tmpimg_path)
+        song["image"] = tmpimg_path
+        break
     return song
 
 def filter_songs(karaoke):
+    filter_re = re.compile("(.*)\s+\(.*\)")
     for song in songs[:]:
-        if not karaoke and song["title"].count("(Karaoke)"):
+        title = song["title"]
+        if not karaoke and title.count("(Karaoke)"):
             songs.remove(song)
+            continue
+        filter_match = filter_re.match(title)
+        if filter_match:
+            short_title = filter_match.group(1)
+            if short_title in [ s["title"] for s in songs ]:
+                songs.remove(song)
+                continue
 
 def make_pdf(output, title, cover_size):
@@ -128 +145 @@
     table_contents = []
     for song in songs:
-        image = None
-        label = Paragraph("%s - %s" % (song["artist"], song["title"]), text_style)
-        if "cover" in song and os.path.exists(song["cover"]):
-            image = song["cover"]
-        elif "background" in song and os.path.exists(song["background"]):
-            image = song["background"]
-        if image:
-            image = Image(image, cover_size, cover_size)
-            #image.drawHeight = cover_size *I.drawHeight / I.drawWidth
-            #image.drawWidth = cover_size
-        else:
-            image = Spacer(1, cover_size)
-        row = [image, label]
-        table_contents.append(row)
+        try:
+            image = None
+            label = Paragraph("%s - %s" % (song["artist"], song["title"]), text_style)
+            if "image" in song:
+                image = Image(song["image"], cover_size, cover_size)
+                #image.drawHeight = cover_size *I.drawHeight / I.drawWidth
+                #image.drawWidth = cover_size
+            else:
+                image = Spacer(1, cover_size)
+            row = [image, label]
+            table_contents.append(row)
+        except Exception:
+            print >>sys.stderr, "Error when printing %s" % repr(song)
+            raise
     Story.append(Table(table_contents,
                        colWidths=(cover_size + 2, col_width - (cover_size + 2)),
@@ -158 +175 @@
     parser.add_option("-c", "--cover-size", dest="cover_size", default=32,
                       type="int", help="Cover size (default: %default pt)")
+    parser.add_option("--dpi", dest="dpi", default=96,
+                  help="Image resolution in Dots Per Inch (default: %default)")
     opts, args = parser.parse_args()
     if len(args) != 0:
@@ -165 +184 @@
 if __name__ == '__main__':
     opts = parse_opts()
-    get_songs()
+    get_songs(opts.cover_size, opts.dpi)
     #pprint(songs)
     filter_songs(opts.karaoke)