#!/usr/bin/python

import sys

import traceback

if (__name__ == '__main__'):

	sys.path.insert(0, "/home/fabien/lib/python")

	sys.path.insert(0, "./lib")

	sys.stderr = sys.stdout

	print "Content-type: text/html; charset=iso-8859-1;"

	print

# import cgitb; cgitb.enable()

import cgi

from htmltmpl import TemplateManager, TemplateProcessor

import my_db

from os import environ

from urlparse import urljoin

def get_selection(form):

	kw = form.getvalue("kw")

	if not isinstance(kw, type([])):

		if kw:

			kw = [kw]

		else:

			kw = [0]

	return map(int, kw)

def set_selection(kw, sel):

	res = []

	for key in kw:

		chk = key[0] in sel

		res.append({'id':key[0],

			'keyword': key[1],

			'count' : key[2],

			'checked' : chk})

	res.sort(lambda l,r: cmp(l["keyword"],r["keyword"]))

	return res

def get_curl():

	return urljoin( 'http://' + environ["HTTP_HOST"] + environ["REQUEST_URI"], 'add.py')

def main(bk, kw, curl, total):

    tmpl = TemplateManager().prepare("index.tmpl")

    tproc = TemplateProcessor()

    tproc.set("Bookmarks", bk)

    tproc.set("Keywords", kw)

    tproc.set("curl", curl)

    tproc.set("total", total)

    print tproc.process(tmpl)

if (__name__ == "__main__"):

    form = cgi.FieldStorage()

    db = my_db.connect()

    keywords = db.get_all_keywords()

    total = keywords[0][2]

    keywords = keywords[1:]

    selection = get_selection(form)

    if selection[0] == 0:

    	exc = map(lambda e: int(e[0]), keywords)

   	bookmarks = db.select_bookmarks([0], exc)

    else:

    	bookmarks = db.select_bookmarks(selection)

    keywords = set_selection(keywords, selection)

    if len(bookmarks) > 0:

    	bookmarks = db.get_bookmarks(map(lambda x: x[0], bookmarks))

    bookmarks = map(lambda bk: bk.dict(), bookmarks)

    curl = get_curl();

    main(bookmarks, keywords, curl, total)