| 1 | """Utility functions used by various web helpers""" | 
|---|
| 2 | import cgi | 
|---|
| 3 | from xml.sax.saxutils import XMLGenerator | 
|---|
| 4 |  | 
|---|
| 5 | def html_escape(s): | 
|---|
| 6 |     """HTML-escape a string or object | 
|---|
| 7 |      | 
|---|
| 8 |     This converts any non-string objects passed into it to strings | 
|---|
| 9 |     (actually, using ``unicode()``).  All values returned are | 
|---|
| 10 |     non-unicode strings (using ``&#num;`` entities for all non-ASCII | 
|---|
| 11 |     characters). | 
|---|
| 12 |      | 
|---|
| 13 |     None is treated specially, and returns the empty string. | 
|---|
| 14 |     """ | 
|---|
| 15 |     if s is None: | 
|---|
| 16 |         return '' | 
|---|
| 17 |     if not isinstance(s, basestring): | 
|---|
| 18 |         if hasattr(s, '__unicode__'): | 
|---|
| 19 |             s = unicode(s) | 
|---|
| 20 |         else: | 
|---|
| 21 |             s = str(s) | 
|---|
| 22 |     s = cgi.escape(s, True) | 
|---|
| 23 |     if isinstance(s, unicode): | 
|---|
| 24 |         s = s.encode('ascii', 'xmlcharrefreplace') | 
|---|
| 25 |     return s | 
|---|
| 26 |  | 
|---|
| 27 | class Partial(object): | 
|---|
| 28 |     """partial object, which will be in Python 2.5""" | 
|---|
| 29 |     def __init__(*args, **kw): | 
|---|
| 30 |         self = args[0] | 
|---|
| 31 |         self.fn, self.args, self.kw = (args[1], args[2:], kw) | 
|---|
| 32 |      | 
|---|
| 33 |     def __call__(self, *args, **kw): | 
|---|
| 34 |         if kw and self.kw: | 
|---|
| 35 |             d = self.kw.copy() | 
|---|
| 36 |             d.update(kw) | 
|---|
| 37 |         else: | 
|---|
| 38 |             d = kw or self.kw | 
|---|
| 39 |         return self.fn(*(self.args + args), **d) | 
|---|
| 40 |  | 
|---|
| 41 | class SimplerXMLGenerator(XMLGenerator): | 
|---|
| 42 |     def addQuickElement(self, name, contents=None, attrs={}): | 
|---|
| 43 |         """Convenience method for adding an element with no children""" | 
|---|
| 44 |         self.startElement(name, attrs) | 
|---|
| 45 |         if contents is not None: | 
|---|
| 46 |             self.characters(contents) | 
|---|
| 47 |         self.endElement(name) | 
|---|