Added support for multiple header levels. Previously only h3 was allowed.

1 files changed, 5 insertions, 4 deletions

diff --git a/pycco/main.py b/pycco/main.py
index f9e5fa4..18bf085 100644
--- a/pycco/main.py
+++ b/pycco/main.py
@@ -171,12 +171,13 @@ def preprocess(comment, section_nr, preserve_paths=True, outdir=None):
                                                             outdir=outdir)))
 
     def replace_section_name(match):
-        return '### <span id="%(id)s" href="%(id)s">%(name)s</span>' % {
-            "id"   : sanitize_section_name(match.group(1)),
-            "name" : match.group(1)
+        return '%(lvl)s <span id="%(id)s" href="%(id)s">%(name)s</span>' % {
+            "lvl"  : re.sub('=', '#', match.group(1)),
+            "id"   : sanitize_section_name(match.group(2)),
+            "name" : match.group(2)
         }
 
-    comment = re.sub('^===(.+)===\\n', replace_section_name, comment)
+    comment = re.sub('^([=]+)([^=]+)[=]*\\n', replace_section_name, comment)
     comment = re.sub('[^`]\[\[(.+)\]\]', replace_crossref, comment)
 
     return comment