~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/symbol_versioning.py

  • Committer: mbp at sourcefrog
  • Date: 2005-03-13 05:47:58 UTC
  • Revision ID: mbp@sourcefrog.net-20050313054757-a86c3f5871069e22
fix up doctest for code rearrangement

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006-2010 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
 
"""Symbol versioning
18
 
 
19
 
The methods here allow for api symbol versioning.
20
 
"""
21
 
 
22
 
__all__ = ['deprecated_function',
23
 
           'deprecated_in',
24
 
           'deprecated_list',
25
 
           'deprecated_method',
26
 
           'DEPRECATED_PARAMETER',
27
 
           'deprecated_passed',
28
 
           'set_warning_method',
29
 
           'warn',
30
 
           ]
31
 
 
32
 
 
33
 
import warnings
34
 
# Import the 'warn' symbol so bzrlib can call it even if we redefine it
35
 
from warnings import warn
36
 
 
37
 
import bzrlib
38
 
 
39
 
 
40
 
DEPRECATED_PARAMETER = "A deprecated parameter marker."
41
 
 
42
 
 
43
 
def deprecated_in(version_tuple):
44
 
    """Generate a message that something was deprecated in a release.
45
 
 
46
 
    >>> deprecated_in((1, 4, 0))
47
 
    '%s was deprecated in version 1.4.0.'
48
 
    """
49
 
    return ("%%s was deprecated in version %s."
50
 
            % bzrlib._format_version_tuple(version_tuple))
51
 
 
52
 
 
53
 
def set_warning_method(method):
54
 
    """Set the warning method to be used by this module.
55
 
 
56
 
    It should take a message and a warning category as warnings.warn does.
57
 
    """
58
 
    global warn
59
 
    warn = method
60
 
 
61
 
 
62
 
# TODO - maybe this would be easier to use as one 'smart' method that
63
 
# guess if it is a method or a class or an attribute ? If so, we can
64
 
# add that on top of the primitives, once we have all three written
65
 
# - RBC 20050105
66
 
 
67
 
 
68
 
def deprecation_string(a_callable, deprecation_version):
69
 
    """Generate an automatic deprecation string for a_callable.
70
 
 
71
 
    :param a_callable: The callable to substitute into deprecation_version.
72
 
    :param deprecation_version: A deprecation format warning string. This should
73
 
        have a single %s operator in it. a_callable will be turned into a nice
74
 
        python symbol and then substituted into deprecation_version.
75
 
    """
76
 
    # We also want to handle old-style classes, in particular exception, and
77
 
    # they don't have an im_class attribute.
78
 
    if getattr(a_callable, 'im_class', None) is None:
79
 
        symbol = "%s.%s" % (a_callable.__module__,
80
 
                            a_callable.__name__)
81
 
    else:
82
 
        symbol = "%s.%s.%s" % (a_callable.im_class.__module__,
83
 
                               a_callable.im_class.__name__,
84
 
                               a_callable.__name__
85
 
                               )
86
 
    return deprecation_version % symbol
87
 
 
88
 
 
89
 
def deprecated_function(deprecation_version):
90
 
    """Decorate a function so that use of it will trigger a warning."""
91
 
 
92
 
    def function_decorator(callable):
93
 
        """This is the function python calls to perform the decoration."""
94
 
 
95
 
        def decorated_function(*args, **kwargs):
96
 
            """This is the decorated function."""
97
 
            from bzrlib import trace
98
 
            trace.mutter_callsite(4, "Deprecated function called")
99
 
            warn(deprecation_string(callable, deprecation_version),
100
 
                DeprecationWarning, stacklevel=2)
101
 
            return callable(*args, **kwargs)
102
 
        _populate_decorated(callable, deprecation_version, "function",
103
 
                            decorated_function)
104
 
        return decorated_function
105
 
    return function_decorator
106
 
 
107
 
 
108
 
def deprecated_method(deprecation_version):
109
 
    """Decorate a method so that use of it will trigger a warning.
110
 
 
111
 
    To deprecate a static or class method, use
112
 
 
113
 
        @staticmethod
114
 
        @deprecated_function
115
 
        def ...
116
 
 
117
 
    To deprecate an entire class, decorate __init__.
118
 
    """
119
 
 
120
 
    def method_decorator(callable):
121
 
        """This is the function python calls to perform the decoration."""
122
 
 
123
 
        def decorated_method(self, *args, **kwargs):
124
 
            """This is the decorated method."""
125
 
            from bzrlib import trace
126
 
            if callable.__name__ == '__init__':
127
 
                symbol = "%s.%s" % (self.__class__.__module__,
128
 
                                    self.__class__.__name__,
129
 
                                    )
130
 
            else:
131
 
                symbol = "%s.%s.%s" % (self.__class__.__module__,
132
 
                                       self.__class__.__name__,
133
 
                                       callable.__name__
134
 
                                       )
135
 
            trace.mutter_callsite(4, "Deprecated method called")
136
 
            warn(deprecation_version % symbol, DeprecationWarning, stacklevel=2)
137
 
            return callable(self, *args, **kwargs)
