#!/usr/bin/python

from xbel_handler import XBELHandler, bookmark
from xml.sax import saxexts
import bkmark

def import_xbel(xbelfile):
    xbel_handler = XBELHandler()
    p=saxexts.XMLParserFactory.make_parser("xml.sax.drivers.drv_xmlproc")
    p.setDocumentHandler( xbel_handler )
    p.parseFile( xbelfile )
    return  xbel_handler.bms

_keywords = {}
    
def import_bookmarks(db, bms):
	global _keywords
	_keywords = {}
	for elem in db.get_all_keywords():
		_keywords[elem[1]] = elem[0]
	for folder in bms.folders:
		import_folder(db,folder)

def import_folder(db, folder, keywords = []):
	global _keywords
	kw = folder.title.strip()
	if _keywords.has_key(kw):
		id = _keywords[kw]
	else:
		#id = db.add_keyword(kw)
		print "add keyword '" +  kw + "'"
		id = len(_keywords)+100
		_keywords[kw] = id
	for child in folder.children:
		if isinstance(child, bookmark.Bookmark):
			import_bookmark(db, child, keywords+[id])
		else:
			import_folder(db, child, keywords+[id])

def import_bookmark(db, bookmark, keywords):
	bk = bkmark.Bookmark()
	bk.name = bookmark.title
	bk.url = bookmark.href
	if has_attr(bookmark, 'desc'):
		bk.desc = bookmark.desc
	try:
		print "add bookmark '" + bk.name + "'"
		# id = db.add_bookmark(bk)
		print "with keywords '" + join(',',keywords)[-1] + "'"
		# db.update_keywords(id, keywords)
	except:
		pass