/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: Andrew Bennetts
  • Date: 2009-10-21 11:13:40 UTC
  • mto: This revision was merged to the branch mainline in revision 4762.
  • Revision ID: andrew.bennetts@canonical.com-20091021111340-w7x4d5yf83qwjncc
Add test that WSGI glue allows request handlers to access paths above that request's. backing transport, so long as it is within the WSGI app's backing transport.

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
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
 
from __future__ import absolute_import
28
 
 
29
23
import re
30
24
 
31
 
from . import errors
32
 
 
33
 
 
34
 
class InvalidPattern(errors.BzrError):
35
 
 
36
 
    _fmt = ('Invalid pattern(s) found. %(msg)s')
37
 
 
38
 
    def __init__(self, msg):
39
 
        self.msg = msg
40
 
 
41
25
 
42
26
class LazyRegex(object):
43
27
    """A proxy around a real regex, which won't be compiled until accessed."""
58
42
    def __init__(self, args=(), kwargs={}):
59
43
        """Create a new proxy object, passing in the args to pass to re.compile
60
44
 
61
 
        :param args: The `*args` to pass to re.compile
62
 
        :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
63
47
        """
64
48
        self._real_regex = None
65
49
        self._regex_args = args
74
58
 
75
59
    def _real_re_compile(self, *args, **kwargs):
76
60
        """Thunk over to the original re.compile"""
77
 
        try:
78
 
            return _real_re_compile(*args, **kwargs)
79
 
        except re.error as e:
80
 
            # raise InvalidPattern instead of re.error as this gives a
81
 
            # cleaner message to the user.
82
 
            raise InvalidPattern('"' + args[0] + '" ' +str(e))
83
 
 
84
 
    def __getstate__(self):
85
 
        """Return the state to use when pickling."""
86
 
        return {
87
 
            "args": self._regex_args,
88
 
            "kwargs": self._regex_kwargs,
89
 
            }
90
 
 
91
 
    def __setstate__(self, dict):
92
 
        """Restore from a pickled state."""
93
 
        self._real_regex = None
94
 
        setattr(self, "_regex_args", dict["args"])
95
 
        setattr(self, "_regex_kwargs", dict["kwargs"])
 
61
        return _real_re_compile(*args, **kwargs)
96
62
 
97
63
    def __getattr__(self, attr):
98
64
        """Return a member from the proxied regex object.
139
105
    raise AssertionError(
140
106
        "re.compile has already been overridden as lazy_compile, but this would" \
141
107
        " cause infinite recursion")
142
 
 
143
 
 
144
 
# Some libraries calls re.finditer which fails it if receives a LazyRegex.
145
 
if getattr(re, 'finditer', False):
146
 
    def finditer_public(pattern, string, flags=0):
147
 
        if isinstance(pattern, LazyRegex):
148
 
            return pattern.finditer(string)
149
 
        else:
150
 
            return _real_re_compile(pattern, flags).finditer(string)
151
 
    re.finditer = finditer_public