Forráskód Böngészése

Fix a bug with the greentexts

oobabooga 3 éve
szülő
commit
538998b43b
1 módosított fájl, 2 hozzáadás és 1 törlés
  1. 2 1
      html_generator.py

+ 2 - 1
html_generator.py

@@ -148,7 +148,8 @@ def generate_html(f):
     output += '</div>'
     output += '</div>'
     output = output.split('\n')
     output = output.split('\n')
     for i in range(len(output)):
     for i in range(len(output)):
-        output[i] = re.sub('^(&gt;[^\n]*(<br>|</div>))', '<span class="greentext">\\1</span>\n', output[i])
+        output[i] = re.sub(r'^(&gt;(.*?)(<br>|</div>))', r'<span class="greentext">\1</span>', output[i])
+        output[i] = re.sub(r'^<blockquote class="message">(&gt;(.*?)(<br>|</div>))', r'<blockquote class="message"><span class="greentext">\1</span>', output[i])
     output = '\n'.join(output)
     output = '\n'.join(output)
 
 
     return output
     return output