X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=page_builder.py;h=235f51192debddc5ca851bdc1ea39334b3c37693;hb=138f8de1ece8b876693d7b8dfc6a139e76aa58b9;hp=fa800d8cf4fb3f6c0f1d5bd1ea3958b6c8dab1c9;hpb=5e241dc47e497c547463cecc07946ea6882835a7;p=kiosk.git

diff --git a/page_builder.py b/page_builder.py
index fa800d8..235f511 100644
--- a/page_builder.py
+++ b/page_builder.py
@@ -1,3 +1,5 @@
+#!/usr/bin/env python3
+
 import sys
 
 
@@ -17,27 +19,27 @@ class page_builder(object):
         self.debug_info = None
         self.custom_html = None
 
-    def set_layout(self, layout):
+    def set_layout(self, layout: int):
         self.layout = layout
         return self
 
-    def set_title(self, title):
+    def set_title(self, title: str):
         self.title = title
         return self
 
-    def set_style(self, style):
+    def set_style(self, style: str):
         self.style = style
         return self
 
-    def add_item(self, item):
+    def add_item(self, item: str):
         self.items.append(item)
         return self
 
-    def set_debug_info(self, debug_info):
+    def set_debug_info(self, debug_info: bool):
         self.debug_info = debug_info
         return self
 
-    def __pick_layout(self):
+    def __pick_layout(self) -> None:
         if len(self.items) == 1:
             self.layout = page_builder.LAYOUT_ONE_ITEM
         elif len(self.items) <= 4:
@@ -45,21 +47,21 @@ class page_builder(object):
         else:
             self.layout = page_builder.LAYOUT_MANY_ITEMS
 
-    def __render_custom_html(self, f):
+    def __render_custom_html(self, f) -> None:
         if self.custom_html is not None:
             f.write(self.custom_html)
 
-    def __render_header(self, f):
+    def __render_header(self, f) -> None:
         if self.title is not None:
             f.write("<H1>%s</H1>\n" % self.title)
         f.write("<HR>\n<TABLE WIDTH=99% BORDER=0>\n<TR>\n")
         if self.style is not None:
             f.write(self.style)
 
-    def __render_footer(self, f):
+    def __render_footer(self, f) -> None:
         f.write("</TR>\n</TABLE>\n")
 
-    def render_html(self, f):
+    def render_html(self, f) -> None:
         if self.layout == page_builder.LAYOUT_AUTO or self.layout is None:
             self.__pick_layout()
         self.__render_custom_html(f)
@@ -84,6 +86,7 @@ class page_builder(object):
 
         else:
             print("Error, unknown layout type: %d" % self.layout)
+            return
 
         count = 0
         self.items.sort(key=len, reverse=True)