Move rewrite_title to external file

2 files changed, 135 insertions, 134 deletions

diff --git a/generators/generators.py b/generators/generators.py
index 8887bddb..9ad773c2 100644
--- a/generators/generators.py
+++ b/generators/generators.py
@@ -383,3 +383,137 @@ def generate_sitemap(feed):
         sitemapfile.write('<?xml version="1.0" encoding="UTF-8"?>\n%s\n' %
                           ET.tostring(sitemap).decode('utf-8'))
         sys.stderr.write("wrote sitemap.xml\n")
+
+
+def rewrite_title():
+    """Override the title of some page for a better render"""
+    sys.stderr.write("Overriding some titles\n")
+
+    filenames =['build/index.html',
+                'build/blog/index.html',
+                'build/notes/index.html']
+
+    for root, dirs, files in os.walk(BUILDDIR):
+        for name in files:
+            if re.search(r'website($)',root):
+                if name.endswith(".html"):
+                    try:
+                        filenames.append("build"+os.path.join(root.split(BUILDDIR)[1], name))
+                    except IndexError:
+                        filenames.append(name)
+            if re.search(r'Documents($)',root):
+                if name.endswith(".html"):
+                    try:
+                        filenames.append("build"+os.path.join(root.split(BUILDDIR)[1], name))
+                    except IndexError:
+                        filenames.append(name)
+            if re.search(r'notes($)',root):
+                if name.endswith(".html"):
+                    try:
+                        filenames.append("build"+os.path.join(root.split(BUILDDIR)[1], name))
+                    except IndexError:
+                        filenames.append(name)
+            if re.search(r'blog($)',root):
+                if name.endswith(".html"):
+                    try:
+                        filenames.append("build"+os.path.join(root.split(BUILDDIR)[1], name))
+                    except IndexError:
+                        filenames.append(name)
+
+    h1_titles_list = []
+    h1_title = []
+
+    h2_titles_list = []
+    h2_title = []
+
+    fd, tmppath = tempfile.mkstemp()
+    os.close(fd)
+    for filename in filenames:
+        soup = bs4.BeautifulSoup(open(filename), "lxml")
+        for myh1 in soup.find_all("h1"):
+            if re.match("^(?!.*article-title).*$", str(myh1)):
+                h1_id = myh1['id']
+                h1_name = myh1.string
+
+                h1_title.append(str(myh1))
+
+
+                myh1['class'] = "h1"
+                myh1.string = ""
+
+                h1_span_left = soup.new_tag("span")
+                h1_span_left['class'] = "left-h1"
+                h1_span_left.string = "█▓▒░"
+
+                h1_span_title = soup.new_tag("span")
+                h1_span_title['class'] = "title-h1"
+                h1_span_title.string = "「"+h1_name+"」"
+
+                h1_span_right = soup.new_tag("span")
+                h1_span_right['class'] = "right-h1"
+                h1_span_right.string = "░▒▓█"
+
+                myh1.string.insert_before(h1_span_left)
+                myh1.span.insert_after(h1_span_right)
+                myh1.span.insert_after(h1_span_title)
+
+                h1_title.append(myh1)
+                h1_title.append(h1_name)
+
+                h1_titles_list.append(h1_title)
+            h1_title = []
+
+        for myh2 in soup.find_all("h2"):
+            if re.match("^(?!.*blog-index-year-title).*$", str(myh2)):
+                h2_id = myh2['id']
+                h2_name = myh2.string
+
+                h2_title.append(str(myh2))
+
+
+                myh2['class'] = "h2"
+                myh2.string = ""
+
+                h2_span_left = soup.new_tag("span")
+                h2_span_left['class'] = "left-h2"
+                h2_span_left.string = ".: "
+
+                h2_span_title = soup.new_tag("span")
+                h2_span_title['class'] = "title-h2"
+                h2_span_title.string = h2_name
+
+                h2_span_right = soup.new_tag("span")
+                h2_span_right['class'] = "right-h2"
+                h2_span_right.string = " :."
+
+                myh2.string.insert_before(h2_span_left)
+                myh2.span.insert_after(h2_span_right)
+                myh2.span.insert_after(h2_span_title)
+
+                h2_title.append(myh2)
+                h2_title.append(h2_name)
+
+                h2_titles_list.append(h2_title)
+            h2_title = []
+
+    tested_title_list = []
+    tested_title = []
+    for filename in filenames:
+        soup = bs4.BeautifulSoup(open(filename), "lxml")
+        if os.path.exists(filename):
+            with open(filename, 'r', encoding='utf-8') as indexmd:
+                lines = indexmd.readlines()
+
+                with open(filename, 'w', encoding='utf-8') as indexmd:
+                    for line in lines:
+                        string = ""
+                        for title in h1_titles_list:
+                            if re.match(".*"+title[0]+".*", line):
+                                string = str(title[1])
+                        for title in h2_titles_list:
+                            if re.match(".*"+title[0]+".*", line):
+                                string = str(title[1])
+                        if string != "":
+                            indexmd.write(re.sub(line, string, line))
+                        else:
+                            indexmd.write(line)
diff --git a/pyblog b/pyblog
index 97a88470..11ea868b 100755
--- a/pyblog
+++ b/pyblog
@@ -51,139 +51,6 @@ from config.config import *
 from generators import generators
 
 
