~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/hooks.py

  • Committer: Andrew Bennetts
  • Date: 2011-02-14 12:03:05 UTC
  • mto: This revision was merged to the branch mainline in revision 5664.
  • Revision ID: andrew.bennetts@canonical.com-20110214120305-7l7iu1h6f13voeo7
Add release note.

Show diffs side-by-side

added added

removed removed

Lines of Context:
16
16
 
17
17
 
18
18
"""Support for plugin hooking logic."""
19
 
 
20
19
from bzrlib import (
 
20
    pyutils,
21
21
    registry,
22
22
    symbol_versioning,
23
23
    )
26
26
import textwrap
27
27
 
28
28
from bzrlib import (
29
 
    _format_version_tuple,
30
 
    errors,
31
 
    pyutils,
32
 
    )
 
29
        _format_version_tuple,
 
30
        errors,
 
31
        )
 
32
from bzrlib.help_topics import help_as_plain_text
33
33
""")
34
34
 
35
35
 
115
115
    FOO hook is triggered.
116
116
    """
117
117
 
118
 
    def __init__(self, module=None, member_name=None):
119
 
        """Create a new hooks dictionary.
120
 
 
121
 
        :param module: The module from which this hooks dictionary should be loaded
122
 
            (used for lazy hooks)
123
 
        :param member_name: Name under which this hooks dictionary should be loaded.
124
 
            (used for lazy hooks)
125
 
        """
 
118
    def __init__(self):
126
119
        dict.__init__(self)
127
120
        self._callable_names = {}
128
 
        self._module = module
129
 
        self._member_name = member_name
130
 
 
131
 
    def add_hook(self, name, doc, introduced, deprecated=None):
132
 
        """Add a hook point to this dictionary.
133
 
 
134
 
        :param name: The name of the hook, for clients to use when registering.
135
 
        :param doc: The docs for the hook.
136
 
        :param introduced: When the hook was introduced (e.g. (0, 15)).
137
 
        :param deprecated: When the hook was deprecated, None for
138
 
            not-deprecated.
139
 
        """
140
 
        if name in self:
141
 
            raise errors.DuplicateKey(name)
142
 
        if self._module:
143
 
            callbacks = _lazy_hooks.setdefault(
144
 
                (self._module, self._member_name, name), [])
145
 
        else:
146
 
            callbacks = None
147
 
        hookpoint = HookPoint(name=name, doc=doc, introduced=introduced,
148
 
                              deprecated=deprecated, callbacks=callbacks)
149
 
        self[name] = hookpoint
150
 
 
151
 
    @symbol_versioning.deprecated_method(symbol_versioning.deprecated_in((2, 4)))
 
121
 
152
122
    def create_hook(self, hook):
153
123
        """Create a hook which can have callbacks registered for it.
154
124
 
196
166
        """
197
167
        return self._callable_names.get(a_callable, "No hook name")
198
168
 
199
 
    def install_named_hook_lazy(self, hook_name, callable_module,
200
 
        callable_member, name):
201
 
        """Install a_callable in to the hook hook_name lazily, and label it.
202
 
 
203
 
        :param hook_name: A hook name. See the __init__ method for the complete
204
 
            list of hooks.
205
 
        :param callable_module: Name of the module in which the callable is
206
 
            present.
207
 
        :param callable_member: Member name of the callable.
208
 
        :param name: A name to associate the callable with, to show users what
209
 
            is running.
210
 
        """
211
 
        try:
212
 
            hook = self[hook_name]
213
 
        except KeyError:
214
 
            raise errors.UnknownHook(self.__class__.__name__, hook_name)
215
 
        try:
216
 
            hook_lazy = getattr(hook, "hook_lazy")
217
 
        except AttributeError:
218
 
            raise errors.UnsupportedOperation(self.install_named_hook_lazy,
219
 
                self)
220
 
        else:
221
 
            hook_lazy(callable_module, callable_member, name)
222
 
 
223
169
    def install_named_hook(self, hook_name, a_callable, name):
224
170
        """Install a_callable in to the hook hook_name, and label it name.
225
171
 
226
 
        :param hook_name: A hook name. See the __init__ method for the complete
227
 
            list of hooks.
 
172
        :param hook_name: A hook name. See the __init__ method of BranchHooks
 
173
            for the complete list of hooks.
228
174
        :param a_callable: The callable to be invoked when the hook triggers.
229
175
            The exact signature will depend on the hook - see the __init__
230
 
            method for details on each hook.
 
176
            method of BranchHooks for details on each hook.
231
177
        :param name: A name to associate a_callable with, to show users what is
232
178
            running.
233
179
        """
243
189
        if name is not None:
244
190
            self.name_hook(a_callable, name)
245
191
 
246
 
    def uninstall_named_hook(self, hook_name, label):
247
 
        """Uninstall named hooks.
248
 
 
249
 
        :param hook_name: Hook point name
250
 
        :param label: Label of the callable to uninstall
251
 
        """
252
 
        try:
253
 
            hook = self[hook_name]
254
 
        except KeyError:
255
 
            raise errors.UnknownHook(self.__class__.__name__, hook_name)
256
 
        try:
257
 
            uninstall = getattr(hook, "uninstall")
258
 
        except AttributeError:
