#!/usr/bin/python

import sys
import traceback
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
import time
from bkmark import Bookmark

def get_bk_from_form(form):
	bk = Bookmark()
	bk.id = int(form['id'].value)
	bk.url = form['url'].value
	if form.has_key('name'):
		bk.name = form['name'].value
	if form.has_key('desc'):
		bk.desc = form['desc'].value
	return bk

def get_kw_from_form(form):
	kw = form.getvalue("kw")
	if not isinstance(kw, type([])):
		if kw:
			kw = [kw]
		else:
			kw = []
	kw = map(int, kw)
	return kw

def get_new_kw_from_form(form):
	if form.has_key('newkw'):
		return map(lambda e: e.strip(),
			form['newkw'].value.split(','))
	else:
		return []

def main(bk, kw):
    tmpl = TemplateManager().prepare("add_result.tmpl")
    tproc = TemplateProcessor()
    tproc.set("url", bk.url)
    tproc.set("name", bk.name)
    tproc.set("added", bk.added)
    tproc.set("desc", bk.desc)
    tproc.set("Keywords", kw)
    print tproc.process(tmpl)

if (__name__ == "__main__"):
    form = cgi.FieldStorage()
    db = my_db.connect()
    bk = get_bk_from_form(form)
    kw = get_kw_from_form(form)
    newkw = get_new_kw_from_form(form)
    (ids, kws, cnt) = map(list,apply(zip, db.get_all_keywords()))
    for elem in newkw:
    	if elem in kws:
		kw.append(ids[kws.index(elem)])
	else:
		id = db.add_keyword(elem)
		kw.append(id)
    if bk.id == -1:
	id = db.add_bookmark(bk)
    else:
    	db.update_bookmark(bk)
	id = bk.id
    db.update_keywords(id, kw)
    kw = map(lambda e: { 'keyword': e[1] }, db.get_keywords(id)[1:])
    main(bk, kw)