PageMaker.py

Go to the documentation of this file.

class PageMaker():
    import os
    import time
    import common_tools as tools
    
    def __init__(self, name, output_dir, page_base_url, base_html=False, verbose=False):
        if base_html:
            self.content = open(base_html,"r").read()
        else:
            self.content = open("%s/core/view/template.html"%self.os.environ["NOVAANAVALID_DIR"],"r").read()
        self.output_dir    = output_dir
        self.page_base_url = page_base_url
        self.verbose       = verbose
        self.name          = name
        # Run standard methods
        
    
    def addFooter(self):
        t_string = self.time.strftime("%Y-%m-%d %H:%M:%S (%Z)", self.time.localtime())
        footer = \
        """
        <div id="footer">
          <div class="container-fluid">
            <p class="muted credit">Page autogenerated at <font color="#0066CC">%s</font> by Validation's robots.</p>
          </div>
        </div>
        """%t_string
        self.addContent("<!--FOOTER_HOOK-->",footer)

    def setNavBarActive(self):
        in_nav_bar = False
        in_dropdown = False
        for line in self.content.split("\n"):
            if "nav navbar-nav" in line: in_nav_bar = True
            if 'class="dropdown-menu"' in line: in_dropdown = line
            if in_dropdown and '</ul>' in line: in_dropdown = False
            if not in_nav_bar: continue
            if self.name in line:
                # This simplistic method can get confused if we're writing in subfolders
                # and haven't specified this in self.name and instead use self.output_dir.
                # Really this shouldn't happen, but I'll hack it here for now.
                index_to_use = -1
                if self.output_dir[-1] == "/": index_to_use = -2
                if ("/" in self.output_dir) and (self.output_dir.split("/")[index_to_use] not in line):
                    continue
                if in_dropdown:
                    new_line = in_dropdown.replace("<li ", '<li class="active" ')
                    self.content = self.content.replace(in_dropdown,new_line)
                else:
                    new_line = line.replace("<li>",'<li class="active">')
                    self.content = self.content.replace(line,new_line)
                break
                                            
    def addContent(self,pattern,replacement,destructive=False):
        if (not destructive) and (pattern not in replacement): replacement = replacement+"\n"+pattern
        self.content = self.content.replace(pattern,replacement)
    
    def write(self):
        if self.verbose: print "PM   :     Writing page: %s%s"%(self.output_dir,self.name)
        self.tools.mkdir(self.output_dir)
        self.addFooter()
        self.setNavBarActive()
        html_file = open("%s/%s"%(self.output_dir,self.name),"wb")
        html_file.write(self.content)
        html_file.close()