Fix setup.py bugs caught by Yuvgoog.

1 files changed, 5 insertions, 3 deletions

diff --git a/setup.py b/setup.py
index d97f24a..95a8a0d 100644
--- a/setup.py
+++ b/setup.py
@@ -24,10 +24,12 @@ def read_description():
     readme_file = open(os.path.join(os.path.dirname(__file__), 'README.txt'))
     readme_text = readme_file.read()
     readme_file.close()
-    main_desc_regexp = r'^argparse\s*[\d.]*\s*\n=======+\s*\n(.*)Requirements '
-    main_desc, = re.findall(main_desc_regexp, readme_text, re.DOTALL)
+    main_desc_regexp = r'^argparse\s*[\w.]*\s*\n=======+\s*\n(.*)Requirements '
+    main_desc_re_c = re.compile(main_desc_regexp, re.DOTALL)
+    main_desc, = main_desc_re_c.findall(readme_text)
     avail_desc_regexp = r'Availability & Documentation\s*\n-----+\s*\n(.*)'
-    avail_desc, = re.findall(avail_desc_regexp, readme_text, re.DOTALL)
+    avail_desc_re_c = re.compile(avail_desc_regexp, re.DOTALL)
+    avail_desc, = avail_desc_re_c.findall(readme_text)
     return main_desc + avail_desc
 
 distutils.core.setup(