138
 
        _populate_decorated(callable, deprecation_version, "method",
139
 
                            decorated_method)
140
 
        return decorated_method
141
 
    return method_decorator
142
 
 
143
 
 
144
 
def deprecated_passed(parameter_value):
145
 
    """Return True if parameter_value was used."""
146
 
    # FIXME: it might be nice to have a parameter deprecation decorator.
147
 
    # it would need to handle positional and *args and **kwargs parameters,
148
 
    # which means some mechanism to describe how the parameter was being
149
 
    # passed before deprecation, and some way to deprecate parameters that
150
 
    # were not at the end of the arg list. Thats needed for __init__ where
151
 
    # we cannot just forward to a new method name.I.e. in the following
152
 
    # examples we would want to have callers that pass any value to 'bad' be
153
 
    # given a warning - because we have applied:
154
 
    # @deprecated_parameter('bad', deprecated_in((1, 5, 0))
155
 
    #
156
 
    # def __init__(self, bad=None)
157
 
    # def __init__(self, bad, other)
158
 
    # def __init__(self, **kwargs)
159
 
    # RBC 20060116
160
 
    return not parameter_value is DEPRECATED_PARAMETER
161
 
 
162
 
 
163
 
def _decorate_docstring(callable, deprecation_version, label,
164
 
                        decorated_callable):
165
 
    if callable.__doc__:
166
 
        docstring_lines = callable.__doc__.split('\n')
167
 
    else:
168
 
        docstring_lines = []
169
 
    if len(docstring_lines) == 0:
170
 
        decorated_callable.__doc__ = deprecation_version % ("This " + label)
171
 
    elif len(docstring_lines) == 1:
172
 
        decorated_callable.__doc__ = (callable.__doc__
173
 
                                    + "\n"
174
 
                                    + "\n"
175
 
                                    + deprecation_version % ("This " + label)
176
 
                                    + "\n")
177
 
    else:
178
 
        spaces = len(docstring_lines[-1])
179
 
        new_doc = callable.__doc__
180
 
        new_doc += "\n" + " " * spaces
181
 
        new_doc += deprecation_version % ("This " + label)
182
 
        new_doc += "\n" + " " * spaces
183
 
        decorated_callable.__doc__ = new_doc
184
 
 
185
 
 
186
 
def _populate_decorated(callable, deprecation_version, label,
187
 
                        decorated_callable):
188
 
    """Populate attributes like __name__ and __doc__ on the decorated callable.
189
 
    """
190
 
    _decorate_docstring(callable, deprecation_version, label,
191
 
                        decorated_callable)
192
 
    decorated_callable.__module__ = callable.__module__
193
 
    decorated_callable.__name__ = callable.__name__
194
 
    decorated_callable.is_deprecated = True
195
 
 
196
 
 
197
 
def _dict_deprecation_wrapper(wrapped_method):
198
 
    """Returns a closure that emits a warning and calls the superclass"""
199
 
    def cb(dep_dict, *args, **kwargs):
200
 
        msg = 'access to %s' % (dep_dict._variable_name, )
201
 
        msg = dep_dict._deprecation_version % (msg,)
202
 
        if dep_dict._advice:
203
 
            msg += ' ' + dep_dict._advice
204
 
        warn(msg, DeprecationWarning, stacklevel=2)
205
 
        return wrapped_method(dep_dict, *args, **kwargs)
206
 
    return cb
207
 
 
208
 
 
209
 
class DeprecatedDict(dict):
210
 
    """A dictionary that complains when read or written."""
211
 
 
212
 
    is_deprecated = True
213
 
 
214
 
    def __init__(self,
215
 
        deprecation_version,
216
 
        variable_name,
217
 
        initial_value,
218
 
        advice,
219
 
        ):
220
 
        """Create a dict that warns when read or modified.
221
 
 
222
 
        :param deprecation_version: string for the warning format to raise,
223
 
            typically from deprecated_in()
224
 
        :param initial_value: The contents of the dict
225
 
        :param variable_name: This allows better warnings to be printed
226
 
        :param advice: String of advice on what callers should do instead
227
 
            of using this variable.
228
 
        """
229
 
        self._deprecation_version = deprecation_version
230
 
        self._variable_name = variable_name
231
 
        self._advice = advice
232
 
        dict.__init__(self, initial_value)
233
 
 
234
 
    # This isn't every possible method but it should trap anyone using the
235
 
    # dict -- add more if desired
236
 
    __len__ = _dict_deprecation_wrapper(dict.__len__)
237
 
    __getitem__ = _dict_deprecation_wrapper(dict.__getitem__)
238
 
    __setitem__ = _dict_deprecation_wrapper(dict.__setitem__)
239
 
    __delitem__ = _dict_deprecation_wrapper(dict.__delitem__)
240
 
    keys = _dict_deprecation_wrapper(dict.keys)
241
 
    __contains__ = _dict_deprecation_wrapper(dict.__contains__)
242
 
 
243
 
 
244
 
def deprecated_list(deprecation_version, variable_name,
245
 
                    initial_value, extra=None):
