# -*- coding: utf-8 -*-
#
# The interface of this HTML generation class is pretty directly based on
# https://pypi.python.org/pypi/html but it uses ElementTree to render the
# HTML output.
#
# Copyright © 2018 Simon Forman
#
# This file is html.py.
#
# html.py is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# html.py is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with html.py. If not, see .
#
from xml.etree.ElementTree import Element, SubElement, tostringlist
HTML4_STRICT_DOCTYPE = (
''
)
class HTML(object):
def __init__(self, element=None):
if element is None:
element = Element('html')
assert isinstance(element, Element), repr(element)
self.root = self.element = element
def __getattr__(self, tag):
e = HTML(SubElement(self.element, tag))
e.root = self.root
return e
def __iadd__(self, other):
return self._append(self.element, other)
def _append(self, to, other):
if isinstance(other, str):
if len(to):
last = to[-1]
if last.tail is None:
last.tail = other
else:
last.tail += other
elif to.text is None:
to.text = other
else:
to.text += other
elif isinstance(other, Element):
to.append(other)
elif isinstance(other, HTML):
if other.root is self.root:
raise ValueError('What are you doing? No recursive HTML.')
to.append(other.element)
else:
raise ValueError('Must only add strings or Elements not %r'
% (other,))
return self
def __call__(self, *content, **kw):
for it in content:
self._append(self.element, it)
self.element.attrib.update(
(k.rstrip('_').replace('_', '-'), v)
for k, v in kw.items()
)
return self
def __enter__(self):
return self
def __exit__(self, exc_type, exc_value, exc_tb):
pass
def __repr__(self):
return '' % (self.element, id(self))
def _stringify(self, encoding='unicode'):
return tostringlist(self.element, method='html', encoding=encoding)
def __str__(self):
return ''.join(self._stringify())
def __iter__(self):
return iter(self._stringify())