From 757e387d84d8014abcf9b07d52d556fd483150d3 Mon Sep 17 00:00:00 2001
From: Cyrille L <echolib@a-lec.org>
Date: Fri, 30 Dec 2022 17:54:44 +0100
Subject: [PATCH] wip: (fix) module titles (remove empty lines, create divs if
 contents after title)

---
 src/var/lib/tyto/program/wip.py | 22 ++++++++++++++++------
 1 file changed, 16 insertions(+), 6 deletions(-)

diff --git a/src/var/lib/tyto/program/wip.py b/src/var/lib/tyto/program/wip.py
index babd408..248c782 100644
--- a/src/var/lib/tyto/program/wip.py
+++ b/src/var/lib/tyto/program/wip.py
@@ -500,22 +500,32 @@ def wip_titles():
   # Add div after title if needed
   for ln, line in enumerate(article_tmp2.rsplit('\n')):
     if line.startswith('<h'):
+      try:    article_tmp2.rsplit('\n')[ln + 1]
+      except: continue
+      
       if article_tmp2.rsplit('\n')[ln + 1].startswith('<h'):
         continue
       else:
         article_temp = article_temp.replace(
           line, '%s\n<div class="content_title">'%line
           )
+        continue
+    else:
+      continue
   
+  # Close div before title if needed
   article_tmp2 = article_temp
-  # Add div after title if needed
   indiv = False
   for ln, line in enumerate(article_tmp2.rsplit('\n')):
-    if line.startswith('<h') and \
-       article_tmp2.rsplit('\n')[ln + 1].startswith('<div'):
-      indiv = True
-    
-    elif line.startswith('<h') and indiv:
+    if line.startswith('<h'):
+      try:    article_tmp2.rsplit('\n')[ln + 1]
+      except: continue
+      
+      if article_tmp2.rsplit('\n')[ln + 1].startswith('<div'):
+        indiv = True
+        continue
+
+    if line.startswith('<h') and indiv:
       article_temp = article_temp.replace(
         line, '</div>\n%s'%line
         )