← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/registry.py

  • Committer: Andrew Bennetts
  • Date: 2010-10-08 04:25:10 UTC
  • mto: This revision was merged to the branch mainline in revision 5472.
  • Revision ID: andrew.bennetts@canonical.com-20101008042510-sg9vdhmnggilzxsk
Fix stray TAB in source.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/registry.py
1
 
# Copyright (C) 2006-2010 Canonical Ltd
 
1
# Copyright (C) 2006, 2008 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
17
17
"""Classes to provide name-to-object registry-like support."""
18
18
 
19
19
 
20
 
from bzrlib.pyutils import get_named_object
21
 
 
22
 
 
23
20
class _ObjectGetter(object):
24
21
    """Maintain a reference to an object, and return the object on request.
25
22
 
35
32
    def __init__(self, obj):
36
33
        self._obj = obj
37
34
 
38
 
    def get_module(self):
39
 
        """Get the module the object was loaded from."""
40
 
        return self._obj.__module__
41
 
 
42
35
    def get_obj(self):
43
36
        """Get the object that was saved at creation time"""
44
37
        return self._obj
58
51
        self._imported = False
59
52
        super(_LazyObjectGetter, self).__init__(None)
60
53
 
61
 
    def get_module(self):
62
 
        """Get the module the referenced object will be loaded from.
63
 
        """
64
 
        return self._module_name
65
 
 
66
54
    def get_obj(self):
67
55
        """Get the referenced object.
68
56
 
70
58
        return the imported object.
71
59
        """
72
60
        if not self._imported:
73
 
            self._obj = get_named_object(self._module_name, self._member_name)
74
 
            self._imported = True
 
61
            self._do_import()
75
62
        return super(_LazyObjectGetter, self).get_obj()
76
63
 
 
64
    def _do_import(self):
 
65
        if self._member_name:
 
66
            segments = self._member_name.split('.')
 
67
            names = segments[0:1]
 
68
        else:
 
69
            names = [self._member_name]
 
70
        obj = __import__(self._module_name, globals(), locals(), names)
 
71
        if self._member_name:
 
72
            for segment in segments:
 
73
                obj = getattr(obj, segment)
 
74
        self._obj = obj
 
75
        self._imported = True
 
76
 
77
77
    def __repr__(self):
78
 
        return "<%s.%s object at %x, module=%r attribute=%r imported=%r>" % (
 
78
        return "<%s.%s object at %x, module=%r attribute=%r>" % (
79
79
            self.__class__.__module__, self.__class__.__name__, id(self),
80
 
            self._module_name, self._member_name, self._imported)
 
80
            self._module_name, self._member_name)
81
81
 
82
82
 
83
83
class Registry(object):
172
172
        """
173
173
        return self._dict[self._get_key_or_default(key)].get_obj()
174
174
 
175
 
    def _get_module(self, key):
176
 
        """Return the module the object will be or was loaded from.
177
 
 
178
 
        :param key: The key to obtain the module for.
179
 
        :return: The name of the module
180
 
        """
181
 
        return self._dict[key].get_module()
182
 
 
183
175
    def get_prefix(self, fullname):
184
176
        """Return an object whose key is a prefix of the supplied value.
185
177
 
256
248
        Registry.__init__(self)
257
249
        self._other_registry = other_registry
258
250
 
259
 
    def register(self, key, obj, help=None, info=None,
260
 
                 override_existing=False):
261
 
        Registry.register(self, key, obj, help=help, info=info,
262
 
            override_existing=override_existing)
263
 
        if self._other_registry is not None:
264
 
            self._other_registry.register(key, obj, help=help,
265
 
                info=info, override_existing=override_existing)
266
 
 
267
251
    def register_lazy(self, key, module_name, member_name,
268
252
                      help=None, info=None,
269
253
                      override_existing=False):
275
259
            self._other_registry.register_lazy(key, module_name, member_name,
276
260
                help=help, info=info, override_existing=override_existing)
277
261
 
278
 
    def remove(self, key):
279
 
        Registry.remove(self, key)
280
 
        if self._other_registry is not None:
281
 
            self._other_registry.remove(key)
282
 
 
283
262
    def get(self, format_string):
284
263
        r = Registry.get(self, format_string)
285
264
        if callable(r):
286
265
            r = r()
287
266
        return r
 
267
 
 
268