Web   ·   Wiki   ·   Activities   ·   Blog   ·   Lists   ·   Chat   ·   Meeting   ·   Bugs   ·   Git   ·   Translate   ·   Archive   ·   People   ·   Donate
summaryrefslogtreecommitdiffstats
path: root/mwlib/htmlwriter.py
diff options
context:
space:
mode:
Diffstat (limited to 'mwlib/htmlwriter.py')
-rwxr-xr-xmwlib/htmlwriter.py447
1 files changed, 447 insertions, 0 deletions
diff --git a/mwlib/htmlwriter.py b/mwlib/htmlwriter.py
new file mode 100755
index 0000000..6a89157
--- /dev/null
+++ b/mwlib/htmlwriter.py
@@ -0,0 +1,447 @@
+#! /usr/bin/env python
+# -*- coding: utf-8 -*-
+
+# Copyright (c) 2007-2008 PediaPress GmbH
+# See README.txt for additional licensing information.
+
+import os
+from mwlib import parser, rendermath, timeline
+
+import urllib
+import cgi
+
+#from PIL import Image
+
+from mwlib.log import Log
+
+log = Log("htmlwriter")
+
+class HTMLWriter(object):
+ imglevel = 0
+ namedLinkCount = 1
+ def __init__(self, out, images=None, math_renderer=None):
+ self.out = out
+ self.level = 0
+ self.images = images
+ # self.images = imgdb.ImageDB(os.path.expanduser("~/images"))
+ self.references = []
+ if math_renderer is None:
+ self.math_renderer = rendermath.Renderer()
+ else:
+ self.math_renderer = math_renderer
+
+ def _write(self, s):
+ self.out.write(cgi.escape(s).encode('utf8'))
+
+ def getCategoryList(self, obj):
+ categories = list(set(c.target for c in obj.find(parser.CategoryLink)))
+ categories.sort()
+ return categories
+
+ def write(self, obj):
+ m = "write" + obj.__class__.__name__
+ m=getattr(self, m)
+ try:
+ m(obj)
+ except:
+ print "Error trying to write %s" % obj
+
+ def ignore(self, obj):
+ pass
+
+ def serializeVList(self,vlist):
+ args = []
+ styleArgs = []
+ gotClass = 0
+ gotExtraClass = 0
+ for (key,value) in vlist.items():
+ if isinstance(value, (basestring, int)):
+ if key=="class":
+ args.append('%s="%s"' % (key, value))
+ gotClass = 1
+ else:
+ args.append('%s="%s"' % (key, value))
+ if isinstance(value, dict) and key=="style":
+ for (_key,_value) in value.items():
+ styleArgs.append("%s:%s" % (_key, _value))
+ args.append(' style="%s"' % ';'.join(styleArgs))
+ gotExtraClass = 1
+ return ' '.join(args)
+
+
+ def writeMagic(self, m):
+ if m.values.get('html'):
+ for x in m.children:
+ self.write(x)
+
+ def writeCaption(self, obj):
+ # todo- A table contained a Caption node, causing an exception in write.
+ # Not sure what the HTML should be, if any.
+ pass
+
+ def writeSection(self, obj):
+ header = "h%s" % (obj.level)
+ self.out.write("<%s>" % header)
+ self.write(obj.children[0])
+ self.out.write("</%s>" % header)
+
+ self.level += 1
+ for x in obj.children[1:]:
+ self.write(x)
+ self.level -= 1
+
+ def writePreFormatted(self, n):
+ self.out.write("<pre>")
+ for x in n:
+ self.write(x)
+ self.out.write("</pre>")
+
+ def writeNode(self, n):
+ for x in n:
+ self.write(x)
+
+ def writeCell(self, cell):
+ svl = ""
+ if cell.vlist:
+ svl = self.serializeVList(cell.vlist)
+
+ self.out.write('<td %s>' % svl)
+ for x in cell:
+ self.write(x)
+ self.out.write("</td>")
+
+ def writeTagNode(self, t):
+ if t.caption == 'ref':
+ if t.children:
+ self.references.append(t)
+ svl = ""
+ if t.vlist:
+ svl = self.serializeVList(t.vlist)
+
+ self.out.write("<sup %s>%s</sup>" % (svl, len(self.references)))
+
+ return
+ elif t.caption == 'references':
+ if not self.references:
+ return
+
+ self.out.write("<ol>")
+ for r in self.references:
+ svl = ""
+ if r.vlist:
+ svl = self.serializeVList(r.vlist)
+ self.out.write("<li %s>" % svl)
+ for x in r:
+ self.write(x)
+ self.out.write("</li>")
+ self.out.write("</ol>")
+
+ self.references = []
+ return
+ elif t.caption=='imagemap':
+ # FIXME. this is not complete. t.imagemap.entries should also be handled.
+ print "WRITEIMAGEMAP:", t.imagemap
+ if t.imagemap.imagelink:
+ self.write(t.imagemap.imagelink)
+ return
+
+
+ self.out.write(t.starttext.encode('utf8'))
+ for x in t:
+ self.write(x)
+ self.out.write(t.endtext.encode('utf8'))
+
+ def writeRow(self, row):
+ svl = ""
+ if row.vlist:
+ svl = self.serializeVList(row.vlist)
+ self.out.write('<tr %s>' % svl)
+ for x in row:
+ self.write(x)
+
+ self.out.write('</tr>')
+
+ def writeTable(self, t):
+ svl = ""
+ if t.vlist:
+ svl = self.serializeVList(t.vlist)
+
+
+
+ self.out.write("<table %s>" % svl)
+ if t.caption:
+ self.out.write("<caption>")
+ self.write(t.caption)
+ self.out.write("<caption>")
+ for x in t:
+ self.write(x)
+ self.out.write("</table>")
+
+ def writeMath(self, obj):
+ latex = obj.caption
+ p = self.math_renderer.render(latex)
+ self.out.write(p)
+
+ def writeURL(self, obj):
+ self.out.write('<a href="%s" class="offsite" ttid="externallink">' % obj.caption)
+ if obj.children:
+ for x in obj.children:
+ self.write(x)
+ else:
+ self.out.write(obj.caption)
+
+ self.out.write('&nbsp;<img src="/static/outgoing_link.gif" /></a>')
+
+ def writeNamedURL(self, obj):
+ self.out.write('<a href="%s" class="offsite" ttid="externallink">' % obj.caption)
+ if obj.children:
+ for x in obj.children:
+ self.write(x)
+ else:
+ name = "[%s]" % self.namedLinkCount
+ self.namedLinkCount += 1
+ self.out.write(name)
+
+ self.out.write('&nbsp;<img src="/static/outgoing_link.gif" /></a>')
+
+
+ def writeParagraph(self, obj):
+ self.out.write("\n<p>")
+ for x in obj:
+ self.write(x)
+ self.out.write("</p>\n")
+
+ def getHREF(self, obj):
+ parts = obj.target.encode('utf-8').split('#')
+ parts[0] = parts[0].replace(" ", "_")
+
+
+ return '../%s/' % ("#".join([urllib.quote(x) for x in parts]))
+
+ writeLangLink = ignore
+
+ def writeLink(self, obj):
+ if obj.target is None:
+ return
+
+ href = self.getHREF(obj)
+ if href is not None:
+ self.out.write('<a href="%s" class="normallink">' % (href,))
+ else:
+ self.out.write('<a class="deadlink">')
+ if obj.children:
+ for x in obj.children:
+ self.write(x)
+ else:
+ self._write(obj.target)
+
+ self.out.write("</a>")
+
+ def writeSpecialLink(self, obj):
+ if obj.children:
+ for x in obj.children:
+ self.write(x)
+ else:
+ self._write(obj.target)
+
+ def writeCategoryLink(self, obj):
+ if obj.colon:
+ if obj.children:
+ for x in obj.children:
+ self.write(x)
+ else:
+ self._write(obj.target)
+
+ def writeTimeline(self, obj):
+ #img = timeline.drawTimeline(obj.caption)
+ img = None
+ if img is None:
+ return
+
+ target = "/timeline/"+os.path.basename(img)
+ width, height = Image.open(img).size
+
+ self.out.write('<img src="%s" width="%s" height="%s" />' % (target, width, height))
+
+ def writeImageLink(self, obj):
+ """
+ <span class='image'>
+ <span class='left'>
+ <img src='bla' />
+ <span class='imagecaption'>bla bla</span>
+ <span/>
+ <span/>
+ """
+
+ if self.images is None:
+ return
+
+ width = obj.width
+ height = obj.height
+
+ #if not width:
+ # width = 400 # what could be a sensible default if no width is given? maybe better 0?
+
+ if width:
+ path = self.images.getPath(obj.target, size=max(width, height))
+ url = self.images.getURL(obj.target, size=max(width, height))
+ else:
+ path = self.images.getPath(obj.target)
+ url = self.images.getURL(obj.target)
+
+ if url is None:
+ return
+
+ if isinstance(path, str):
+ path = unicode(path, 'utf8')
+
+ if self.imglevel==0:
+ self.imglevel += 1
+
+ # WTB: Added the ability to not specify width & height since images may not be found locally.
+ # This may have to be redone eventually, perhaps we need a database of image dimensions,
+ # but I doubt it. Besides, more hardcoded pathnames in 'getimg'?
+ try:
+ def getimg():
+ return Image.open(path)
+ img = None
+
+ if not width:
+ if not img:
+ img = getimg()
+ size = img.size
+ width = min(400, size[0])
+
+ if not height:
+ if not img:
+ img = getimg()
+ size = img.size
+ height = size[1]*width/size[0]
+ except IOError, err:
+ log.warn("Image.open failed:", err, "path=", repr(path))
+ # WTB: Removed following return as images will not always be found locally.
+ #self.imglevel -= 1
+ #return
+
+ attr = ''
+ attr_css = ''
+
+ if width:
+ attr += "width='%d' " % width
+ attr_css += "width:%dpx " % width
+
+ if height:
+ attr += "height='%d' " % height
+ # WTB: Note- height not applied to CSS.
+
+ if obj.isInline():
+ self.out.write('<img src="%s" %s/>' % (url.encode("utf8"), attr.encode("utf8")))
+ else:
+ # WTB: This looked like a mistake to me, it was modifying obj.align instead of align.
+ # This function should not modify obj at all.
+ align = obj.align
+ if obj.thumb == True and not align:
+ align = "clear right"
+ self.out.write('''<div class="bbotstyle image %s" style="%s">'''% (align, attr_css))
+ self.out.write('<img src="%s" %s/>' % (url.encode("utf8"), attr.encode("utf8")))
+
+ self.out.write('<span class="imagecaption">')
+ for x in obj.children:
+ self.write(x)
+ self.out.write('</span></div>')
+ self.imglevel -= 1
+ else:
+ self.out.write('<a href="%s">' % url)
+ for x in obj.children:
+ self.write(x)
+ self.out.write('</a>')
+
+ def writeText(self, t):
+ #self.out.write(cgi.escape(t.caption).encode('ascii', 'xmlcharrefreplace'))
+ self._write(t.caption)
+
+ writeControl = writeText
+
+ def writeArticle(self, a):
+
+ for x in a:
+ self.write(x)
+
+ self.out.write("\n<br/>")
+
+ def writeStyle(self, s):
+ if s.caption == "''":
+ tag = 'em'
+ elif s.caption=="'''''":
+ self.out.write("<strong><em>")
+ for x in s:
+ self.write(x)
+ self.out.write("</em></strong>")
+ return
+ elif s.caption == "'''":
+ tag = 'strong'
+ elif s.caption == ";":
+ self.out.write("<div><strong>")
+ for x in s:
+ self.write(x)
+ self.out.write("</strong></div>")
+ return
+
+ elif s.caption.startswith(":"):
+ self.out.write("<blockquote>"*len(s.caption))
+ for x in s:
+ self.write(x)
+ self.out.write("</blockquote>"*len(s.caption))
+ return
+ elif s.caption == "overline":
+ self.out.write('<u style="text-decoration: overline;">')
+ for x in s:
+ self.write(x)
+ self.out.write('</u>')
+ return
+ else:
+ tag = s.caption
+
+
+ self.out.write("<%s>" % tag)
+ for x in s:
+ self.write(x)
+ self.out.write("</%s>" % tag)
+
+ def writeItem(self, item):
+ svl = ""
+ if item.vlist:
+ svl = self.serializeVList(item.vlist)
+ self.out.write("<li %s>" % svl)
+ for x in item:
+ self.write(x)
+ self.out.write("</li>\n")
+
+ def writeItemList(self, lst):
+ if lst.numbered:
+ tag = "ol"
+ else:
+ tag = "ul"
+
+ self.out.write("<%s>" % tag)
+
+ for x in lst:
+ self.write(x)
+ self.out.write("\n")
+
+ self.out.write("</%s>" % tag)
+
+
+class NoLinksWriter(HTMLWriter):
+ """Subclass that ignores (non-outgoing) links"""
+
+ def writeLink(self, obj):
+ if obj.target is None:
+ return
+
+ if obj.children:
+ for x in obj.children:
+ self.write(x)
+ else:
+ self._write(obj.target)
+
generated by cgit v0.9.1 at 2024-05-04 03:24:34 (GMT)