blob: a53735b07ded420ca2bc9fa636f40b9eb2241500 [file] [log] [blame]
Guido van Rossum7627c0d2000-03-31 14:58:54 +00001#
2# Secret Labs' Regular Expression Engine
Guido van Rossum7627c0d2000-03-31 14:58:54 +00003#
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +00004# convert re-style regular expression to sre pattern
Guido van Rossum7627c0d2000-03-31 14:58:54 +00005#
Fredrik Lundh770617b2001-01-14 15:06:11 +00006# Copyright (c) 1998-2001 by Secret Labs AB. All rights reserved.
Guido van Rossum7627c0d2000-03-31 14:58:54 +00007#
Fredrik Lundh29c4ba92000-08-01 18:20:07 +00008# See the sre.py file for information on usage and redistribution.
Guido van Rossum7627c0d2000-03-31 14:58:54 +00009#
10
Fred Drakeb8f22742001-09-04 19:10:20 +000011"""Internal support module for sre"""
12
Fredrik Lundh470ea5a2001-01-14 21:00:44 +000013# XXX: show string offset and offending character for all errors
14
Guido van Rossum7627c0d2000-03-31 14:58:54 +000015from sre_constants import *
16
17SPECIAL_CHARS = ".\\[{()*+?^$|"
Fredrik Lundh143328b2000-09-02 11:03:34 +000018REPEAT_CHARS = "*+?{"
Guido van Rossum7627c0d2000-03-31 14:58:54 +000019
Serhiy Storchakae2ccf562014-10-10 11:14:49 +030020DIGITS = frozenset("0123456789")
Guido van Rossumb81e70e2000-04-10 17:10:48 +000021
Serhiy Storchakae2ccf562014-10-10 11:14:49 +030022OCTDIGITS = frozenset("01234567")
23HEXDIGITS = frozenset("0123456789abcdefABCDEF")
Serhiy Storchakaa54aae02015-03-24 22:58:14 +020024ASCIILETTERS = frozenset("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ")
Guido van Rossum7627c0d2000-03-31 14:58:54 +000025
Serhiy Storchakae2ccf562014-10-10 11:14:49 +030026WHITESPACE = frozenset(" \t\n\r\v\f")
27
Raymond Hettingerdf1b6992014-11-09 15:56:33 -080028_REPEATCODES = frozenset({MIN_REPEAT, MAX_REPEAT})
29_UNITCODES = frozenset({ANY, RANGE, IN, LITERAL, NOT_LITERAL, CATEGORY})
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +000030
Guido van Rossum7627c0d2000-03-31 14:58:54 +000031ESCAPES = {
Fredrik Lundhf2989b22001-02-18 12:05:16 +000032 r"\a": (LITERAL, ord("\a")),
33 r"\b": (LITERAL, ord("\b")),
34 r"\f": (LITERAL, ord("\f")),
35 r"\n": (LITERAL, ord("\n")),
36 r"\r": (LITERAL, ord("\r")),
37 r"\t": (LITERAL, ord("\t")),
38 r"\v": (LITERAL, ord("\v")),
Fredrik Lundh0640e112000-06-30 13:55:15 +000039 r"\\": (LITERAL, ord("\\"))
Guido van Rossum7627c0d2000-03-31 14:58:54 +000040}
41
42CATEGORIES = {
Fredrik Lundh770617b2001-01-14 15:06:11 +000043 r"\A": (AT, AT_BEGINNING_STRING), # start of string
Fredrik Lundh01016fe2000-06-30 00:27:46 +000044 r"\b": (AT, AT_BOUNDARY),
45 r"\B": (AT, AT_NON_BOUNDARY),
46 r"\d": (IN, [(CATEGORY, CATEGORY_DIGIT)]),
47 r"\D": (IN, [(CATEGORY, CATEGORY_NOT_DIGIT)]),
48 r"\s": (IN, [(CATEGORY, CATEGORY_SPACE)]),
49 r"\S": (IN, [(CATEGORY, CATEGORY_NOT_SPACE)]),
50 r"\w": (IN, [(CATEGORY, CATEGORY_WORD)]),
51 r"\W": (IN, [(CATEGORY, CATEGORY_NOT_WORD)]),
Fredrik Lundh770617b2001-01-14 15:06:11 +000052 r"\Z": (AT, AT_END_STRING), # end of string
Guido van Rossum7627c0d2000-03-31 14:58:54 +000053}
54
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +000055FLAGS = {
Fredrik Lundh436c3d582000-06-29 08:58:44 +000056 # standard flags
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +000057 "i": SRE_FLAG_IGNORECASE,
58 "L": SRE_FLAG_LOCALE,
59 "m": SRE_FLAG_MULTILINE,
60 "s": SRE_FLAG_DOTALL,
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +000061 "x": SRE_FLAG_VERBOSE,
Fredrik Lundh436c3d582000-06-29 08:58:44 +000062 # extensions
Antoine Pitroufd036452008-08-19 17:56:33 +000063 "a": SRE_FLAG_ASCII,
Fredrik Lundh436c3d582000-06-29 08:58:44 +000064 "t": SRE_FLAG_TEMPLATE,
65 "u": SRE_FLAG_UNICODE,
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +000066}
67
Serhiy Storchaka3557b052017-10-24 23:31:42 +030068TYPE_FLAGS = SRE_FLAG_ASCII | SRE_FLAG_LOCALE | SRE_FLAG_UNICODE
69GLOBAL_FLAGS = SRE_FLAG_DEBUG | SRE_FLAG_TEMPLATE
Serhiy Storchakabe9a4e52016-09-10 00:57:55 +030070
71class Verbose(Exception):
72 pass
73
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +000074class Pattern:
75 # master pattern object. keeps track of global attributes
Guido van Rossum7627c0d2000-03-31 14:58:54 +000076 def __init__(self):
Fredrik Lundh90a07912000-06-30 07:50:59 +000077 self.flags = 0
Fredrik Lundh90a07912000-06-30 07:50:59 +000078 self.groupdict = {}
Serhiy Storchakab5d0a212015-11-05 17:49:26 +020079 self.groupwidths = [None] # group 0
Serhiy Storchaka4eea62f2015-02-21 10:07:35 +020080 self.lookbehindgroups = None
81 @property
82 def groups(self):
Serhiy Storchakab5d0a212015-11-05 17:49:26 +020083 return len(self.groupwidths)
Fredrik Lundhebc37b22000-10-28 19:30:41 +000084 def opengroup(self, name=None):
Fredrik Lundh90a07912000-06-30 07:50:59 +000085 gid = self.groups
Serhiy Storchakab5d0a212015-11-05 17:49:26 +020086 self.groupwidths.append(None)
Serhiy Storchaka9baa5b22014-09-29 22:49:23 +030087 if self.groups > MAXGROUPS:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +020088 raise error("too many groups")
Raymond Hettingerf13eb552002-06-02 00:40:05 +000089 if name is not None:
Tim Peters75335872001-11-03 19:35:43 +000090 ogid = self.groupdict.get(name, None)
91 if ogid is not None:
Serhiy Storchakaad446d52014-11-10 13:49:00 +020092 raise error("redefinition of group name %r as group %d; "
93 "was group %d" % (name, gid, ogid))
Fredrik Lundh90a07912000-06-30 07:50:59 +000094 self.groupdict[name] = gid
95 return gid
Serhiy Storchaka4eea62f2015-02-21 10:07:35 +020096 def closegroup(self, gid, p):
Serhiy Storchakab5d0a212015-11-05 17:49:26 +020097 self.groupwidths[gid] = p.getwidth()
Fredrik Lundhebc37b22000-10-28 19:30:41 +000098 def checkgroup(self, gid):
Serhiy Storchakab5d0a212015-11-05 17:49:26 +020099 return gid < self.groups and self.groupwidths[gid] is not None
Serhiy Storchaka4eea62f2015-02-21 10:07:35 +0200100
101 def checklookbehindgroup(self, gid, source):
102 if self.lookbehindgroups is not None:
103 if not self.checkgroup(gid):
104 raise source.error('cannot refer to an open group')
105 if gid >= self.lookbehindgroups:
106 raise source.error('cannot refer to group defined in the same '
107 'lookbehind subpattern')
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000108
109class SubPattern:
110 # a subpattern, in intermediate form
111 def __init__(self, pattern, data=None):
Fredrik Lundh90a07912000-06-30 07:50:59 +0000112 self.pattern = pattern
Raymond Hettingerf13eb552002-06-02 00:40:05 +0000113 if data is None:
Fredrik Lundh90a07912000-06-30 07:50:59 +0000114 data = []
115 self.data = data
116 self.width = None
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300117
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000118 def dump(self, level=0):
Serhiy Storchaka44dae8b2014-09-21 22:47:55 +0300119 nl = True
Guido van Rossum13257902007-06-07 23:15:56 +0000120 seqtypes = (tuple, list)
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000121 for op, av in self.data:
Serhiy Storchakac7f7d382014-11-09 20:48:36 +0200122 print(level*" " + str(op), end='')
Serhiy Storchakaab140882014-11-11 21:13:28 +0200123 if op is IN:
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000124 # member sublanguage
Serhiy Storchaka44dae8b2014-09-21 22:47:55 +0300125 print()
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000126 for op, a in av:
Serhiy Storchakac7f7d382014-11-09 20:48:36 +0200127 print((level+1)*" " + str(op), a)
Serhiy Storchakaab140882014-11-11 21:13:28 +0200128 elif op is BRANCH:
Serhiy Storchaka44dae8b2014-09-21 22:47:55 +0300129 print()
130 for i, a in enumerate(av[1]):
131 if i:
Serhiy Storchakac7f7d382014-11-09 20:48:36 +0200132 print(level*" " + "OR")
Serhiy Storchaka44dae8b2014-09-21 22:47:55 +0300133 a.dump(level+1)
Serhiy Storchakaab140882014-11-11 21:13:28 +0200134 elif op is GROUPREF_EXISTS:
Serhiy Storchaka44dae8b2014-09-21 22:47:55 +0300135 condgroup, item_yes, item_no = av
136 print('', condgroup)
137 item_yes.dump(level+1)
138 if item_no:
Serhiy Storchakac7f7d382014-11-09 20:48:36 +0200139 print(level*" " + "ELSE")
Serhiy Storchaka44dae8b2014-09-21 22:47:55 +0300140 item_no.dump(level+1)
Guido van Rossum13257902007-06-07 23:15:56 +0000141 elif isinstance(av, seqtypes):
Serhiy Storchaka44dae8b2014-09-21 22:47:55 +0300142 nl = False
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000143 for a in av:
144 if isinstance(a, SubPattern):
Serhiy Storchaka44dae8b2014-09-21 22:47:55 +0300145 if not nl:
146 print()
147 a.dump(level+1)
148 nl = True
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000149 else:
Serhiy Storchaka44dae8b2014-09-21 22:47:55 +0300150 if not nl:
151 print(' ', end='')
152 print(a, end='')
153 nl = False
154 if not nl:
155 print()
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000156 else:
Serhiy Storchaka44dae8b2014-09-21 22:47:55 +0300157 print('', av)
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000158 def __repr__(self):
Fredrik Lundh90a07912000-06-30 07:50:59 +0000159 return repr(self.data)
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000160 def __len__(self):
Fredrik Lundh90a07912000-06-30 07:50:59 +0000161 return len(self.data)
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000162 def __delitem__(self, index):
Fredrik Lundh90a07912000-06-30 07:50:59 +0000163 del self.data[index]
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000164 def __getitem__(self, index):
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000165 if isinstance(index, slice):
166 return SubPattern(self.pattern, self.data[index])
Fredrik Lundh90a07912000-06-30 07:50:59 +0000167 return self.data[index]
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000168 def __setitem__(self, index, code):
Fredrik Lundh90a07912000-06-30 07:50:59 +0000169 self.data[index] = code
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000170 def insert(self, index, code):
Fredrik Lundh90a07912000-06-30 07:50:59 +0000171 self.data.insert(index, code)
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000172 def append(self, code):
Fredrik Lundh90a07912000-06-30 07:50:59 +0000173 self.data.append(code)
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000174 def getwidth(self):
Fredrik Lundh90a07912000-06-30 07:50:59 +0000175 # determine the width (min, max) for this subpattern
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300176 if self.width is not None:
Fredrik Lundh90a07912000-06-30 07:50:59 +0000177 return self.width
Guido van Rossume2a383d2007-01-15 16:59:06 +0000178 lo = hi = 0
Fredrik Lundh90a07912000-06-30 07:50:59 +0000179 for op, av in self.data:
180 if op is BRANCH:
Serhiy Storchaka9d965422013-08-19 22:50:54 +0300181 i = MAXREPEAT - 1
Fredrik Lundh2f2c67d2000-08-01 21:05:41 +0000182 j = 0
Fredrik Lundh90a07912000-06-30 07:50:59 +0000183 for av in av[1]:
Fredrik Lundh2f2c67d2000-08-01 21:05:41 +0000184 l, h = av.getwidth()
185 i = min(i, l)
Fredrik Lundhe1869832000-08-01 22:47:49 +0000186 j = max(j, h)
Fredrik Lundh90a07912000-06-30 07:50:59 +0000187 lo = lo + i
188 hi = hi + j
189 elif op is CALL:
190 i, j = av.getwidth()
191 lo = lo + i
192 hi = hi + j
193 elif op is SUBPATTERN:
Serhiy Storchakabe9a4e52016-09-10 00:57:55 +0300194 i, j = av[-1].getwidth()
Fredrik Lundh90a07912000-06-30 07:50:59 +0000195 lo = lo + i
196 hi = hi + j
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300197 elif op in _REPEATCODES:
Fredrik Lundh90a07912000-06-30 07:50:59 +0000198 i, j = av[2].getwidth()
Serhiy Storchaka9d965422013-08-19 22:50:54 +0300199 lo = lo + i * av[0]
200 hi = hi + j * av[1]
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300201 elif op in _UNITCODES:
Fredrik Lundh90a07912000-06-30 07:50:59 +0000202 lo = lo + 1
203 hi = hi + 1
Serhiy Storchaka4eea62f2015-02-21 10:07:35 +0200204 elif op is GROUPREF:
Serhiy Storchakab5d0a212015-11-05 17:49:26 +0200205 i, j = self.pattern.groupwidths[av]
Serhiy Storchaka4eea62f2015-02-21 10:07:35 +0200206 lo = lo + i
207 hi = hi + j
208 elif op is GROUPREF_EXISTS:
209 i, j = av[1].getwidth()
210 if av[2] is not None:
211 l, h = av[2].getwidth()
212 i = min(i, l)
213 j = max(j, h)
214 else:
215 i = 0
216 lo = lo + i
217 hi = hi + j
218 elif op is SUCCESS:
Fredrik Lundh90a07912000-06-30 07:50:59 +0000219 break
Serhiy Storchaka9d965422013-08-19 22:50:54 +0300220 self.width = min(lo, MAXREPEAT - 1), min(hi, MAXREPEAT)
Fredrik Lundh90a07912000-06-30 07:50:59 +0000221 return self.width
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000222
223class Tokenizer:
224 def __init__(self, string):
Antoine Pitrou463badf2012-06-23 13:29:19 +0200225 self.istext = isinstance(string, str)
Serhiy Storchakaad446d52014-11-10 13:49:00 +0200226 self.string = string
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300227 if not self.istext:
228 string = str(string, 'latin1')
Serhiy Storchakaad446d52014-11-10 13:49:00 +0200229 self.decoded_string = string
Fredrik Lundhc13222c2000-07-01 23:49:14 +0000230 self.index = 0
Serhiy Storchakab99c1322014-11-10 14:38:16 +0200231 self.next = None
Fredrik Lundhc13222c2000-07-01 23:49:14 +0000232 self.__next()
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000233 def __next(self):
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300234 index = self.index
235 try:
Serhiy Storchakaad446d52014-11-10 13:49:00 +0200236 char = self.decoded_string[index]
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300237 except IndexError:
Fredrik Lundhc13222c2000-07-01 23:49:14 +0000238 self.next = None
239 return
Guido van Rossum75a902d2007-10-19 22:06:24 +0000240 if char == "\\":
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300241 index += 1
Fredrik Lundh90a07912000-06-30 07:50:59 +0000242 try:
Serhiy Storchakaad446d52014-11-10 13:49:00 +0200243 char += self.decoded_string[index]
Fredrik Lundh90a07912000-06-30 07:50:59 +0000244 except IndexError:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200245 raise error("bad escape (end of pattern)",
Serhiy Storchaka1b2004f2014-11-10 18:28:53 +0200246 self.string, len(self.string) - 1) from None
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300247 self.index = index + 1
Fredrik Lundhc13222c2000-07-01 23:49:14 +0000248 self.next = char
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300249 def match(self, char):
Fredrik Lundh90a07912000-06-30 07:50:59 +0000250 if char == self.next:
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300251 self.__next()
252 return True
253 return False
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000254 def get(self):
Fredrik Lundh90a07912000-06-30 07:50:59 +0000255 this = self.next
Fredrik Lundhc13222c2000-07-01 23:49:14 +0000256 self.__next()
Fredrik Lundh90a07912000-06-30 07:50:59 +0000257 return this
Antoine Pitrou463badf2012-06-23 13:29:19 +0200258 def getwhile(self, n, charset):
259 result = ''
260 for _ in range(n):
261 c = self.next
262 if c not in charset:
263 break
264 result += c
265 self.__next()
266 return result
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300267 def getuntil(self, terminator):
268 result = ''
269 while True:
270 c = self.next
271 self.__next()
272 if c is None:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200273 if not result:
274 raise self.error("missing group name")
275 raise self.error("missing %s, unterminated name" % terminator,
276 len(result))
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300277 if c == terminator:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200278 if not result:
279 raise self.error("missing group name", 1)
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300280 break
281 result += c
282 return result
Serhiy Storchakabd48d272016-09-11 12:50:02 +0300283 @property
284 def pos(self):
285 return self.index - len(self.next or '')
Fredrik Lundhc13222c2000-07-01 23:49:14 +0000286 def tell(self):
Serhiy Storchakaad446d52014-11-10 13:49:00 +0200287 return self.index - len(self.next or '')
Fredrik Lundhc13222c2000-07-01 23:49:14 +0000288 def seek(self, index):
Serhiy Storchakaad446d52014-11-10 13:49:00 +0200289 self.index = index
290 self.__next()
291
292 def error(self, msg, offset=0):
293 return error(msg, self.string, self.tell() - offset)
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000294
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +0000295def _class_escape(source, escape):
296 # handle escape code inside character class
297 code = ESCAPES.get(escape)
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000298 if code:
Fredrik Lundh90a07912000-06-30 07:50:59 +0000299 return code
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +0000300 code = CATEGORIES.get(escape)
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300301 if code and code[0] is IN:
Fredrik Lundh90a07912000-06-30 07:50:59 +0000302 return code
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000303 try:
Gustavo Niemeyera01a2ee2004-09-03 17:06:10 +0000304 c = escape[1:2]
305 if c == "x":
Fredrik Lundh0c4fdba2000-08-31 22:57:55 +0000306 # hexadecimal escape (exactly two digits)
Antoine Pitrou463badf2012-06-23 13:29:19 +0200307 escape += source.getwhile(2, HEXDIGITS)
308 if len(escape) != 4:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200309 raise source.error("incomplete escape %s" % escape, len(escape))
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300310 return LITERAL, int(escape[2:], 16)
Antoine Pitrou463badf2012-06-23 13:29:19 +0200311 elif c == "u" and source.istext:
312 # unicode escape (exactly four digits)
313 escape += source.getwhile(4, HEXDIGITS)
314 if len(escape) != 6:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200315 raise source.error("incomplete escape %s" % escape, len(escape))
Antoine Pitrou463badf2012-06-23 13:29:19 +0200316 return LITERAL, int(escape[2:], 16)
317 elif c == "U" and source.istext:
318 # unicode escape (exactly eight digits)
319 escape += source.getwhile(8, HEXDIGITS)
320 if len(escape) != 10:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200321 raise source.error("incomplete escape %s" % escape, len(escape))
Antoine Pitrou463badf2012-06-23 13:29:19 +0200322 c = int(escape[2:], 16)
323 chr(c) # raise ValueError for invalid code
324 return LITERAL, c
Gustavo Niemeyera01a2ee2004-09-03 17:06:10 +0000325 elif c in OCTDIGITS:
Fredrik Lundh0c4fdba2000-08-31 22:57:55 +0000326 # octal escape (up to three digits)
Antoine Pitrou463badf2012-06-23 13:29:19 +0200327 escape += source.getwhile(2, OCTDIGITS)
Serhiy Storchakac563caf2014-09-23 23:22:41 +0300328 c = int(escape[1:], 8)
329 if c > 0o377:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200330 raise source.error('octal escape value %s outside of '
Serhiy Storchakaad446d52014-11-10 13:49:00 +0200331 'range 0-0o377' % escape, len(escape))
Serhiy Storchakac563caf2014-09-23 23:22:41 +0300332 return LITERAL, c
Gustavo Niemeyera01a2ee2004-09-03 17:06:10 +0000333 elif c in DIGITS:
Antoine Pitrou463badf2012-06-23 13:29:19 +0200334 raise ValueError
Fredrik Lundh90a07912000-06-30 07:50:59 +0000335 if len(escape) == 2:
Serhiy Storchakaa54aae02015-03-24 22:58:14 +0200336 if c in ASCIILETTERS:
Serhiy Storchaka9bd85b82016-06-11 19:15:00 +0300337 raise source.error('bad escape %s' % escape, len(escape))
Fredrik Lundh0640e112000-06-30 13:55:15 +0000338 return LITERAL, ord(escape[1])
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000339 except ValueError:
Fredrik Lundh90a07912000-06-30 07:50:59 +0000340 pass
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200341 raise source.error("bad escape %s" % escape, len(escape))
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000342
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +0000343def _escape(source, escape, state):
344 # handle escape code in expression
345 code = CATEGORIES.get(escape)
346 if code:
Fredrik Lundh90a07912000-06-30 07:50:59 +0000347 return code
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +0000348 code = ESCAPES.get(escape)
349 if code:
Fredrik Lundh90a07912000-06-30 07:50:59 +0000350 return code
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +0000351 try:
Gustavo Niemeyera01a2ee2004-09-03 17:06:10 +0000352 c = escape[1:2]
353 if c == "x":
Fredrik Lundh0c4fdba2000-08-31 22:57:55 +0000354 # hexadecimal escape
Antoine Pitrou463badf2012-06-23 13:29:19 +0200355 escape += source.getwhile(2, HEXDIGITS)
Fredrik Lundh143328b2000-09-02 11:03:34 +0000356 if len(escape) != 4:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200357 raise source.error("incomplete escape %s" % escape, len(escape))
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300358 return LITERAL, int(escape[2:], 16)
Antoine Pitrou463badf2012-06-23 13:29:19 +0200359 elif c == "u" and source.istext:
360 # unicode escape (exactly four digits)
361 escape += source.getwhile(4, HEXDIGITS)
362 if len(escape) != 6:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200363 raise source.error("incomplete escape %s" % escape, len(escape))
Antoine Pitrou463badf2012-06-23 13:29:19 +0200364 return LITERAL, int(escape[2:], 16)
365 elif c == "U" and source.istext:
366 # unicode escape (exactly eight digits)
367 escape += source.getwhile(8, HEXDIGITS)
368 if len(escape) != 10:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200369 raise source.error("incomplete escape %s" % escape, len(escape))
Antoine Pitrou463badf2012-06-23 13:29:19 +0200370 c = int(escape[2:], 16)
371 chr(c) # raise ValueError for invalid code
372 return LITERAL, c
Gustavo Niemeyera01a2ee2004-09-03 17:06:10 +0000373 elif c == "0":
Fredrik Lundh0c4fdba2000-08-31 22:57:55 +0000374 # octal escape
Antoine Pitrou463badf2012-06-23 13:29:19 +0200375 escape += source.getwhile(2, OCTDIGITS)
Serhiy Storchakac563caf2014-09-23 23:22:41 +0300376 return LITERAL, int(escape[1:], 8)
Gustavo Niemeyera01a2ee2004-09-03 17:06:10 +0000377 elif c in DIGITS:
Fredrik Lundh0c4fdba2000-08-31 22:57:55 +0000378 # octal escape *or* decimal group reference (sigh)
Fredrik Lundh0c4fdba2000-08-31 22:57:55 +0000379 if source.next in DIGITS:
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300380 escape += source.get()
Fredrik Lundh143328b2000-09-02 11:03:34 +0000381 if (escape[1] in OCTDIGITS and escape[2] in OCTDIGITS and
382 source.next in OCTDIGITS):
Fredrik Lundh0c4fdba2000-08-31 22:57:55 +0000383 # got three octal digits; this is an octal escape
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300384 escape += source.get()
Serhiy Storchakac563caf2014-09-23 23:22:41 +0300385 c = int(escape[1:], 8)
386 if c > 0o377:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200387 raise source.error('octal escape value %s outside of '
Serhiy Storchakaad446d52014-11-10 13:49:00 +0200388 'range 0-0o377' % escape,
389 len(escape))
Serhiy Storchakac563caf2014-09-23 23:22:41 +0300390 return LITERAL, c
Gustavo Niemeyera01a2ee2004-09-03 17:06:10 +0000391 # not an octal escape, so this is a group reference
392 group = int(escape[1:])
393 if group < state.groups:
Fredrik Lundhebc37b22000-10-28 19:30:41 +0000394 if not state.checkgroup(group):
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200395 raise source.error("cannot refer to an open group",
Serhiy Storchakaad446d52014-11-10 13:49:00 +0200396 len(escape))
Serhiy Storchaka4eea62f2015-02-21 10:07:35 +0200397 state.checklookbehindgroup(group, source)
Fredrik Lundh0c4fdba2000-08-31 22:57:55 +0000398 return GROUPREF, group
Serhiy Storchaka662cef62016-10-23 12:11:19 +0300399 raise source.error("invalid group reference %d" % group, len(escape) - 1)
Fredrik Lundh90a07912000-06-30 07:50:59 +0000400 if len(escape) == 2:
Serhiy Storchakaa54aae02015-03-24 22:58:14 +0200401 if c in ASCIILETTERS:
Serhiy Storchaka9bd85b82016-06-11 19:15:00 +0300402 raise source.error("bad escape %s" % escape, len(escape))
Fredrik Lundh0640e112000-06-30 13:55:15 +0000403 return LITERAL, ord(escape[1])
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +0000404 except ValueError:
Fredrik Lundh90a07912000-06-30 07:50:59 +0000405 pass
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200406 raise source.error("bad escape %s" % escape, len(escape))
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +0000407
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300408def _uniq(items):
409 if len(set(items)) == len(items):
410 return items
411 newitems = []
412 for item in items:
413 if item not in newitems:
414 newitems.append(item)
415 return newitems
416
Serhiy Storchakac7ac7282017-05-16 15:16:15 +0300417def _parse_sub(source, state, verbose, nested):
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000418 # parse an alternation: a|b|c
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000419
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000420 items = []
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000421 itemsappend = items.append
422 sourcematch = source.match
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200423 start = source.tell()
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300424 while True:
Serhiy Storchakac7ac7282017-05-16 15:16:15 +0300425 itemsappend(_parse(source, state, verbose, nested + 1,
426 not nested and not items))
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300427 if not sourcematch("|"):
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000428 break
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000429
430 if len(items) == 1:
431 return items[0]
432
433 subpattern = SubPattern(state)
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +0000434
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000435 # check if all items share a common prefix
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300436 while True:
Fredrik Lundh90a07912000-06-30 07:50:59 +0000437 prefix = None
438 for item in items:
439 if not item:
440 break
441 if prefix is None:
442 prefix = item[0]
443 elif item[0] != prefix:
444 break
445 else:
446 # all subitems start with a common "prefix".
447 # move it out of the branch
448 for item in items:
449 del item[0]
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300450 subpattern.append(prefix)
Fredrik Lundh90a07912000-06-30 07:50:59 +0000451 continue # check next one
452 break
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000453
454 # check if the branch can be replaced by a character set
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300455 set = []
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000456 for item in items:
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300457 if len(item) != 1:
458 break
459 op, av = item[0]
460 if op is LITERAL:
461 set.append((op, av))
462 elif op is IN and av[0][0] is not NEGATE:
463 set.extend(av)
464 else:
Fredrik Lundh90a07912000-06-30 07:50:59 +0000465 break
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000466 else:
Fredrik Lundh90a07912000-06-30 07:50:59 +0000467 # we can store this as a character set instead of a
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000468 # branch (the compiler may optimize this even more)
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300469 subpattern.append((IN, _uniq(set)))
Fredrik Lundh90a07912000-06-30 07:50:59 +0000470 return subpattern
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000471
472 subpattern.append((BRANCH, (None, items)))
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +0000473 return subpattern
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000474
Serhiy Storchakac7ac7282017-05-16 15:16:15 +0300475def _parse(source, state, verbose, nested, first=False):
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000476 # parse a simple pattern
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +0000477 subpattern = SubPattern(state)
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000478
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000479 # precompute constants into local variables
480 subpatternappend = subpattern.append
481 sourceget = source.get
482 sourcematch = source.match
483 _len = len
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300484 _ord = ord
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000485
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300486 while True:
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000487
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300488 this = source.next
Fredrik Lundh90a07912000-06-30 07:50:59 +0000489 if this is None:
490 break # end of pattern
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300491 if this in "|)":
492 break # end of subpattern
493 sourceget()
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000494
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300495 if verbose:
Fredrik Lundh90a07912000-06-30 07:50:59 +0000496 # skip whitespace and comments
497 if this in WHITESPACE:
498 continue
499 if this == "#":
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300500 while True:
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000501 this = sourceget()
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300502 if this is None or this == "\n":
Fredrik Lundh90a07912000-06-30 07:50:59 +0000503 break
504 continue
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +0000505
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300506 if this[0] == "\\":
507 code = _escape(source, this, state)
508 subpatternappend(code)
509
510 elif this not in SPECIAL_CHARS:
511 subpatternappend((LITERAL, _ord(this)))
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000512
Fredrik Lundh90a07912000-06-30 07:50:59 +0000513 elif this == "[":
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200514 here = source.tell() - 1
Fredrik Lundh90a07912000-06-30 07:50:59 +0000515 # character set
516 set = []
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000517 setappend = set.append
518## if sourcematch(":"):
Fredrik Lundh90a07912000-06-30 07:50:59 +0000519## pass # handle character classes
Serhiy Storchaka05cb7282017-11-16 12:38:26 +0200520 if source.next == '[':
521 import warnings
522 warnings.warn(
523 'Possible nested set at position %d' % source.tell(),
524 FutureWarning, stacklevel=nested + 6
525 )
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300526 negate = sourcematch("^")
Fredrik Lundh90a07912000-06-30 07:50:59 +0000527 # check remaining characters
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300528 while True:
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000529 this = sourceget()
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300530 if this is None:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200531 raise source.error("unterminated character set",
532 source.tell() - here)
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300533 if this == "]" and set:
Fredrik Lundh90a07912000-06-30 07:50:59 +0000534 break
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300535 elif this[0] == "\\":
Fredrik Lundh90a07912000-06-30 07:50:59 +0000536 code1 = _class_escape(source, this)
Fredrik Lundh90a07912000-06-30 07:50:59 +0000537 else:
Serhiy Storchaka05cb7282017-11-16 12:38:26 +0200538 if set and this in '-&~|' and source.next == this:
539 import warnings
540 warnings.warn(
541 'Possible set %s at position %d' % (
542 'difference' if this == '-' else
543 'intersection' if this == '&' else
544 'symmetric difference' if this == '~' else
545 'union',
546 source.tell() - 1),
547 FutureWarning, stacklevel=nested + 6
548 )
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300549 code1 = LITERAL, _ord(this)
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000550 if sourcematch("-"):
Fredrik Lundh90a07912000-06-30 07:50:59 +0000551 # potential range
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200552 that = sourceget()
553 if that is None:
554 raise source.error("unterminated character set",
555 source.tell() - here)
556 if that == "]":
Fredrik Lundh025468d2000-10-07 10:16:19 +0000557 if code1[0] is IN:
558 code1 = code1[1][0]
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000559 setappend(code1)
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300560 setappend((LITERAL, _ord("-")))
Fredrik Lundh90a07912000-06-30 07:50:59 +0000561 break
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200562 if that[0] == "\\":
563 code2 = _class_escape(source, that)
Guido van Rossum41c99e72003-04-14 17:59:34 +0000564 else:
Serhiy Storchaka05cb7282017-11-16 12:38:26 +0200565 if that == '-':
566 import warnings
567 warnings.warn(
568 'Possible set difference at position %d' % (
569 source.tell() - 2),
570 FutureWarning, stacklevel=nested + 6
571 )
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200572 code2 = LITERAL, _ord(that)
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300573 if code1[0] != LITERAL or code2[0] != LITERAL:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200574 msg = "bad character range %s-%s" % (this, that)
575 raise source.error(msg, len(this) + 1 + len(that))
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300576 lo = code1[1]
577 hi = code2[1]
578 if hi < lo:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200579 msg = "bad character range %s-%s" % (this, that)
580 raise source.error(msg, len(this) + 1 + len(that))
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300581 setappend((RANGE, (lo, hi)))
Fredrik Lundh90a07912000-06-30 07:50:59 +0000582 else:
583 if code1[0] is IN:
584 code1 = code1[1][0]
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000585 setappend(code1)
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000586
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300587 set = _uniq(set)
Fredrik Lundh770617b2001-01-14 15:06:11 +0000588 # XXX: <fl> should move set optimization to compiler!
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300589 if _len(set) == 1 and set[0][0] is LITERAL:
590 # optimization
591 if negate:
592 subpatternappend((NOT_LITERAL, set[0][1]))
593 else:
594 subpatternappend(set[0])
Fredrik Lundh90a07912000-06-30 07:50:59 +0000595 else:
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300596 if negate:
597 set.insert(0, (NEGATE, None))
598 # charmap optimization can't be added here because
599 # global flags still are not known
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000600 subpatternappend((IN, set))
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000601
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300602 elif this in REPEAT_CHARS:
Fredrik Lundh90a07912000-06-30 07:50:59 +0000603 # repeat previous item
Serhiy Storchakaad446d52014-11-10 13:49:00 +0200604 here = source.tell()
Fredrik Lundh90a07912000-06-30 07:50:59 +0000605 if this == "?":
606 min, max = 0, 1
607 elif this == "*":
608 min, max = 0, MAXREPEAT
Fredrik Lundhc0c7ee32001-02-18 21:04:48 +0000609
Fredrik Lundh90a07912000-06-30 07:50:59 +0000610 elif this == "+":
611 min, max = 1, MAXREPEAT
612 elif this == "{":
Gustavo Niemeyer6fa0c5a2005-09-14 08:54:39 +0000613 if source.next == "}":
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300614 subpatternappend((LITERAL, _ord(this)))
Gustavo Niemeyer6fa0c5a2005-09-14 08:54:39 +0000615 continue
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300616
Fredrik Lundh90a07912000-06-30 07:50:59 +0000617 min, max = 0, MAXREPEAT
618 lo = hi = ""
619 while source.next in DIGITS:
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300620 lo += sourceget()
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000621 if sourcematch(","):
Fredrik Lundh90a07912000-06-30 07:50:59 +0000622 while source.next in DIGITS:
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300623 hi += sourceget()
Fredrik Lundh90a07912000-06-30 07:50:59 +0000624 else:
625 hi = lo
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000626 if not sourcematch("}"):
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300627 subpatternappend((LITERAL, _ord(this)))
Fredrik Lundhc13222c2000-07-01 23:49:14 +0000628 source.seek(here)
629 continue
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300630
Fredrik Lundh90a07912000-06-30 07:50:59 +0000631 if lo:
Barry Warsaw8bee7612004-08-25 02:22:30 +0000632 min = int(lo)
Serhiy Storchaka70ca0212013-02-16 16:47:47 +0200633 if min >= MAXREPEAT:
634 raise OverflowError("the repetition number is too large")
Fredrik Lundh90a07912000-06-30 07:50:59 +0000635 if hi:
Barry Warsaw8bee7612004-08-25 02:22:30 +0000636 max = int(hi)
Serhiy Storchaka70ca0212013-02-16 16:47:47 +0200637 if max >= MAXREPEAT:
638 raise OverflowError("the repetition number is too large")
639 if max < min:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200640 raise source.error("min repeat greater than max repeat",
Serhiy Storchakaad446d52014-11-10 13:49:00 +0200641 source.tell() - here)
Fredrik Lundh90a07912000-06-30 07:50:59 +0000642 else:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200643 raise AssertionError("unsupported quantifier %r" % (char,))
Fredrik Lundh90a07912000-06-30 07:50:59 +0000644 # figure out which item to repeat
645 if subpattern:
646 item = subpattern[-1:]
647 else:
Fredrik Lundhc0c7ee32001-02-18 21:04:48 +0000648 item = None
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300649 if not item or item[0][0] is AT:
Serhiy Storchakaad446d52014-11-10 13:49:00 +0200650 raise source.error("nothing to repeat",
651 source.tell() - here + len(this))
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300652 if item[0][0] in _REPEATCODES:
Serhiy Storchakaad446d52014-11-10 13:49:00 +0200653 raise source.error("multiple repeat",
654 source.tell() - here + len(this))
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300655 if item[0][0] is SUBPATTERN:
656 group, add_flags, del_flags, p = item[0][1]
657 if group is None and not add_flags and not del_flags:
658 item = p
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000659 if sourcematch("?"):
Fredrik Lundh90a07912000-06-30 07:50:59 +0000660 subpattern[-1] = (MIN_REPEAT, (min, max, item))
661 else:
662 subpattern[-1] = (MAX_REPEAT, (min, max, item))
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +0000663
Fredrik Lundh90a07912000-06-30 07:50:59 +0000664 elif this == ".":
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000665 subpatternappend((ANY, None))
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +0000666
Fredrik Lundh90a07912000-06-30 07:50:59 +0000667 elif this == "(":
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200668 start = source.tell() - 1
669 group = True
Fredrik Lundh90a07912000-06-30 07:50:59 +0000670 name = None
Serhiy Storchakabe9a4e52016-09-10 00:57:55 +0300671 add_flags = 0
672 del_flags = 0
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000673 if sourcematch("?"):
Fredrik Lundh90a07912000-06-30 07:50:59 +0000674 # options
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300675 char = sourceget()
676 if char is None:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200677 raise source.error("unexpected end of pattern")
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300678 if char == "P":
Fredrik Lundh90a07912000-06-30 07:50:59 +0000679 # python extensions
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000680 if sourcematch("<"):
Fredrik Lundh90a07912000-06-30 07:50:59 +0000681 # named group: skip forward to end of name
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300682 name = source.getuntil(">")
Georg Brandl1d472b72013-04-14 11:40:00 +0200683 if not name.isidentifier():
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200684 msg = "bad character in group name %r" % name
685 raise source.error(msg, len(name) + 1)
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000686 elif sourcematch("="):
Fredrik Lundh90a07912000-06-30 07:50:59 +0000687 # named backreference
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300688 name = source.getuntil(")")
Georg Brandl1d472b72013-04-14 11:40:00 +0200689 if not name.isidentifier():
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200690 msg = "bad character in group name %r" % name
691 raise source.error(msg, len(name) + 1)
Fredrik Lundhb71624e2000-06-30 09:13:06 +0000692 gid = state.groupdict.get(name)
693 if gid is None:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200694 msg = "unknown group name %r" % name
Serhiy Storchakaad446d52014-11-10 13:49:00 +0200695 raise source.error(msg, len(name) + 1)
Serhiy Storchaka485407c2015-07-18 23:27:00 +0300696 if not state.checkgroup(gid):
697 raise source.error("cannot refer to an open group",
698 len(name) + 1)
Serhiy Storchaka4eea62f2015-02-21 10:07:35 +0200699 state.checklookbehindgroup(gid, source)
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000700 subpatternappend((GROUPREF, gid))
Fredrik Lundh7cafe4d2000-07-02 17:33:27 +0000701 continue
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300702
Fredrik Lundh90a07912000-06-30 07:50:59 +0000703 else:
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000704 char = sourceget()
Fredrik Lundh90a07912000-06-30 07:50:59 +0000705 if char is None:
Serhiy Storchakaad446d52014-11-10 13:49:00 +0200706 raise source.error("unexpected end of pattern")
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200707 raise source.error("unknown extension ?P" + char,
708 len(char) + 2)
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300709 elif char == ":":
Fredrik Lundh90a07912000-06-30 07:50:59 +0000710 # non-capturing group
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200711 group = None
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300712 elif char == "#":
Fredrik Lundh90a07912000-06-30 07:50:59 +0000713 # comment
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300714 while True:
715 if source.next is None:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200716 raise source.error("missing ), unterminated comment",
717 source.tell() - start)
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300718 if sourceget() == ")":
Fredrik Lundh90a07912000-06-30 07:50:59 +0000719 break
Fredrik Lundh0c4fdba2000-08-31 22:57:55 +0000720 continue
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300721
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300722 elif char in "=!<":
Fredrik Lundh43b3b492000-06-30 10:41:31 +0000723 # lookahead assertions
Fredrik Lundh6f013982000-07-03 18:44:21 +0000724 dir = 1
725 if char == "<":
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300726 char = sourceget()
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200727 if char is None:
728 raise source.error("unexpected end of pattern")
729 if char not in "=!":
730 raise source.error("unknown extension ?<" + char,
731 len(char) + 2)
Fredrik Lundh6f013982000-07-03 18:44:21 +0000732 dir = -1 # lookbehind
Serhiy Storchaka4eea62f2015-02-21 10:07:35 +0200733 lookbehindgroups = state.lookbehindgroups
734 if lookbehindgroups is None:
735 state.lookbehindgroups = state.groups
Serhiy Storchakac7ac7282017-05-16 15:16:15 +0300736 p = _parse_sub(source, state, verbose, nested + 1)
Serhiy Storchaka4eea62f2015-02-21 10:07:35 +0200737 if dir < 0:
738 if lookbehindgroups is None:
739 state.lookbehindgroups = None
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000740 if not sourcematch(")"):
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200741 raise source.error("missing ), unterminated subpattern",
742 source.tell() - start)
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000743 if char == "=":
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000744 subpatternappend((ASSERT, (dir, p)))
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000745 else:
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000746 subpatternappend((ASSERT_NOT, (dir, p)))
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000747 continue
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300748
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300749 elif char == "(":
Gustavo Niemeyerad3fc442003-10-17 22:13:16 +0000750 # conditional backreference group
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300751 condname = source.getuntil(")")
Georg Brandl1d472b72013-04-14 11:40:00 +0200752 if condname.isidentifier():
Gustavo Niemeyerad3fc442003-10-17 22:13:16 +0000753 condgroup = state.groupdict.get(condname)
754 if condgroup is None:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200755 msg = "unknown group name %r" % condname
Serhiy Storchakaad446d52014-11-10 13:49:00 +0200756 raise source.error(msg, len(condname) + 1)
Gustavo Niemeyerad3fc442003-10-17 22:13:16 +0000757 else:
758 try:
Barry Warsaw8bee7612004-08-25 02:22:30 +0000759 condgroup = int(condname)
Serhiy Storchaka9baa5b22014-09-29 22:49:23 +0300760 if condgroup < 0:
761 raise ValueError
Gustavo Niemeyerad3fc442003-10-17 22:13:16 +0000762 except ValueError:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200763 msg = "bad character in group name %r" % condname
764 raise source.error(msg, len(condname) + 1) from None
Serhiy Storchaka9baa5b22014-09-29 22:49:23 +0300765 if not condgroup:
Serhiy Storchakaad446d52014-11-10 13:49:00 +0200766 raise source.error("bad group number",
767 len(condname) + 1)
Serhiy Storchaka9baa5b22014-09-29 22:49:23 +0300768 if condgroup >= MAXGROUPS:
Serhiy Storchaka662cef62016-10-23 12:11:19 +0300769 msg = "invalid group reference %d" % condgroup
770 raise source.error(msg, len(condname) + 1)
Serhiy Storchaka4eea62f2015-02-21 10:07:35 +0200771 state.checklookbehindgroup(condgroup, source)
Serhiy Storchakac7ac7282017-05-16 15:16:15 +0300772 item_yes = _parse(source, state, verbose, nested + 1)
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300773 if source.match("|"):
Serhiy Storchakac7ac7282017-05-16 15:16:15 +0300774 item_no = _parse(source, state, verbose, nested + 1)
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300775 if source.next == "|":
776 raise source.error("conditional backref with more than two branches")
777 else:
778 item_no = None
779 if not source.match(")"):
780 raise source.error("missing ), unterminated subpattern",
781 source.tell() - start)
782 subpatternappend((GROUPREF_EXISTS, (condgroup, item_yes, item_no)))
783 continue
784
Serhiy Storchakabe9a4e52016-09-10 00:57:55 +0300785 elif char in FLAGS or char == "-":
Fredrik Lundh90a07912000-06-30 07:50:59 +0000786 # flags
Serhiy Storchakabe9a4e52016-09-10 00:57:55 +0300787 flags = _parse_flags(source, state, char)
788 if flags is None: # global flags
Serhiy Storchaka305ccbe2017-05-10 06:05:20 +0300789 if not first or subpattern:
Serhiy Storchakabd48d272016-09-11 12:50:02 +0300790 import warnings
Serhiy Storchakaabf275a2016-09-17 01:29:58 +0300791 warnings.warn(
Roy Williams171b9a32017-06-09 22:01:16 -0700792 'Flags not at the start of the expression %r%s' % (
Serhiy Storchakaabf275a2016-09-17 01:29:58 +0300793 source.string[:20], # truncate long regexes
794 ' (truncated)' if len(source.string) > 20 else '',
795 ),
Serhiy Storchakac7ac7282017-05-16 15:16:15 +0300796 DeprecationWarning, stacklevel=nested + 6
Serhiy Storchakaabf275a2016-09-17 01:29:58 +0300797 )
Serhiy Storchaka305ccbe2017-05-10 06:05:20 +0300798 if (state.flags & SRE_FLAG_VERBOSE) and not verbose:
799 raise Verbose
Serhiy Storchakabe9a4e52016-09-10 00:57:55 +0300800 continue
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300801
Serhiy Storchakabe9a4e52016-09-10 00:57:55 +0300802 add_flags, del_flags = flags
803 group = None
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300804 else:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200805 raise source.error("unknown extension ?" + char,
806 len(char) + 1)
807
808 # parse group contents
809 if group is not None:
810 try:
811 group = state.opengroup(name)
812 except error as err:
813 raise source.error(err.msg, len(name) + 1) from None
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300814 sub_verbose = ((verbose or (add_flags & SRE_FLAG_VERBOSE)) and
815 not (del_flags & SRE_FLAG_VERBOSE))
Serhiy Storchakac7ac7282017-05-16 15:16:15 +0300816 p = _parse_sub(source, state, sub_verbose, nested + 1)
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200817 if not source.match(")"):
818 raise source.error("missing ), unterminated subpattern",
819 source.tell() - start)
820 if group is not None:
821 state.closegroup(group, p)
Serhiy Storchakabe9a4e52016-09-10 00:57:55 +0300822 subpatternappend((SUBPATTERN, (group, add_flags, del_flags, p)))
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000823
Fredrik Lundh90a07912000-06-30 07:50:59 +0000824 elif this == "^":
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000825 subpatternappend((AT, AT_BEGINNING))
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000826
Fredrik Lundh90a07912000-06-30 07:50:59 +0000827 elif this == "$":
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300828 subpatternappend((AT, AT_END))
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000829
Fredrik Lundh90a07912000-06-30 07:50:59 +0000830 else:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200831 raise AssertionError("unsupported special character %r" % (char,))
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000832
Serhiy Storchaka821a9d12017-05-14 08:32:33 +0300833 # unpack non-capturing groups
834 for i in range(len(subpattern))[::-1]:
835 op, av = subpattern[i]
836 if op is SUBPATTERN:
837 group, add_flags, del_flags, p = av
838 if group is None and not add_flags and not del_flags:
839 subpattern[i: i+1] = p
840
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000841 return subpattern
842
Serhiy Storchakabe9a4e52016-09-10 00:57:55 +0300843def _parse_flags(source, state, char):
844 sourceget = source.get
845 add_flags = 0
846 del_flags = 0
847 if char != "-":
848 while True:
Serhiy Storchaka3557b052017-10-24 23:31:42 +0300849 flag = FLAGS[char]
850 if source.istext:
851 if char == 'L':
852 msg = "bad inline flags: cannot use 'L' flag with a str pattern"
853 raise source.error(msg)
854 else:
855 if char == 'u':
856 msg = "bad inline flags: cannot use 'u' flag with a bytes pattern"
857 raise source.error(msg)
858 add_flags |= flag
859 if (flag & TYPE_FLAGS) and (add_flags & TYPE_FLAGS) != flag:
860 msg = "bad inline flags: flags 'a', 'u' and 'L' are incompatible"
861 raise source.error(msg)
Serhiy Storchakabe9a4e52016-09-10 00:57:55 +0300862 char = sourceget()
863 if char is None:
864 raise source.error("missing -, : or )")
865 if char in ")-:":
866 break
867 if char not in FLAGS:
868 msg = "unknown flag" if char.isalpha() else "missing -, : or )"
869 raise source.error(msg, len(char))
870 if char == ")":
Serhiy Storchakabe9a4e52016-09-10 00:57:55 +0300871 state.flags |= add_flags
872 return None
873 if add_flags & GLOBAL_FLAGS:
874 raise source.error("bad inline flags: cannot turn on global flag", 1)
875 if char == "-":
876 char = sourceget()
877 if char is None:
878 raise source.error("missing flag")
879 if char not in FLAGS:
880 msg = "unknown flag" if char.isalpha() else "missing flag"
881 raise source.error(msg, len(char))
882 while True:
Serhiy Storchaka3557b052017-10-24 23:31:42 +0300883 flag = FLAGS[char]
884 if flag & TYPE_FLAGS:
885 msg = "bad inline flags: cannot turn off flags 'a', 'u' and 'L'"
886 raise source.error(msg)
887 del_flags |= flag
Serhiy Storchakabe9a4e52016-09-10 00:57:55 +0300888 char = sourceget()
889 if char is None:
890 raise source.error("missing :")
891 if char == ":":
892 break
893 if char not in FLAGS:
894 msg = "unknown flag" if char.isalpha() else "missing :"
895 raise source.error(msg, len(char))
896 assert char == ":"
897 if del_flags & GLOBAL_FLAGS:
898 raise source.error("bad inline flags: cannot turn off global flag", 1)
899 if add_flags & del_flags:
900 raise source.error("bad inline flags: flag turned on and off", 1)
901 return add_flags, del_flags
902
Antoine Pitroufd036452008-08-19 17:56:33 +0000903def fix_flags(src, flags):
904 # Check and fix flags according to the type of pattern (str or bytes)
905 if isinstance(src, str):
Serhiy Storchaka22a309a2014-12-01 11:50:07 +0200906 if flags & SRE_FLAG_LOCALE:
Serhiy Storchaka9bd85b82016-06-11 19:15:00 +0300907 raise ValueError("cannot use LOCALE flag with a str pattern")
Antoine Pitroufd036452008-08-19 17:56:33 +0000908 if not flags & SRE_FLAG_ASCII:
909 flags |= SRE_FLAG_UNICODE
910 elif flags & SRE_FLAG_UNICODE:
911 raise ValueError("ASCII and UNICODE flags are incompatible")
912 else:
913 if flags & SRE_FLAG_UNICODE:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200914 raise ValueError("cannot use UNICODE flag with a bytes pattern")
Serhiy Storchaka22a309a2014-12-01 11:50:07 +0200915 if flags & SRE_FLAG_LOCALE and flags & SRE_FLAG_ASCII:
Serhiy Storchaka9bd85b82016-06-11 19:15:00 +0300916 raise ValueError("ASCII and LOCALE flags are incompatible")
Antoine Pitroufd036452008-08-19 17:56:33 +0000917 return flags
918
Fredrik Lundh7898c3e2000-08-07 20:59:04 +0000919def parse(str, flags=0, pattern=None):
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +0000920 # parse 're' pattern into list of (opcode, argument) tuples
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000921
922 source = Tokenizer(str)
923
Fredrik Lundh7898c3e2000-08-07 20:59:04 +0000924 if pattern is None:
925 pattern = Pattern()
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000926 pattern.flags = flags
Fredrik Lundh470ea5a2001-01-14 21:00:44 +0000927 pattern.str = str
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000928
Serhiy Storchakabe9a4e52016-09-10 00:57:55 +0300929 try:
Serhiy Storchakac7ac7282017-05-16 15:16:15 +0300930 p = _parse_sub(source, pattern, flags & SRE_FLAG_VERBOSE, 0)
Serhiy Storchakabe9a4e52016-09-10 00:57:55 +0300931 except Verbose:
932 # the VERBOSE flag was switched on inside the pattern. to be
933 # on the safe side, we'll parse the whole thing again...
934 pattern = Pattern()
935 pattern.flags = flags | SRE_FLAG_VERBOSE
936 pattern.str = str
Serhiy Storchakad65cd092016-09-11 01:39:01 +0300937 source.seek(0)
Serhiy Storchakac7ac7282017-05-16 15:16:15 +0300938 p = _parse_sub(source, pattern, True, 0)
Serhiy Storchakabe9a4e52016-09-10 00:57:55 +0300939
Antoine Pitroufd036452008-08-19 17:56:33 +0000940 p.pattern.flags = fix_flags(str, p.pattern.flags)
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000941
Serhiy Storchakae2ccf562014-10-10 11:14:49 +0300942 if source.next is not None:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200943 assert source.next == ")"
944 raise source.error("unbalanced parenthesis")
Fredrik Lundh8a3ebf82000-07-23 21:46:17 +0000945
Serhiy Storchakaa01a1442016-03-06 09:15:47 +0200946 if flags & SRE_FLAG_DEBUG:
947 p.dump()
948
Guido van Rossum7627c0d2000-03-31 14:58:54 +0000949 return p
950
Fredrik Lundh436c3d582000-06-29 08:58:44 +0000951def parse_template(source, pattern):
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +0000952 # parse 're' replacement string into list of literals and
953 # group references
954 s = Tokenizer(source)
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000955 sget = s.get
Serhiy Storchaka9c15ec12013-10-23 22:27:52 +0300956 groups = []
957 literals = []
958 literal = []
959 lappend = literal.append
Serhiy Storchaka662cef62016-10-23 12:11:19 +0300960 def addgroup(index, pos):
961 if index > pattern.groups:
962 raise s.error("invalid group reference %d" % index, pos)
Serhiy Storchaka9c15ec12013-10-23 22:27:52 +0300963 if literal:
964 literals.append(''.join(literal))
965 del literal[:]
966 groups.append((len(literals), index))
967 literals.append(None)
Serhiy Storchaka07360df2015-03-30 01:01:48 +0300968 groupindex = pattern.groupindex
Serhiy Storchaka9c15ec12013-10-23 22:27:52 +0300969 while True:
Raymond Hettinger968c56a2004-03-26 23:24:00 +0000970 this = sget()
Fredrik Lundh90a07912000-06-30 07:50:59 +0000971 if this is None:
972 break # end of replacement string
Serhiy Storchaka9c15ec12013-10-23 22:27:52 +0300973 if this[0] == "\\":
Fredrik Lundh90a07912000-06-30 07:50:59 +0000974 # group
Serhiy Storchaka9c15ec12013-10-23 22:27:52 +0300975 c = this[1]
Gustavo Niemeyera01a2ee2004-09-03 17:06:10 +0000976 if c == "g":
Fredrik Lundh90a07912000-06-30 07:50:59 +0000977 name = ""
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200978 if not s.match("<"):
979 raise s.error("missing <")
980 name = s.getuntil(">")
981 if name.isidentifier():
Fredrik Lundh90a07912000-06-30 07:50:59 +0000982 try:
Serhiy Storchaka07360df2015-03-30 01:01:48 +0300983 index = groupindex[name]
Fredrik Lundh90a07912000-06-30 07:50:59 +0000984 except KeyError:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200985 raise IndexError("unknown group name %r" % name)
986 else:
987 try:
988 index = int(name)
989 if index < 0:
990 raise ValueError
991 except ValueError:
992 raise s.error("bad character in group name %r" % name,
993 len(name) + 1) from None
994 if index >= MAXGROUPS:
Serhiy Storchaka662cef62016-10-23 12:11:19 +0300995 raise s.error("invalid group reference %d" % index,
Serhiy Storchaka632a77e2015-03-25 21:03:47 +0200996 len(name) + 1)
Serhiy Storchaka662cef62016-10-23 12:11:19 +0300997 addgroup(index, len(name) + 1)
Gustavo Niemeyera01a2ee2004-09-03 17:06:10 +0000998 elif c == "0":
999 if s.next in OCTDIGITS:
Serhiy Storchaka9c15ec12013-10-23 22:27:52 +03001000 this += sget()
Gustavo Niemeyera01a2ee2004-09-03 17:06:10 +00001001 if s.next in OCTDIGITS:
Serhiy Storchaka9c15ec12013-10-23 22:27:52 +03001002 this += sget()
1003 lappend(chr(int(this[1:], 8) & 0xff))
Gustavo Niemeyera01a2ee2004-09-03 17:06:10 +00001004 elif c in DIGITS:
1005 isoctal = False
1006 if s.next in DIGITS:
Serhiy Storchaka9c15ec12013-10-23 22:27:52 +03001007 this += sget()
Gustavo Niemeyerf5a15992004-09-03 20:15:56 +00001008 if (c in OCTDIGITS and this[2] in OCTDIGITS and
1009 s.next in OCTDIGITS):
Serhiy Storchaka9c15ec12013-10-23 22:27:52 +03001010 this += sget()
Gustavo Niemeyera01a2ee2004-09-03 17:06:10 +00001011 isoctal = True
Serhiy Storchakac563caf2014-09-23 23:22:41 +03001012 c = int(this[1:], 8)
1013 if c > 0o377:
Serhiy Storchaka632a77e2015-03-25 21:03:47 +02001014 raise s.error('octal escape value %s outside of '
Serhiy Storchakaad446d52014-11-10 13:49:00 +02001015 'range 0-0o377' % this, len(this))
Serhiy Storchakac563caf2014-09-23 23:22:41 +03001016 lappend(chr(c))
Gustavo Niemeyera01a2ee2004-09-03 17:06:10 +00001017 if not isoctal:
Serhiy Storchaka662cef62016-10-23 12:11:19 +03001018 addgroup(int(this[1:]), len(this) - 1)
Fredrik Lundh90a07912000-06-30 07:50:59 +00001019 else:
1020 try:
Serhiy Storchaka9c15ec12013-10-23 22:27:52 +03001021 this = chr(ESCAPES[this][1])
Fredrik Lundh90a07912000-06-30 07:50:59 +00001022 except KeyError:
Serhiy Storchakaa54aae02015-03-24 22:58:14 +02001023 if c in ASCIILETTERS:
Serhiy Storchaka9bd85b82016-06-11 19:15:00 +03001024 raise s.error('bad escape %s' % this, len(this))
Serhiy Storchaka9c15ec12013-10-23 22:27:52 +03001025 lappend(this)
Fredrik Lundh90a07912000-06-30 07:50:59 +00001026 else:
Serhiy Storchaka9c15ec12013-10-23 22:27:52 +03001027 lappend(this)
1028 if literal:
1029 literals.append(''.join(literal))
1030 if not isinstance(source, str):
Ezio Melottib92ed7c2010-03-06 15:24:08 +00001031 # The tokenizer implicitly decodes bytes objects as latin-1, we must
1032 # therefore re-encode the final representation.
Serhiy Storchaka9c15ec12013-10-23 22:27:52 +03001033 literals = [None if s is None else s.encode('latin-1') for s in literals]
Fredrik Lundhb25e1ad2001-03-22 15:50:10 +00001034 return groups, literals
Andrew M. Kuchling815d5b92000-06-09 14:08:07 +00001035
Fredrik Lundh436c3d582000-06-29 08:58:44 +00001036def expand_template(template, match):
Fredrik Lundhb25e1ad2001-03-22 15:50:10 +00001037 g = match.group
Serhiy Storchaka7438e4b2014-10-10 11:06:31 +03001038 empty = match.string[:0]
Fredrik Lundhb25e1ad2001-03-22 15:50:10 +00001039 groups, literals = template
1040 literals = literals[:]
1041 try:
1042 for index, group in groups:
Serhiy Storchaka7438e4b2014-10-10 11:06:31 +03001043 literals[index] = g(group) or empty
Fredrik Lundhb25e1ad2001-03-22 15:50:10 +00001044 except IndexError:
Serhiy Storchaka662cef62016-10-23 12:11:19 +03001045 raise error("invalid group reference %d" % index)
Serhiy Storchaka7438e4b2014-10-10 11:06:31 +03001046 return empty.join(literals)