students/Vincent_Aquila/session07/html_render2.py

"""
Python210A Vincent Aquila  Fall2018
using Python 3.7.0
Session 7 HTML Render - A class-based system for rendering html
"""


# This is the framework for the base class
class Element(object):
    indent = "    "
    tag = "html"

    def __init__(self, content=None, **kwargs):
        if content is None:
            self.content = []
        else:
            self.content = [content]
        self.attrs = dict(**kwargs)

    def append(self, new_content):
        self.content.append(new_content)

    def _open_tag(self):
        if self.attrs is None:
            return f"<{self.tag}>\n"
        else:
            open_tag = f"<{self.tag}"
            for tag, attribute in self.attrs.items():
                open_tag += f' {tag}="{attribute}"'
            open_tag += ">\n"
            return open_tag

    def _close_tag(self):
        return f"</{self.tag}>\n"

    def render(self, out_file, cur_ind=""):
        out_file.write(cur_ind + self._open_tag())

        for content in self.content:
            try:
                content.render(out_file, cur_ind + self.indent)
            except AttributeError:
                out_file.write(f"{cur_ind + self.indent}{content}\n")
        out_file.write(cur_ind + self._close_tag())


class Html(Element):
    tag = "html"

    def render(self, out_file, cur_ind=""):
        out_file.write(cur_ind + "<!DOCTYPE html>\n")
        super().render(out_file, cur_ind)


class Body(Element):
    tag = "body"


class P(Element):
    tag = "p"


class Head(Element):
    tag = "head"


class Ul(Element):
    tag = "ul"


class Li(Element):
    tag = "li"


# OneLineTag
class OneLineTag(Element):
    def render(self, out_file, cur_ind=""):
        if self.attrs is None:
            out_file.write(f"{cur_ind}<{self.tag}>{self.content[0]}</{self.tag}>\n")
        else:
            out_file.write(f"{cur_ind}<{self.tag}")
            for tag, attribute in self.attrs.items():
                out_file.write(f' {tag}="{attribute}"')
            out_file.write(f">{self.content[0]}</{self.tag}>\n")

    def append(self, new_content):
        raise NotImplementedError("You cannot add content to a OneLineTag")


class Title(OneLineTag):
    tag = "title"


class A(OneLineTag):
    tag = "a"

    def __init__(self, link, content=None, **kwargs):
        kwargs['href'] = link
        super().__init__(content, **kwargs)


class H(OneLineTag):
    def __init__(self, num, content=None, **kwargs):
        header_tag = "h" + str(num)

        super().__init__(content, **kwargs)
        self.tag = header_tag


# SelfClosingTag
class SelfClosingTag(Element):
    def __init__(self, content=None, **kwargs):
        if content:
            raise TypeError("SelfClosingTag can not contain any content.")
        super().__init__(content, **kwargs)

    def append(self, new_content):
        if new_content:
            raise TypeError("You can not add content to a SelfClosingTag")

    def render(self, out_file, cur_ind=""):
        if self.attrs is None:
            out_file.write(f"{cur_ind}<{self.tag} />\n")
        else:
            out_file.write(f"{cur_ind}<{self.tag}")
            for tag, attribute in self.attrs.items():
                out_file.write(f' {tag}="{attribute}"')
            out_file.write(" />\n")


class Hr(SelfClosingTag):
    tag = "hr"


class Br(SelfClosingTag):
    tag = "br"


class Meta(SelfClosingTag):
    tag = "meta"