1
# Copyright (C) 2006, 2008 Canonical Ltd
1
# Copyright (C) 2006-2010 Canonical Ltd
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."""
19
from .pyutils import get_named_object
20
22
class _ObjectGetter(object):
21
23
"""Maintain a reference to an object, and return the object on request.
58
69
return the imported object.
60
71
if not self._imported:
72
self._obj = get_named_object(self._module_name, self._member_name)
62
74
return super(_LazyObjectGetter, self).get_obj()
66
segments = self._member_name.split('.')
69
names = [self._member_name]
70
obj = __import__(self._module_name, globals(), locals(), names)
72
for segment in segments:
73
obj = getattr(obj, segment)
77
76
def __repr__(self):
78
return "<%s.%s object at %x, module=%r attribute=%r>" % (
77
return "<%s.%s object at %x, module=%r attribute=%r imported=%r>" % (
79
78
self.__class__.__module__, self.__class__.__name__, id(self),
80
self._module_name, self._member_name)
79
self._module_name, self._member_name, self._imported)
83
82
class Registry(object):
101
100
self._default_key = None
102
101
# Map from key => (is_lazy, info)
104
104
self._help_dict = {}
105
105
self._info_dict = {}
108
"""Return a set of the format names which are aliases."""
109
return dict(self._aliases.items())
113
for alias, target in self._aliases.items():
114
ret.setdefault(target, []).append(alias)
107
117
def register(self, key, obj, help=None, info=None,
108
118
override_existing=False):
109
119
"""Register a new object to a name.
132
142
override_existing=False):
133
143
"""Register a new object to be loaded on request.
145
:param key: This is the key to use to request the object later.
135
146
:param module_name: The python path to the module. Such as 'os.path'.
136
147
:param member_name: The member of the module to return. If empty or
137
148
None, get() will return the module itself.
138
149
:param help: Help text for this entry. This may be a string or
140
:param info: More information for this entry. Registry
151
:param info: More information for this entry. Registry.get_info()
152
can be used to get this information. Registry treats this as an
153
opaque storage location (it is defined by the caller).
141
154
:param override_existing: If True, replace the existing object
142
155
with the new one. If False, if there is already something
143
156
registered with the same key, raise a KeyError
148
161
self._dict[key] = _LazyObjectGetter(module_name, member_name)
149
162
self._add_help_and_info(key, help=help, info=info)
164
def register_alias(self, key, target, info=None):
165
"""Register an alias.
167
:param key: Alias name
168
:param target: Target key name
170
if key in self._dict and key not in self._aliases:
171
raise KeyError('Key %r already registered and not an alias' % key)
172
self._dict[key] = self._dict[target]
173
self._aliases[key] = target
175
info = self._info_dict[target]
176
self._add_help_and_info(key, help=self._help_dict[target], info=info)
151
178
def _add_help_and_info(self, key, help=None, info=None):
152
179
"""Add the help and information about this key"""
153
180
self._help_dict[key] = help
173
200
return self._dict[self._get_key_or_default(key)].get_obj()
202
def _get_module(self, key):
203
"""Return the module the object will be or was loaded from.
205
:param key: The key to obtain the module for.
206
:return: The name of the module
208
return self._dict[key].get_module()
175
210
def get_prefix(self, fullname):
176
211
"""Return an object whose key is a prefix of the supplied value.
217
252
"""Get a list of registered entries"""
218
return sorted(self._dict.keys())
253
return sorted(self._dict)
220
255
def iteritems(self):
221
for key, getter in self._dict.iteritems():
222
yield key, getter.get_obj()
256
for key in self._dict:
257
yield key, self._dict[key].get_obj()
225
260
# We should not use the iteritems() implementation below (see bug
227
return sorted([(key, getter.get_obj())
228
for key, getter in self._dict.items()])
262
return [(key, self._dict[key].get_obj()) for key in self.keys()]
230
264
def _set_default_key(self, key):
231
if not self._dict.has_key(key):
265
if key not in self._dict:
232
266
raise KeyError('No object registered under key %s.' % key)
234
268
self._default_key = key
237
271
return self._default_key
239
273
default_key = property(_get_default_key, _set_default_key,
240
doc="Current value of the default key."
241
" Can be set to any existing key.")
274
doc="Current value of the default key."
275
" Can be set to any existing key.")
244
278
class FormatRegistry(Registry):
248
282
Registry.__init__(self)
249
283
self._other_registry = other_registry
285
def register(self, key, obj, help=None, info=None,
286
override_existing=False):
287
Registry.register(self, key, obj, help=help, info=info,
288
override_existing=override_existing)
289
if self._other_registry is not None:
290
self._other_registry.register(key, obj, help=help,
291
info=info, override_existing=override_existing)
251
293
def register_lazy(self, key, module_name, member_name,
252
294
help=None, info=None,
253
295
override_existing=False):
254
296
# Overridden to allow capturing registrations to two seperate
255
297
# registries in a single call.
256
298
Registry.register_lazy(self, key, module_name, member_name,
257
help=help, info=info, override_existing=override_existing)
299
help=help, info=info, override_existing=override_existing)
258
300
if self._other_registry is not None:
259
301
self._other_registry.register_lazy(key, module_name, member_name,
260
help=help, info=info, override_existing=override_existing)
302
help=help, info=info, override_existing=override_existing)
304
def remove(self, key):
305
Registry.remove(self, key)
306
if self._other_registry is not None:
307
self._other_registry.remove(key)
262
309
def get(self, format_string):
263
310
r = Registry.get(self, format_string)