xbelweb: lib/xbelimp.py@d79722064d8d (annotated)

24 f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	1	#!/usr/bin/python
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	2
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	3	from xbel_handler import XBELHandler, bookmark
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	4	from xml.sax import saxexts
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	5	import bkmark
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	6
26 17b0cd274530 [svn r1560] Add import functionality and correct a bug in the merge_kw. fabien parents: 25 diff changeset	7	def parse_xbel(xbelfile):
24 f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	8	xbel_handler = XBELHandler()
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	9	p=saxexts.XMLParserFactory.make_parser("xml.sax.drivers.drv_xmlproc")
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	10	p.setDocumentHandler( xbel_handler )
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	11	p.parseFile( xbelfile )
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	12	return xbel_handler.bms
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	13
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	14	_keywords = {}
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	15
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	16	def import_bookmarks(db, bms):
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	17	global _keywords
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	18	_keywords = {}
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	19	for elem in db.get_all_keywords():
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	20	_keywords[elem[1]] = elem[0]
26 17b0cd274530 [svn r1560] Add import functionality and correct a bug in the merge_kw. fabien parents: 25 diff changeset	21	#print "<body>"
17b0cd274530 [svn r1560] Add import functionality and correct a bug in the merge_kw. fabien parents: 25 diff changeset	22	#print "<ol>"
25 777bcb36f7be [svn r1559] Unicode work around. fabien parents: 24 diff changeset	23	for child in bms.folders:
777bcb36f7be [svn r1559] Unicode work around. fabien parents: 24 diff changeset	24	if isinstance(child, bookmark.Bookmark):
777bcb36f7be [svn r1559] Unicode work around. fabien parents: 24 diff changeset	25	import_bookmark(db, child)
777bcb36f7be [svn r1559] Unicode work around. fabien parents: 24 diff changeset	26	else:
777bcb36f7be [svn r1559] Unicode work around. fabien parents: 24 diff changeset	27	import_folder(db, child)
26 17b0cd274530 [svn r1560] Add import functionality and correct a bug in the merge_kw. fabien parents: 25 diff changeset	28	#print "</ol>"
24 f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	29
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	30	def import_folder(db, folder, keywords = []):
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	31	global _keywords
25 777bcb36f7be [svn r1559] Unicode work around. fabien parents: 24 diff changeset	32	kw = folder.title.strip().encode('utf-8')
26 17b0cd274530 [svn r1560] Add import functionality and correct a bug in the merge_kw. fabien parents: 25 diff changeset	33	#print "<li>" + kw
24 f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	34	if _keywords.has_key(kw):
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	35	id = _keywords[kw]
26 17b0cd274530 [svn r1560] Add import functionality and correct a bug in the merge_kw. fabien parents: 25 diff changeset	36	#print " [%d]" % (id,)
24 f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	37	else:
26 17b0cd274530 [svn r1560] Add import functionality and correct a bug in the merge_kw. fabien parents: 25 diff changeset	38	id = db.add_keyword(kw)
17b0cd274530 [svn r1560] Add import functionality and correct a bug in the merge_kw. fabien parents: 25 diff changeset	39	id = len(_keywords)
17b0cd274530 [svn r1560] Add import functionality and correct a bug in the merge_kw. fabien parents: 25 diff changeset	40	#print " [%d] (added)" % (id,)
24 f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	41	_keywords[kw] = id
26 17b0cd274530 [svn r1560] Add import functionality and correct a bug in the merge_kw. fabien parents: 25 diff changeset	42	#print "<ol>"
24 f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	43	for child in folder.children:
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	44	if isinstance(child, bookmark.Bookmark):
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	45	import_bookmark(db, child, keywords+[id])
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	46	else:
f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	47	import_folder(db, child, keywords+[id])
26 17b0cd274530 [svn r1560] Add import functionality and correct a bug in the merge_kw. fabien parents: 25 diff changeset	48	#print "</ol></li>"
24 f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	49
25 777bcb36f7be [svn r1559] Unicode work around. fabien parents: 24 diff changeset	50	def import_bookmark(db, bookmark, keywords = []):
24 f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	51	bk = bkmark.Bookmark()
25 777bcb36f7be [svn r1559] Unicode work around. fabien parents: 24 diff changeset	52	bk.name = bookmark.title.encode('utf-8')
777bcb36f7be [svn r1559] Unicode work around. fabien parents: 24 diff changeset	53	bk.url = bookmark.href.encode('utf-8')
777bcb36f7be [svn r1559] Unicode work around. fabien parents: 24 diff changeset	54	if hasattr(bookmark, 'desc'):
777bcb36f7be [svn r1559] Unicode work around. fabien parents: 24 diff changeset	55	bk.desc = bookmark.desc.encode('utf-8')
26 17b0cd274530 [svn r1560] Add import functionality and correct a bug in the merge_kw. fabien parents: 25 diff changeset	56	id = db.add_bookmark(bk)
17b0cd274530 [svn r1560] Add import functionality and correct a bug in the merge_kw. fabien parents: 25 diff changeset	57	#print "<li>add bookmark '" + bk.name + "' [%d]" % (id,)
17b0cd274530 [svn r1560] Add import functionality and correct a bug in the merge_kw. fabien parents: 25 diff changeset	58	#print " with keywords '" + ','.join(map(str,keywords)) + "'</li>"
17b0cd274530 [svn r1560] Add import functionality and correct a bug in the merge_kw. fabien parents: 25 diff changeset	59	db.update_keywords(id, keywords)
24 f3a926c13ce0 [svn r1558] First version of xbelimp. fabien parents: diff changeset	60

author	fabien@tzone.org
	Wed, 31 Dec 2008 00:07:25 -0500
changeset 69	d79722064d8d
parent 26	17b0cd274530
permissions	-rw-r--r--