~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/symbol_versioning.py

  • Committer: Martin Pool
  • Date: 2005-09-12 09:50:44 UTC
  • Revision ID: mbp@sourcefrog.net-20050912095044-6acfdb5611729987
- no tests in bzrlib.fetch anymore

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