259
 
            raise errors.UnsupportedOperation(self.install_named_hook_lazy,
260
 
                self)
261
 
        else:
262
 
            uninstall(label)
263
 
 
264
192
    def name_hook(self, a_callable, name):
265
193
        """Associate name with a_callable to show users what is running."""
266
194
        self._callable_names[a_callable] = name
279
207
        should describe the recommended replacement hook to register for.
280
208
    """
281
209
 
282
 
    def __init__(self, name, doc, introduced, deprecated=None, callbacks=None):
 
210
    def __init__(self, name, doc, introduced, deprecated):
283
211
        """Create a HookPoint.
284
212
 
285
213
        :param name: The name of the hook, for clients to use when registering.
292
220
        self.__doc__ = doc
293
221
        self.introduced = introduced
294
222
        self.deprecated = deprecated
295
 
        if callbacks is None:
296
 
            self._callbacks = []
297
 
        else:
298
 
            self._callbacks = callbacks
 
223
        self._callbacks = []
 
224
        self._callback_names = {}
299
225
 
300
226
    def docs(self):
301
227
        """Generate the documentation for this HookPoint.
321
247
        return '\n'.join(strings)
322
248
 
323
249
    def __eq__(self, other):
324
 
        return (type(other) == type(self) and other.__dict__ == self.__dict__)
325
 
 
326
 
    def hook_lazy(self, callback_module, callback_member, callback_label):
327
 
        """Lazily register a callback to be called when this HookPoint fires.
328
 
 
329
 
        :param callback_module: Module of the callable to use when this
330
 
            HookPoint fires.
331
 
        :param callback_member: Member name of the callback.
332
 
        :param callback_label: A label to show in the UI while this callback is
333
 
            processing.
334
 
        """
335
 
        obj_getter = registry._LazyObjectGetter(callback_module,
336
 
            callback_member)
337
 
        self._callbacks.append((obj_getter, callback_label))
 
250
        return (type(other) == type(self) and 
 
251
            other.__dict__ == self.__dict__)
338
252
 
339
253
    def hook(self, callback, callback_label):
340
254
        """Register a callback to be called when this HookPoint fires.
343
257
        :param callback_label: A label to show in the UI while this callback is
344
258
            processing.
345
259
        """
346
 
        obj_getter = registry._ObjectGetter(callback)
347
 
        self._callbacks.append((obj_getter, callback_label))
348
 
 
349
 
    def uninstall(self, label):
350
 
        """Uninstall the callback with the specified label.
351
 
 
352
 
        :param label: Label of the entry to uninstall
353
 
        """
354
 
        entries_to_remove = []
355
 
        for entry in self._callbacks:
356
 
            (entry_callback, entry_label) = entry
357
 
            if entry_label == label:
358
 
                entries_to_remove.append(entry)
359
 
        if entries_to_remove == []:
360
 
            raise KeyError("No entry with label %r" % label)
361
 
        for entry in entries_to_remove:
362
 
            self._callbacks.remove(entry)
 
260
        self._callbacks.append(callback)
 
261
        if callback_label is not None:
 
262
            self._callback_names[callback] = callback_label
363
263
 
364
264
    def __iter__(self):
365
 
        return (callback.get_obj() for callback, name in self._callbacks)
 
265
        return iter(self._callbacks)
366
266
 
367
267
    def __len__(self):
368
268
        return len(self._callbacks)
372
272
        strings.append("<%s(" % type(self).__name__)
373
273
        strings.append(self.name)
374
274
        strings.append("), callbacks=[")
375
 
        callbacks = self._callbacks
376
 
        for (callback, callback_name) in callbacks:
377
 
            strings.append(repr(callback.get_obj()))
 
275
        for callback in self._callbacks:
 
276
            strings.append(repr(callback))
378
277
            strings.append("(")
379
 
            strings.append(callback_name)
 
278
            strings.append(self._callback_names[callback])
380
279
            strings.append("),")
381
 
        if len(callbacks) == 1:
 
280
        if len(self._callbacks) == 1:
382
281
            strings[-1] = ")"
383
282
        strings.append("]>")
384
283
        return ''.join(strings)
423
322
        hooks = known_hooks_key_to_object(hook_key)
424
323
        segments.append(hooks.docs())
425
324
    return '\n'.join(segments)
426
 
 
427
 
 
428
 
# Lazily registered hooks. Maps (module, name, hook_name) tuples
429
 
# to lists of tuples with objectgetters and names
430
 
_lazy_hooks = {}
431
 
 
432
 
 
433
 
def install_lazy_named_hook(hookpoints_module, hookpoints_name, hook_name,
434
 
    a_callable, name):
435
 
    """Install a callable in to a hook lazily, and label it name.
436
 
 
437
 
    :param hookpoints_module: Module name of the hook points.
438
 
    :param hookpoints_name: Name of the hook points.
439
 
    :param hook_name: A hook name.
440
 
    :param callable: a callable to call for the hook.
441
 
    :param name: A name to associate a_callable with, to show users what is
442
 
        running.
443
 
    """
444
 
    key = (hookpoints_module, hookpoints_name, hook_name)
445
 
    obj_getter = registry._ObjectGetter(a_callable)
446
 
    _lazy_hooks.setdefault(key, []).append((obj_getter, name))