re-pcre: Support .group() method without args.

pfalcon · pfalcon · commit b8718dc7ccb7 · 2017-07-06T00:15:17.000+03:00
Defaults to 0.
diff --git a/re-pcre/re.py b/re-pcre/re.py
@@ -41,6 +41,8 @@ def __init__(self, s, num_matches, offsets):
         self.offsets = offsets
 
     def group(self, *n):
+        if not n:
+            return self.s[self.offsets[0]:self.offsets[1]]
         if len(n) == 1:
             return self.s[self.offsets[n[0]*2]:self.offsets[n[0]*2+1]]
         return tuple(self.s[self.offsets[i*2]:self.offsets[i*2+1]] for i in n)
diff --git a/re-pcre/test_re.py b/re-pcre/test_re.py
@@ -2,6 +2,7 @@
 
 m = re.search(r"a+", "caaab")
 assert m.group(0) == "aaa"
+assert m.group() == "aaa"
 
 m = re.match(r"(?ms)foo.*\Z", "foo\nbar")
 assert m.group(0) == "foo\nbar"
