| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485 |
- from __future__ import absolute_import
-
- import sys
-
-
- class WarningMessage(object):
-
- """Holds the result of a single showwarning() call."""
-
- _WARNING_DETAILS = ('message', 'category', 'filename', 'lineno', 'file',
- 'line')
-
- def __init__(self, message, category, filename, lineno, file=None,
- line=None):
- local_values = locals()
- for attr in self._WARNING_DETAILS:
- setattr(self, attr, local_values[attr])
-
- self._category_name = category and category.__name__ or None
-
- def __str__(self):
- return ('{message : %r, category : %r, filename : %r, lineno : %s, '
- 'line : %r}' % (self.message, self._category_name,
- self.filename, self.lineno, self.line))
-
-
- class catch_warnings(object):
-
- """A context manager that copies and restores the warnings filter upon
- exiting the context.
-
- The 'record' argument specifies whether warnings should be captured by a
- custom implementation of warnings.showwarning() and be appended to a list
- returned by the context manager. Otherwise None is returned by the context
- manager. The objects appended to the list are arguments whose attributes
- mirror the arguments to showwarning().
-
- The 'module' argument is to specify an alternative module to the module
- named 'warnings' and imported under that name. This argument is only
- useful when testing the warnings module itself.
-
- """
-
- def __init__(self, record=False, module=None):
- """Specify whether to record warnings and if an alternative module
- should be used other than sys.modules['warnings'].
-
- For compatibility with Python 3.0, please consider all arguments to be
- keyword-only.
-
- """
- self._record = record
- self._module = module is None and sys.modules['warnings'] or module
- self._entered = False
-
- def __repr__(self):
- args = []
- if self._record:
- args.append('record=True')
- if self._module is not sys.modules['warnings']:
- args.append('module=%r' % self._module)
- name = type(self).__name__
- return '%s(%s)' % (name, ', '.join(args))
-
- def __enter__(self):
- if self._entered:
- raise RuntimeError('Cannot enter %r twice' % self)
- self._entered = True
- self._filters = self._module.filters
- self._module.filters = self._filters[:]
- self._showwarning = self._module.showwarning
- if self._record:
- log = []
-
- def showwarning(*args, **kwargs):
- log.append(WarningMessage(*args, **kwargs))
-
- self._module.showwarning = showwarning
- return log
-
- def __exit__(self, *exc_info):
- if not self._entered:
- raise RuntimeError('Cannot exit %r without entering first' % self)
- self._module.filters = self._filters
- self._module.showwarning = self._showwarning
|
