from templates import Template
from keywords import set_selection, sort_keywords

def sort_kw_with_removed(l, r):
	result = l['removed'] - r['removed']
	if result == 0:
		result = l['count'] - r['count']
	return -int(result)

def output(prefs, folder, parents, children, bookmarks, keywords, status, status_msg):
    tmpl = Template("folders.tmpl", prefs, status, status_msg)
    tmpl.set('Parents', parents)
    tmpl.set('folder', folder['id'])
    tmpl.set('name', folder['name'])
    tmpl.set('total', folder['count'])
    tmpl.set('subtotal', len(bookmarks))
    bookmarks.sort(lambda l,r: cmp(l['name'],r['name']))
    tmpl.set('Bookmarks', bookmarks)
    tmpl.set('Subfolders', children)
    keywords.sort(sort_kw_with_removed)
    tmpl.set('Keywords', keywords)
    print tmpl.process()

def do_folders(db, prefs, form, status, status_msg):
    folder = 0
    if form.has_key('folder'):
    	folder = int(form['folder'].value)
    parents = db.get_folder_parents_and_self(folder)
    parents.reverse()
    children = db.get_subfolders(folder)
    selection = map(lambda e: e['keyword'], parents)
    exclusion = map(lambda e: e['keyword'], children)
    folders_keywords = []
    for parent in parents:
	folders_keywords.append(parent['keyword'])
	parent['count'] = db.get_bookmarks_count(folders_keywords)[0][1]
    children_keywords = db.get_bookmarks_count(selection, [], exclusion)
    for child in children:
	folders_keywords.append(child['keyword'])
	# This is really selection, not folders_keywords
	child['count'] = 0
	for kw in children_keywords:
		if kw[0] == child['keyword']:
			child['count'] = kw[1]
			children_keywords.remove(kw)
			break
    del children_keywords
    bookmarks = map(lambda bk: bk[0], db.select_bookmarks(selection))
    allkw = filter(lambda kw: kw[0] not in folders_keywords,
   			 db.get_keywords(bookmarks))
    keywords = map(lambda kw: { 'id': kw[0], 'keyword': kw[1] }, allkw)
    allkw = map(lambda kw: kw[0], allkw)
    count = db.get_bookmarks_count(selection, [], allkw)
    removed = db.get_bookmarks_count(selection, exclusion, allkw)
    for kw in keywords:
    	kw['count'] = 0
	kw['removed'] = 0
    	for cnt in count:
		if kw['id'] == cnt[0]:
			kw['count'] = cnt[1]
			count.remove(cnt)
			break
    	for cnt in removed:
		if kw['id'] == cnt[0]:
			kw['removed'] = cnt[1]
			removed.remove(cnt)
			break
    del allkw, count, removed
    bookmarks = db.select_bookmarks(selection, exclusion)
    if len(bookmarks) <> 0:
   	bookmarks = db.get_bookmarks(map(lambda bk: bk[0], bookmarks))
    	bookmarks = map(lambda bk: bk.dict(), bookmarks)
    output(prefs, parents.pop(), parents, children, bookmarks, keywords, status, status_msg)