246
 
    """Create a list that warns when modified
247
 
 
248
 
    :param deprecation_version: string for the warning format to raise,
249
 
        typically from deprecated_in()
250
 
    :param initial_value: The contents of the list
251
 
    :param variable_name: This allows better warnings to be printed
252
 
    :param extra: Extra info to print when printing a warning
253
 
    """
254
 
 
255
 
    subst_text = 'Modifying %s' % (variable_name,)
256
 
    msg = deprecation_version % (subst_text,)
257
 
    if extra:
258
 
        msg += ' ' + extra
259
 
 
260
 
    class _DeprecatedList(list):
261
 
        __doc__ = list.__doc__ + msg
262
 
 
263
 
        is_deprecated = True
264
 
 
265
 
        def _warn_deprecated(self, func, *args, **kwargs):
266
 
            warn(msg, DeprecationWarning, stacklevel=3)
267
 
            return func(self, *args, **kwargs)
268
 
 
269
 
        def append(self, obj):
270
 
            """appending to %s is deprecated""" % (variable_name,)
271
 
            return self._warn_deprecated(list.append, obj)
272
 
 
273
 
        def insert(self, index, obj):
274
 
            """inserting to %s is deprecated""" % (variable_name,)
275
 
            return self._warn_deprecated(list.insert, index, obj)
276
 
 
277
 
        def extend(self, iterable):
278
 
            """extending %s is deprecated""" % (variable_name,)
279
 
            return self._warn_deprecated(list.extend, iterable)
280
 
 
281
 
        def remove(self, value):
282
 
            """removing from %s is deprecated""" % (variable_name,)
283
 
            return self._warn_deprecated(list.remove, value)
284
 
 
285
 
        def pop(self, index=None):
286
 
            """pop'ing from %s is deprecated""" % (variable_name,)
287
 
            if index:
288
 
                return self._warn_deprecated(list.pop, index)
289
 
            else:
290
 
                # Can't pass None
291
 
                return self._warn_deprecated(list.pop)
292
 
 
293
 
    return _DeprecatedList(initial_value)
294
 
 
295
 
 
296
 
def _check_for_filter(error_only):
297
 
    """Check if there is already a filter for deprecation warnings.
298
 
 
299
 
    :param error_only: Only match an 'error' filter
300
 
    :return: True if a filter is found, False otherwise
301
 
    """
302
 
    for filter in warnings.filters:
303
 
        if issubclass(DeprecationWarning, filter[2]):
304
 
            # This filter will effect DeprecationWarning
305
 
            if not error_only or filter[0] == 'error':
306
 
                return True
307
 
    return False
308
 
 
309
 
 
310
 
def _remove_filter_callable(filter):
311
 
    """Build and returns a callable removing filter from the warnings.
312
 
 
313
 
    :param filter: The filter to remove (can be None).
314
 
 
315
 
    :return: A callable that will remove filter from warnings.filters.
316
 
    """
317
 
    def cleanup():
318
 
        if filter:
319
 
            warnings.filters.remove(filter)
320
 
    return cleanup
321
 
 
322
 
 
323
 
def suppress_deprecation_warnings(override=True):
324
 
    """Call this function to suppress all deprecation warnings.
325
 
 
326
 
    When this is a final release version, we don't want to annoy users with
327
 
    lots of deprecation warnings. We only want the deprecation warnings when
328
 
    running a dev or release candidate.
329
 
 
330
 
    :param override: If True, always set the ignore, if False, only set the
331
 
        ignore if there isn't already a filter.
332
 
 
333
 
    :return: A callable to remove the new warnings this added.
334
 
    """
335
 
    if not override and _check_for_filter(error_only=False):
336
 
        # If there is already a filter effecting suppress_deprecation_warnings,
337
 
        # then skip it.
338
 
        filter = None
339
 
    else:
340
 
        warnings.filterwarnings('ignore', category=DeprecationWarning)
341
 
        filter = warnings.filters[0]
342
 
    return _remove_filter_callable(filter)
343
 
 
344
 
 
345
 
def activate_deprecation_warnings(override=True):
346
 
    """Call this function to activate deprecation warnings.
347
 
 
348
 
    When running in a 'final' release we suppress deprecation warnings.
349
 
    However, the test suite wants to see them. So when running selftest, we
350
 
    re-enable the deprecation warnings.
351
 
 
352
 
    Note: warnings that have already been issued under 'ignore' will not be
353
 
    reported after this point. The 'warnings' module has already marked them as
354
 
    handled, so they don't get issued again.
355
 
 
356
 
    :param override: If False, only add a filter if there isn't an error filter
357
 
        already. (This slightly differs from suppress_deprecation_warnings, in
358
 
        because it always overrides everything but -Werror).
359
 
 
360
 
    :return: A callable to remove the new warnings this added.
361
 
    """
362
 
    if not override and _check_for_filter(error_only=True):
363
 
        # DeprecationWarnings are already turned into errors, don't downgrade
364
 
        # them to 'default'.
365
 
        filter = None
366
 
    else:
367
 
        warnings.filterwarnings('default', category=DeprecationWarning)
368
 
        filter = warnings.filters[0]
369
 
    return _remove_filter_callable(filter)