lyrics.py

Go to the documentation of this file.

from __future__ import with_statement

import urllib,sys

import re
from htmlentitydefs import name2codepoint

import gtk,gtk.glade
from plai import config


_entity_re = re.compile(r'&(?:(#)(\d+)|([^;]+));')

def _FixHTML(html):
        """Turn HTML codepoints etc into single characters.
        taken from http://groups.google.com/group/comp.lang.python/browse_frm/thread/9b7bb3f621b4b8e4"""
        def _repl_func(match):
                if match.group(1): # Numeric character reference
                        return unichr(int(match.group(2)))
                else:
                        return unichr(name2codepoint[match.group(3)])
                        
        return _entity_re.sub(_repl_func,html).decode('ascii','ignore')    


def GetLyrics(artist,title):
        """Block and return song lyrics retreived from lyrc.com.ar or None."""
        site='http://lyrc.com.ar/en/tema1en.php'
        args='?artist='+urllib.quote_plus(artist)+'&songname='+urllib.quote_plus(title)

        f=urllib.urlopen(site+args)
        pagecontents=f.read()
        f.close()

        startmarker='</script></td></tr></table>'

        # The second of these is for the case where there are no thanks given.
        endmarkers=["<p><hr size=1 noshade color=white width=100%><font color='#E9AF03'>Thk",
                                """<br><br><a href="#" onClick="javascript:window.open('badsong.php?songname="""]
                                
        start=pagecontents.find(startmarker)+len(startmarker)

        for marker in endmarkers:
                end=pagecontents.find(marker)
                if end!=-1: break
                
        if start==len(startmarker)-1 or end==-1: 
                return None
        else:
                return _FixHTML(pagecontents[start:end].replace('<br />','')).strip()
                
        
def LyricsWindow(lyrics,artist,title):
        """Pops up a window containing 'lyrics' for 'title' by 'artist'."""
                
        def onSave(widget):
                """Sub function, displays file dialog and saves lyrics where chosen."""
                dlg=gtk.FileChooserDialog('Save lyrics where?',None,gtk.FILE_CHOOSER_ACTION_SAVE,(gtk.STOCK_SAVE,gtk.RESPONSE_ACCEPT))
                dlg.set_current_name(artist+' - '+title+'.txt')
                dlg.run()
                fname=dlg.get_filename()
                dlg.hide()
                with open(fname,'w') as f:
                        f.write(buf.get_text(buf.get_start_iter(),buf.get_end_iter())+'\n')
                        
        xml=gtk.glade.XML(config.gladelyricsfname)
        win=xml.get_widget('LyricsWindow')
        win.set_title('Lyrics for '+artist+' - '+title)
        buf=gtk.TextBuffer()
        buf.set_text(lyrics if lyrics else 'Sorry - no lyrics found.')
        view=xml.get_widget('Lyrics')
        view.set_buffer(buf)
        xml.get_widget('SaveButton').connect('clicked',onSave)
        xml.get_widget('QuitButton').connect('clicked',lambda junk:win.destroy())
                

---