-def rewrite_title():
-    """Override the title of some page for a better render"""
-    sys.stderr.write("Overriding some titles\n")
-
-    filenames =['build/index.html',
-                'build/blog/index.html',
-                'build/notes/index.html']
-
-    for root, dirs, files in os.walk(BUILDDIR):
-        for name in files:
-            if re.search(r'website($)',root):
-                if name.endswith(".html"):
-                    try:
-                        filenames.append("build"+os.path.join(root.split(BUILDDIR)[1], name))
-                    except IndexError:
-                        filenames.append(name)
-            if re.search(r'Documents($)',root):
-                if name.endswith(".html"):
-                    try:
-                        filenames.append("build"+os.path.join(root.split(BUILDDIR)[1], name))
-                    except IndexError:
-                        filenames.append(name)
-            if re.search(r'notes($)',root):
-                if name.endswith(".html"):
-                    try:
-                        filenames.append("build"+os.path.join(root.split(BUILDDIR)[1], name))
-                    except IndexError:
-                        filenames.append(name)
-            if re.search(r'blog($)',root):
-                if name.endswith(".html"):
-                    try:
-                        filenames.append("build"+os.path.join(root.split(BUILDDIR)[1], name))
-                    except IndexError:
-                        filenames.append(name)
-
-    h1_titles_list = []
-    h1_title = []
-
-    h2_titles_list = []
-    h2_title = []
-
-    fd, tmppath = tempfile.mkstemp()
-    os.close(fd)
-    for filename in filenames:
-        soup = bs4.BeautifulSoup(open(filename), "lxml")
-        for myh1 in soup.find_all("h1"):
-            if re.match("^(?!.*article-title).*$", str(myh1)):
-                h1_id = myh1['id']
-                h1_name = myh1.string
-
-                h1_title.append(str(myh1))
-
-
-                myh1['class'] = "h1"
-                myh1.string = ""
-
-                h1_span_left = soup.new_tag("span")
-                h1_span_left['class'] = "left-h1"
-                h1_span_left.string = "█▓▒░"
-
-                h1_span_title = soup.new_tag("span")
-                h1_span_title['class'] = "title-h1"
-                h1_span_title.string = "「"+h1_name+"」"
-
-                h1_span_right = soup.new_tag("span")
-                h1_span_right['class'] = "right-h1"
-                h1_span_right.string = "░▒▓█"
-
-                myh1.string.insert_before(h1_span_left)
-                myh1.span.insert_after(h1_span_right)
-                myh1.span.insert_after(h1_span_title)
-
-                h1_title.append(myh1)
-                h1_title.append(h1_name)
-
-                h1_titles_list.append(h1_title)
-            h1_title = []
-
-        for myh2 in soup.find_all("h2"):
-            if re.match("^(?!.*blog-index-year-title).*$", str(myh2)):
-                h2_id = myh2['id']
-                h2_name = myh2.string
-
-                h2_title.append(str(myh2))
-
-
-                myh2['class'] = "h2"
-                myh2.string = ""
-
-                h2_span_left = soup.new_tag("span")
-                h2_span_left['class'] = "left-h2"
-                h2_span_left.string = ".: "
-
-                h2_span_title = soup.new_tag("span")
-                h2_span_title['class'] = "title-h2"
-                h2_span_title.string = h2_name
-
-                h2_span_right = soup.new_tag("span")
-                h2_span_right['class'] = "right-h2"
-                h2_span_right.string = " :."
-
-                myh2.string.insert_before(h2_span_left)
-                myh2.span.insert_after(h2_span_right)
-                myh2.span.insert_after(h2_span_title)
-
-                h2_title.append(myh2)
-                h2_title.append(h2_name)
-
-                h2_titles_list.append(h2_title)
-            h2_title = []
-
-    tested_title_list = []
-    tested_title = []
-    for filename in filenames:
-        soup = bs4.BeautifulSoup(open(filename), "lxml")
-        if os.path.exists(filename):
-            with open(filename, 'r', encoding='utf-8') as indexmd:
-                lines = indexmd.readlines()
-
-                with open(filename, 'w', encoding='utf-8') as indexmd:
-                    for line in lines:
-                        string = ""
-                        for title in h1_titles_list:
-                            if re.match(".*"+title[0]+".*", line):
-                                string = str(title[1])
-                        for title in h2_titles_list:
-                            if re.match(".*"+title[0]+".*", line):
-                                string = str(title[1])
-                        if string != "":
-                            indexmd.write(re.sub(line, string, line))
-                        else:
-                            indexmd.write(line)
-
 
 def generate_index_and_feed():
     """Generate index.html and feeds (atom and rss)."""
@@ -327,7 +194,7 @@ def generate_index_and_feed():
     generators.generate_table()
     generators.generate_blog_list(feed)
     generators.generate_notes_list()
-    rewrite_title()
+    generators.rewrite_title()
 
     feed.updated_datetime = utils.current_datetime()
     feed.updated = ET.Element("updated")