From c0a6a23c954176e73572ff54cd6031a48aed0507 Mon Sep 17 00:00:00 2001 From: Trey Hunner Date: Sat, 31 Mar 2012 11:59:07 -0700 Subject: Fix regular expression grouping (missing period) --- pycco/main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pycco/main.py b/pycco/main.py index 68d4823..105a7cb 100644 --- a/pycco/main.py +++ b/pycco/main.py @@ -324,8 +324,8 @@ for ext, l in languages.items(): def get_language(source): """Get the current language we're documenting, based on the extension.""" - m = re.match(r'.*\.(.+)', os.path.basename(source)) - if m and languages.has_key(m.group(1)): + m = re.match(r'.*(\..+)', os.path.basename(source)) + if m and m.group(1) in languages: return languages[m.group(1)] else: source = open(source, "r") -- cgit v1.2.1