/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: Martin Pool
  • Date: 2005-06-28 03:02:31 UTC
  • Revision ID: mbp@sourcefrog.net-20050628030231-d311e4ebcd467ef4
Merge John's import-speedup branch:

                                                                                         
  777 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 22:20:32 -0500
      revision-id: john@arbash-meinel.com-20050627032031-e82a50db3863b18e
      bzr selftest was not using the correct bzr

  776 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 22:20:22 -0500
      revision-id: john@arbash-meinel.com-20050627032021-c9f21fde989ddaee
      Add was using an old mutter

  775 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 22:02:33 -0500
      revision-id: john@arbash-meinel.com-20050627030233-9165cfe98fc63298
      Cleaned up to be less different

  774 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 21:54:53 -0500
      revision-id: john@arbash-meinel.com-20050627025452-4260d0e744edef43
      Allow BZR_PLUGIN_PATH='' to negate plugin loading.

  773 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 21:49:34 -0500
      revision-id: john@arbash-meinel.com-20050627024933-b7158f67b7b9eae5
      Finished the previous cleanup (allowing load_plugins to be called twice)

  772 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 21:45:08 -0500
      revision-id: john@arbash-meinel.com-20050627024508-723b1df510d196fc
      Work on making the tests pass. versioning.py is calling run_cmd directly, but plugins have been loaded.

  771 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 21:32:29 -0500
      revision-id: john@arbash-meinel.com-20050627023228-79972744d7c53e15
      Got it down a little bit more by removing import of tree and inventory.

  770 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 21:26:05 -0500
      revision-id: john@arbash-meinel.com-20050627022604-350b9773ef622f95
      Reducing the number of import from bzrlib/__init__.py and bzrlib/branch.py

  769 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 20:32:25 -0500
      revision-id: john@arbash-meinel.com-20050627013225-32dd044f10d23948
      Updated revision.py and xml.py to include SubElement.

  768 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 20:03:56 -0500
      revision-id: john@arbash-meinel.com-20050627010356-ee66919e1c377faf
      Minor typo

  767 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 20:03:13 -0500
      revision-id: john@arbash-meinel.com-20050627010312-40d024007eb85051
      Caching the import

  766 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 19:51:47 -0500
      revision-id: john@arbash-meinel.com-20050627005147-5281c99e48ed1834
      Created wrapper functions for lazy import of ElementTree

  765 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 19:46:37 -0500
      revision-id: john@arbash-meinel.com-20050627004636-bf432902004a94c5
      Removed all of the test imports of cElementTree

  764 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 19:43:59 -0500
      revision-id: john@arbash-meinel.com-20050627004358-d137fbe9570dd71b
      Trying to make bzr startup faster.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/lazy_regex.py
1
 
# Copyright (C) 2006 Canonical Ltd
2
 
#
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.
7
 
#
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.
12
 
#
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
16
 
 
17
 
"""Lazily compiled regex objects.
18
 
 
19
 
This module defines a class which creates proxy objects for regex compilation.
20
 
This allows overriding re.compile() to return lazily compiled objects.
21
 
"""
22
 
 
23
 
import re
24
 
 
25
 
 
26
 
class LazyRegex(object):
27
 
    """A proxy around a real regex, which won't be compiled until accessed."""
28
 
 
29
 
 
30
 
    # These are the parameters on a real _sre.SRE_Pattern object, which we
31
 
    # will map to local members so that we don't have the proxy overhead.
32
 
    _regex_attributes_to_copy = [
33
 
                 '__copy__', '__deepcopy__', 'findall', 'finditer', 'match',
34
 
                 'scanner', 'search', 'split', 'sub', 'subn'
35
 
                 ]
36
 
 
37
 
    # We use slots to keep the overhead low. But we need a slot entry for
38
 
    # all of the attributes we will copy
39
 
    __slots__ = ['_real_regex', '_regex_args', '_regex_kwargs',
40
 
                ] + _regex_attributes_to_copy
41
 
 
42
 
    def __init__(self, args=(), kwargs={}):
43
 
        """Create a new proxy object, passing in the args to pass to re.compile
44
 
 
45
 
        :param args: The *args to pass to re.compile
46
 
        :param kwargs: The **kwargs to pass to re.compile
47
 
        """
48
 
        self._real_regex = None
49
 
        self._regex_args = args
50
 
        self._regex_kwargs = kwargs
51
 
 
52
 
    def _compile_and_collapse(self):
53
 
        """Actually compile the requested regex"""
54
 
        self._real_regex = self._real_re_compile(*self._regex_args,
55
 
                                                 **self._regex_kwargs)
56
 
        for attr in self._regex_attributes_to_copy:
57
 
            setattr(self, attr, getattr(self._real_regex, attr))
58
 
 
59
 
    def _real_re_compile(self, *args, **kwargs):
60
 
        """Thunk over to the original re.compile"""
61
 
        return _real_re_compile(*args, **kwargs)
62
 
 
63
 
    def __getattr__(self, attr):
64
 
        """Return a member from the proxied regex object.
65
 
 
66
 
        If the regex hasn't been compiled yet, compile it
67
 
        """
68
 
        if self._real_regex is None:
69
 
            self._compile_and_collapse()
70
 
        # Once we have compiled, the only time we should come here
71
 
        # is actually if the attribute is missing.
72
 
        return getattr(self._real_regex, attr)
73
 
 
74
 
 
75
 
def lazy_compile(*args, **kwargs):
76
 
    """Create a proxy object which will compile the regex on demand.
77
 
 
78
 
    :return: a LazyRegex proxy object.
79
 
    """
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
 
if _real_re_compile is lazy_compile:
105
 
    raise AssertionError(
106
 
        "re.compile has already been overridden as lazy_compile, but this would" \
107
 
        " cause infinite recursion")