#!/usr/bin/python

import sqlite

from htmltmpl import TemplateManager, TemplateProcessor

import os

import ID3

from sys import stderr

from ogg.vorbis import VorbisFile

_imms_db = os.environ['HOME'] + '/.imms/imms.db'

_template = __file__ + '.tmpl'

def rating_to_color(rating):

	i = rating - 75

    	if i <= 25:

       		red = 255

		green = i * 255 / 25

		blue = 0

	elif i <= 50:

		red = (50-i) * 255 / 25

		green = 255

		blue = 0

	else:

		red = 0

		green = 255

		blue = (i-50) * 255 / 25

	return "#%02X%02X%02X" % (red, green, blue)

def grab_tunes():

	sql = sqlite.connect(_imms_db, autocommit = 1)

	cu = sql.cursor()

	cu.execute(''' SELECT r.rating, l.path

		FROM Library l, Rating r

		WHERE r.uid = l.uid AND r.rating >= 125

		ORDER BY r.rating DESC;''')

	return map(lambda x: {

		'rating' : x[0],

		'path': x[1]},

		cu.fetchall() )

def check_tunes(tunes):

	res = []

	for tune in tunes:

		song = tune['path']

		try:

			os.stat(song)

		except:

			continue

		tune['color'] = rating_to_color(tune['rating'])

		if song[-4:] == '.mp3':

			id3 = ID3.ID3(song)

			try:

				tune['path'] = \

					id3['ARTIST'] + \

					' - ' + id3['TITLE']

			except:

				pass

		elif song[-4:] == '.ogg':

			vf = VorbisFile(song)

			vc = vf.comment()

			try:

				tune['path'] = \

					vc['ARTIST'][0] + \

					u' - ' + vc['TITLE'][0]

			except:

				pass

		res.append(tune)

	return res

def output_web():

	tproc = TemplateProcessor()

	tmpl = TemplateManager().prepare(_template)

	tproc.set('Bestof', check_tunes(grab_tunes()))

	print tproc.process(tmpl)

if __name__ == '__main__':

	output_web()