/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/lazy_regex.py

  • Committer: Robert Collins
  • Date: 2007-04-19 02:27:44 UTC
  • mto: This revision was merged to the branch mainline in revision 2426.
  • Revision ID: robertc@robertcollins.net-20070419022744-pfdqz42kp1wizh43
``make docs`` now creates a man page at ``man1/bzr.1`` fixing bug 107388.
(Robert Collins)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/lazy_regex.py
1
 
# Copyright (C) 2006, 2008-2011, 2017 Canonical Ltd
 
1
# Copyright (C) 2006 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
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
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
"""Lazily compiled regex objects.
18
18
 
19
 
This module defines a class which creates proxy objects for regex
20
 
compilation.  This allows overriding re.compile() to return lazily compiled
21
 
objects.
22
 
 
23
 
We do this rather than just providing a new interface so that it will also
24
 
be used by existing Python modules that create regexs.
 
19
This module defines a class which creates proxy objects for regex compilation.
 
20
This allows overriding re.compile() to return lazily compiled objects.
25
21
"""
26
22
 
27
23
import re
28
24
 
29
 
from . import errors
30
 
 
31
 
 
32
 
class InvalidPattern(errors.BzrError):
33
 
 
34
 
    _fmt = ('Invalid pattern(s) found. %(msg)s')
35
 
 
36
 
    def __init__(self, msg):
37
 
        self.msg = msg
38
 
 
39
25
 
40
26
class LazyRegex(object):
41
27
    """A proxy around a real regex, which won't be compiled until accessed."""
42
28
 
 
29
 
43
30
    # These are the parameters on a real _sre.SRE_Pattern object, which we
44
31
    # will map to local members so that we don't have the proxy overhead.
45
32
    _regex_attributes_to_copy = [
46
 
        '__copy__', '__deepcopy__', 'findall', 'finditer', 'match',
47
 
        'scanner', 'search', 'split', 'sub', 'subn'
48
 
        ]
 
33
                 '__copy__', '__deepcopy__', 'findall', 'finditer', 'match',
 
34
                 'scanner', 'search', 'split', 'sub', 'subn'
 
35
                 ]
49
36
 
50
37
    # We use slots to keep the overhead low. But we need a slot entry for
51
38
    # all of the attributes we will copy
52
39
    __slots__ = ['_real_regex', '_regex_args', '_regex_kwargs',
53
 
                 ] + _regex_attributes_to_copy
 
40
                ] + _regex_attributes_to_copy
54
41
 
55
 
    def __init__(self, args, kwargs):
 
42
    def __init__(self, args=(), kwargs={}):
56
43
        """Create a new proxy object, passing in the args to pass to re.compile
57
44
 
58
 
        :param args: The `*args` to pass to re.compile
59
 
        :param kwargs: The `**kwargs` to pass to re.compile
 
45
        :param args: The *args to pass to re.compile
 
46
        :param kwargs: The **kwargs to pass to re.compile
60
47
        """
61
48
        self._real_regex = None
62
49
        self._regex_args = args
71
58
 
72
59
    def _real_re_compile(self, *args, **kwargs):
73
60
        """Thunk over to the original re.compile"""
74
 
        try:
75
 
            return re.compile(*args, **kwargs)
76
 
        except re.error as e:
77
 
            # raise InvalidPattern instead of re.error as this gives a
78
 
            # cleaner message to the user.
79
 
            raise InvalidPattern('"' + args[0] + '" ' + str(e))
80
 
 
81
 
    def __getstate__(self):
82
 
        """Return the state to use when pickling."""
83
 
        return {
84
 
            "args": self._regex_args,
85
 
            "kwargs": self._regex_kwargs,
86
 
            }
87
 
 
88
 
    def __setstate__(self, dict):
89
 
        """Restore from a pickled state."""
90
 
        self._real_regex = None
91
 
        setattr(self, "_regex_args", dict["args"])
92
 
        setattr(self, "_regex_kwargs", dict["kwargs"])
 
61
        return _real_re_compile(*args, **kwargs)
93
62
 
94
63
    def __getattr__(self, attr):
95
64
        """Return a member from the proxied regex object.
109
78
    :return: a LazyRegex proxy object.
110
79
    """
111
80
    return LazyRegex(args, kwargs)
 
81
 
 
82
 
 
83
def install_lazy_compile():
 
84
    """Make lazy_compile the default compile mode for regex compilation.
 
85
 
 
86
    This overrides re.compile with lazy_compile. To restore the original
 
87
    functionality, call reset_compile().
 
88
    """
 
89
    re.compile = lazy_compile
 
90
 
 
91
 
 
92
def reset_compile():
 
93
    """Restore the original function to re.compile().
 
94
    
 
95
    It is safe to call reset_compile() multiple times, it will always
 
96
    restore re.compile() to the value that existed at import time.
 
97
    Though the first call will reset back to the original (it doesn't
 
98
    track nesting level)
 
99
    """
 
100
    re.compile = _real_re_compile
 
101
 
 
102
 
 
103
_real_re_compile = re.compile
 
104
assert _real_re_compile is not lazy_compile, \
 
105
    "re.compile has already been overridden as lazy_compile, but this would" \
 
106
    " cause infinite recursion"