Fixed regex to better match reST output

1 files changed, 3 insertions, 2 deletions

diff --git a/paste/docsupport/gensite.py b/paste/docsupport/gensite.py
index f8bebb9..c0c7023 100755
--- a/paste/docsupport/gensite.py
+++ b/paste/docsupport/gensite.py
@@ -105,8 +105,9 @@ class RestFile(object):
             print 'No title in %s' % self.filename
             props['title'] = ''
 
-    _start_re = re.compile(r'<div class=".*?" id="contents">')
-    _end_re = re.compile(r'</div>[ \n]*</div>[ \n]*</body>')
+    #_start_re = re.compile(r'<div class=".*?" id="contents">')
+    _start_re = re.compile(r'<body>[ \n]*(?:<div.*?>[ \n]*)?<h1.*?>.*?</h1>')
+    _end_re = re.compile(r'(?:</div>[ \n]*)?</div>[ \n]*</body>')
     _bad_res = [
         (re.compile(r'<link rel="stylesheet".*?>'), ''),
         (re.compile(r'<h1 class="title">.*?</h1>'), ''),