From 46c7d47ff182e2d0fe831f3546041487a8952de0 Mon Sep 17 00:00:00 2001 From: neodarz Date: Sun, 26 May 2019 16:00:44 +0200 Subject: Move rewrite_title to external file --- pyblog | 135 +---------------------------------------------------------------- 1 file changed, 1 insertion(+), 134 deletions(-) (limited to 'pyblog') 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") -- cgit v1.2.1