1
# Copyright (C) 2006-2011 Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
"""Tools for converting globs to regular expressions.
19
This module provides functions for converting shell-like globs to regular
29
from bzrlib.trace import (
35
class Replacer(object):
36
"""Do a multiple-pattern substitution.
38
The patterns and substitutions are combined into one, so the result of
39
one replacement is never substituted again. Add the patterns and
40
replacements via the add method and then call the object. The patterns
41
must not contain capturing groups.
44
_expand = lazy_regex.lazy_compile(ur'\\&')
46
def __init__(self, source=None):
49
self._pats = list(source._pats)
50
self._funs = list(source._funs)
55
def add(self, pat, fun):
56
r"""Add a pattern and replacement.
58
The pattern must not contain capturing groups.
59
The replacement might be either a string template in which \& will be
60
replaced with the match, or a function that will get the matching text
61
as argument. It does not get match object, because capturing is
65
self._pats.append(pat)
66
self._funs.append(fun)
68
def add_replacer(self, replacer):
69
r"""Add all patterns from another replacer.
71
All patterns and replacements from replacer are appended to the ones
75
self._pats.extend(replacer._pats)
76
self._funs.extend(replacer._funs)
78
def __call__(self, text):
80
self._pat = lazy_regex.lazy_compile(
81
u'|'.join([u'(%s)' % p for p in self._pats]),
83
return self._pat.sub(self._do_sub, text)
86
fun = self._funs[m.lastindex - 1]
87
if hasattr(fun, '__call__'):
88
return fun(m.group(0))
90
return self._expand.sub(m.group(0), fun)
93
_sub_named = Replacer()
94
_sub_named.add(ur'\[:digit:\]', ur'\d')
95
_sub_named.add(ur'\[:space:\]', ur'\s')
96
_sub_named.add(ur'\[:alnum:\]', ur'\w')
97
_sub_named.add(ur'\[:ascii:\]', ur'\0-\x7f')
98
_sub_named.add(ur'\[:blank:\]', ur' \t')
99
_sub_named.add(ur'\[:cntrl:\]', ur'\0-\x1f\x7f-\x9f')
103
if m[1] in (u'!', u'^'):
104
return u'[^' + _sub_named(m[2:-1]) + u']'
105
return u'[' + _sub_named(m[1:-1]) + u']'
108
def _invalid_regex(repl):
110
warning(u"'%s' not allowed within a regular expression. "
111
"Replacing with '%s'" % (m, repl))
116
def _trailing_backslashes_regex(m):
117
"""Check trailing backslashes.
119
Does a head count on trailing backslashes to ensure there isn't an odd
120
one on the end that would escape the brackets we wrap the RE in.
122
if (len(m) % 2) != 0:
123
warning(u"Regular expressions cannot end with an odd number of '\\'. "
124
"Dropping the final '\\'.")
130
_sub_re.add(u'^RE:', u'')
131
_sub_re.add(u'\((?!\?)', u'(?:')
132
_sub_re.add(u'\(\?P<.*>', _invalid_regex(u'(?:'))
133
_sub_re.add(u'\(\?P=[^)]*\)', _invalid_regex(u''))
134
_sub_re.add(ur'\\+$', _trailing_backslashes_regex)
137
_sub_fullpath = Replacer()
138
_sub_fullpath.add(ur'^RE:.*', _sub_re) # RE:<anything> is a regex
139
_sub_fullpath.add(ur'\[\^?\]?(?:[^][]|\[:[^]]+:\])+\]', _sub_group) # char group
140
_sub_fullpath.add(ur'(?:(?<=/)|^)(?:\.?/)+', u'') # canonicalize path
141
_sub_fullpath.add(ur'\\.', ur'\&') # keep anything backslashed
142
_sub_fullpath.add(ur'[(){}|^$+.]', ur'\\&') # escape specials
143
_sub_fullpath.add(ur'(?:(?<=/)|^)\*\*+/', ur'(?:.*/)?') # **/ after ^ or /
144
_sub_fullpath.add(ur'\*+', ur'[^/]*') # * elsewhere
145
_sub_fullpath.add(ur'\?', ur'[^/]') # ? everywhere
148
_sub_basename = Replacer()
149
_sub_basename.add(ur'\[\^?\]?(?:[^][]|\[:[^]]+:\])+\]', _sub_group) # char group
150
_sub_basename.add(ur'\\.', ur'\&') # keep anything backslashed
151
_sub_basename.add(ur'[(){}|^$+.]', ur'\\&') # escape specials
152
_sub_basename.add(ur'\*+', ur'.*') # * everywhere
153
_sub_basename.add(ur'\?', ur'.') # ? everywhere
156
def _sub_extension(pattern):
157
return _sub_basename(pattern[2:])
160
class Globster(object):
161
"""A simple wrapper for a set of glob patterns.
163
Provides the capability to search the patterns to find a match for
164
a given filename (including the full path).
166
Patterns are translated to regular expressions to expidite matching.
168
The regular expressions for multiple patterns are aggregated into
169
a super-regex containing groups of up to 99 patterns.
170
The 99 limitation is due to the grouping limit of the Python re module.
171
The resulting super-regex and associated patterns are stored as a list of
172
(regex,[patterns]) in _regex_patterns.
174
For performance reasons the patterns are categorised as extension patterns
175
(those that match against a file extension), basename patterns
176
(those that match against the basename of the filename),
177
and fullpath patterns (those that match against the full path).
178
The translations used for extensions and basenames are relatively simpler
179
and therefore faster to perform than the fullpath patterns.
181
Also, the extension patterns are more likely to find a match and
182
so are matched first, then the basename patterns, then the fullpath
185
# We want to _add_patterns in a specific order (as per type_list below)
186
# starting with the shortest and going to the longest.
187
# As some Python version don't support ordered dicts the list below is
188
# used to select inputs for _add_pattern in a specific order.
189
pattern_types = [ "extension", "basename", "fullpath" ]
193
"translator" : _sub_extension,
194
"prefix" : r'(?:.*/)?(?!.*/)(?:.*\.)'
197
"translator" : _sub_basename,
198
"prefix" : r'(?:.*/)?(?!.*/)'
201
"translator" : _sub_fullpath,
206
def __init__(self, patterns):
207
self._regex_patterns = []
214
pat = normalize_pattern(pat)
215
pattern_lists[Globster.identify(pat)].append(pat)
216
pi = Globster.pattern_info
217
for t in Globster.pattern_types:
218
self._add_patterns(pattern_lists[t], pi[t]["translator"],
221
def _add_patterns(self, patterns, translator, prefix=''):
224
'(%s)' % translator(pat) for pat in patterns[:99]]
225
joined_rule = '%s(?:%s)$' % (prefix, '|'.join(grouped_rules))
226
# Explicitly use lazy_compile here, because we count on its
227
# nicer error reporting.
228
self._regex_patterns.append((
229
lazy_regex.lazy_compile(joined_rule, re.UNICODE),
231
patterns = patterns[99:]
233
def match(self, filename):
234
"""Searches for a pattern that matches the given filename.
236
:return A matching pattern or None if there is no matching pattern.
239
for regex, patterns in self._regex_patterns:
240
match = regex.match(filename)
242
return patterns[match.lastindex -1]
243
except errors.InvalidPattern, e:
244
# We can't show the default e.msg to the user as thats for
245
# the combined pattern we sent to regex. Instead we indicate to
246
# the user that an ignore file needs fixing.
247
mutter('Invalid pattern found in regex: %s.', e.msg)
248
e.msg = "File ~/.bazaar/ignore or .bzrignore contains error(s)."
250
for _, patterns in self._regex_patterns:
252
if not Globster.is_pattern_valid(p):
253
bad_patterns += ('\n %s' % p)
254
e.msg += bad_patterns
259
def identify(pattern):
260
"""Returns pattern category.
262
:param pattern: normalized pattern.
263
Identify if a pattern is fullpath, basename or extension
264
and returns the appropriate type.
266
if pattern.startswith(u'RE:') or u'/' in pattern:
268
elif pattern.startswith(u'*.'):
274
def is_pattern_valid(pattern):
275
"""Returns True if pattern is valid.
277
:param pattern: Normalized pattern.
278
is_pattern_valid() assumes pattern to be normalized.
279
see: globbing.normalize_pattern
282
translator = Globster.pattern_info[Globster.identify(pattern)]["translator"]
283
tpattern = '(%s)' % translator(pattern)
285
re_obj = lazy_regex.lazy_compile(tpattern, re.UNICODE)
286
re_obj.search("") # force compile
287
except errors.InvalidPattern, e:
292
class ExceptionGlobster(object):
293
"""A Globster that supports exception patterns.
295
Exceptions are ignore patterns prefixed with '!'. Exception
296
patterns take precedence over regular patterns and cause a
297
matching filename to return None from the match() function.
298
Patterns using a '!!' prefix are highest precedence, and act
299
as regular ignores. '!!' patterns are useful to establish ignores
300
that apply under paths specified by '!' exception patterns.
303
def __init__(self,patterns):
304
ignores = [[], [], []]
306
if p.startswith(u'!!'):
307
ignores[2].append(p[2:])
308
elif p.startswith(u'!'):
309
ignores[1].append(p[1:])
312
self._ignores = [Globster(i) for i in ignores]
314
def match(self, filename):
315
"""Searches for a pattern that matches the given filename.
317
:return A matching pattern or None if there is no matching pattern.
319
double_neg = self._ignores[2].match(filename)
321
return "!!%s" % double_neg
322
elif self._ignores[1].match(filename):
325
return self._ignores[0].match(filename)
327
class _OrderedGlobster(Globster):
328
"""A Globster that keeps pattern order."""
330
def __init__(self, patterns):
333
:param patterns: sequence of glob patterns
335
# Note: This could be smarter by running like sequences together
336
self._regex_patterns = []
338
pat = normalize_pattern(pat)
339
t = Globster.identify(pat)
340
self._add_patterns([pat], Globster.pattern_info[t]["translator"],
341
Globster.pattern_info[t]["prefix"])
344
_slashes = lazy_regex.lazy_compile(r'[\\/]+')
345
def normalize_pattern(pattern):
346
"""Converts backslashes in path patterns to forward slashes.
348
Doesn't normalize regular expressions - they may contain escapes.
350
if not (pattern.startswith('RE:') or pattern.startswith('!RE:')):
351
pattern = _slashes.sub('/', pattern)
353
pattern = pattern.rstrip('/')