Skip to content

Commit b8718dc

Browse files
committed
re-pcre: Support .group() method without args.
Defaults to 0.
1 parent 607c62a commit b8718dc

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

re-pcre/re.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,8 @@ def __init__(self, s, num_matches, offsets):
4141
self.offsets = offsets
4242

4343
def group(self, *n):
44+
if not n:
45+
return self.s[self.offsets[0]:self.offsets[1]]
4446
if len(n) == 1:
4547
return self.s[self.offsets[n[0]*2]:self.offsets[n[0]*2+1]]
4648
return tuple(self.s[self.offsets[i*2]:self.offsets[i*2+1]] for i in n)

re-pcre/test_re.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
m = re.search(r"a+", "caaab")
44
assert m.group(0) == "aaa"
5+
assert m.group() == "aaa"
56

67
m = re.match(r"(?ms)foo.*\Z", "foo\nbar")
78
assert m.group(0) == "foo\nbar"

0 commit comments

Comments
 (0)