Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit 87538e7

Browse files
pitroupitrou
committed
Issue #4180: The warnings registries are now reset when the filters are modified.
2 parents 605a64b + cb0a006 commit 87538e7

4 files changed

Lines changed: 101 additions & 9 deletions

File tree

‎Lib/test/test_warnings.py‎

Lines changed: 47 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,16 @@ def test_error(self):
9292
self.assertRaises(UserWarning, self.module.warn,
9393
"FilterTests.test_error")
9494

95+
def test_error_after_default(self):
96+
with original_warnings.catch_warnings(module=self.module) as w:
97+
self.module.resetwarnings()
98+
message = "FilterTests.test_ignore_after_default"
99+
def f():
100+
self.module.warn(message, UserWarning)
101+
f()
102+
self.module.filterwarnings("error", category=UserWarning)
103+
self.assertRaises(UserWarning, f)
104+
95105
def test_ignore(self):
96106
with original_warnings.catch_warnings(record=True,
97107
module=self.module) as w:
@@ -100,6 +110,19 @@ def test_ignore(self):
100110
self.module.warn("FilterTests.test_ignore", UserWarning)
101111
self.assertEqual(len(w), 0)
102112

113+
def test_ignore_after_default(self):
114+
with original_warnings.catch_warnings(record=True,
115+
module=self.module) as w:
116+
self.module.resetwarnings()
117+
message = "FilterTests.test_ignore_after_default"
118+
def f():
119+
self.module.warn(message, UserWarning)
120+
f()
121+
self.module.filterwarnings("ignore", category=UserWarning)
122+
f()
123+
f()
124+
self.assertEqual(len(w), 1)
125+
103126
def test_always(self):
104127
with original_warnings.catch_warnings(record=True,
105128
module=self.module) as w:
@@ -111,6 +134,26 @@ def test_always(self):
111134
self.module.warn(message, UserWarning)
112135
self.assertTrue(w[-1].message, message)
113136

137+
def test_always_after_default(self):
138+
with original_warnings.catch_warnings(record=True,
139+
module=self.module) as w:
140+
self.module.resetwarnings()
141+
message = "FilterTests.test_always_after_ignore"
142+
def f():
143+
self.module.warn(message, UserWarning)
144+
f()
145+
self.assertEqual(len(w), 1)
146+
self.assertEqual(w[-1].message.args[0], message)
147+
f()
148+
self.assertEqual(len(w), 1)
149+
self.module.filterwarnings("always", category=UserWarning)
150+
f()
151+
self.assertEqual(len(w), 2)
152+
self.assertEqual(w[-1].message.args[0], message)
153+
f()
154+
self.assertEqual(len(w), 3)
155+
self.assertEqual(w[-1].message.args[0], message)
156+
114157
def test_default(self):
115158
with original_warnings.catch_warnings(record=True,
116159
module=self.module) as w:
@@ -541,7 +584,9 @@ def test_default_action(self):
541584
registry=registry)
542585
self.assertEqual(w[-1].message, message)
543586
self.assertEqual(len(w), 1)
544-
self.assertEqual(len(registry), 1)
587+
# One actual registry key plus the "version" key
588+
self.assertEqual(len(registry), 2)
589+
self.assertIn("version", registry)
545590
del w[:]
546591
# Test removal.
547592
del self.module.defaultaction
@@ -551,7 +596,7 @@ def test_default_action(self):
551596
registry=registry)
552597
self.assertEqual(w[-1].message, message)
553598
self.assertEqual(len(w), 1)
554-
self.assertEqual(len(registry), 1)
599+
self.assertEqual(len(registry), 2)
555600
del w[:]
556601
# Test setting.
557602
self.module.defaultaction = "ignore"

‎Lib/warnings.py‎

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,7 @@ def filterwarnings(action, message="", category=Warning, module="", lineno=0,
5353
filters.append(item)
5454
else:
5555
filters.insert(0, item)
56+
_filters_mutated()
5657

5758
def simplefilter(action, category=Warning, lineno=0, append=False):
5859
"""Insert a simple entry into the list of warnings filters (at the front).
@@ -73,10 +74,12 @@ def simplefilter(action, category=Warning, lineno=0, append=False):
7374
filters.append(item)
7475
else:
7576
filters.insert(0, item)
77+
_filters_mutated()
7678

7779
def resetwarnings():
7880
"""Clear the list of warning filters, so that no filters are active."""
7981
filters[:] = []
82+
_filters_mutated()
8083

8184
class _OptionError(Exception):
8285
"""Exception used by option processing helpers."""
@@ -206,6 +209,9 @@ def warn_explicit(message, category, filename, lineno,
206209
module = module[:-3] # XXX What about leading pathname?
207210
if registry is None:
208211
registry = {}
212+
if registry.get('version', 0) != _filters_version:
213+
registry.clear()
214+
registry['version'] = _filters_version
209215
if isinstance(message, Warning):
210216
text = str(message)
211217
category = message.__class__
@@ -331,6 +337,7 @@ def __enter__(self):
331337
self._entered = True
332338
self._filters = self._module.filters
333339
self._module.filters = self._filters[:]
340+
self._module._filters_mutated()
334341
self._showwarning = self._module.showwarning
335342
if self._record:
336343
log = []
@@ -345,6 +352,7 @@ def __exit__(self, *exc_info):
345352
if not self._entered:
346353
raise RuntimeError("Cannot exit %r without entering first" % self)
347354
self._module.filters = self._filters
355+
self._module._filters_mutated()
348356
self._module.showwarning = self._showwarning
349357

350358

@@ -359,15 +367,22 @@ def __exit__(self, *exc_info):
359367
_warnings_defaults = False
360368
try:
361369
from _warnings import (filters, _defaultaction, _onceregistry,
362-
warn, warn_explicit)
370+
warn, warn_explicit, _filters_mutated)
363371
defaultaction = _defaultaction
364372
onceregistry = _onceregistry
365373
_warnings_defaults = True
374+
366375
except ImportError:
367376
filters = []
368377
defaultaction = "default"
369378
onceregistry = {}
370379

380+
_filters_version = 1
381+
382+
def _filters_mutated():
383+
global _filters_version
384+
_filters_version += 1
385+
371386

372387
# Module initialization
373388
_processoptions(sys.warnoptions)

‎Misc/NEWS‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -132,6 +132,9 @@ Core and Builtins
132132
Library
133133
-------
134134

135+
- Issue #4180: The warnings registries are now reset when the filters
136+
are modified.
137+
135138
- Issue #22419: Limit the length of incoming HTTP request in wsgiref server to
136139
65536 bytes and send a 414 error code for higher lengths. Patch contributed
137140
by Devin Cook.

‎Python/_warnings.c‎

Lines changed: 35 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ MODULE_NAME " provides basic warning filtering support.\n"
1212
static PyObject *_filters; /* List */
1313
static PyObject *_once_registry; /* Dict */
1414
static PyObject *_default_action; /* String */
15+
static long _filters_version;
1516

1617
_Py_IDENTIFIER(argv);
1718
_Py_IDENTIFIER(stderr);
@@ -178,16 +179,33 @@ get_filter(PyObject *category, PyObject *text, Py_ssize_t lineno,
178179
static int
179180
already_warned(PyObject *registry, PyObject *key, int should_set)
180181
{
181-
PyObject *already_warned;
182+
PyObject *version_obj, *already_warned;
183+
_Py_IDENTIFIER(version);
182184

183185
if (key == NULL)
184186
return -1;
185187

186-
already_warned = PyDict_GetItem(registry, key);
187-
if (already_warned != NULL) {
188-
int rc = PyObject_IsTrue(already_warned);
189-
if (rc != 0)
190-
return rc;
188+
version_obj = _PyDict_GetItemId(registry, &PyId_version);
189+
if (version_obj == NULL
190+
|| !PyLong_CheckExact(version_obj)
191+
|| PyLong_AsLong(version_obj) != _filters_version) {
192+
PyDict_Clear(registry);
193+
version_obj = PyLong_FromLong(_filters_version);
194+
if (version_obj == NULL)
195+
return -1;
196+
if (_PyDict_SetItemId(registry, &PyId_version, version_obj) < 0) {
197+
Py_DECREF(version_obj);
198+
return -1;
199+
}
200+
Py_DECREF(version_obj);
201+
}
202+
else {
203+
already_warned = PyDict_GetItem(registry, key);
204+
if (already_warned != NULL) {
205+
int rc = PyObject_IsTrue(already_warned);
206+
if (rc != 0)
207+
return rc;
208+
}
191209
}
192210

193211
/* This warning wasn't found in the registry, set it. */
@@ -751,6 +769,13 @@ warnings_warn_explicit(PyObject *self, PyObject *args, PyObject *kwds)
751769
registry, NULL);
752770
}
753771

772+
static PyObject *
773+
warnings_filters_mutated(PyObject *self, PyObject *args)
774+
{
775+
_filters_version++;
776+
Py_RETURN_NONE;
777+
}
778+
754779

755780
/* Function to issue a warning message; may raise an exception. */
756781

@@ -918,6 +943,8 @@ static PyMethodDef warnings_functions[] = {
918943
warn_doc},
919944
{"warn_explicit", (PyCFunction)warnings_warn_explicit,
920945
METH_VARARGS | METH_KEYWORDS, warn_explicit_doc},
946+
{"_filters_mutated", (PyCFunction)warnings_filters_mutated, METH_NOARGS,
947+
NULL},
921948
/* XXX(brett.cannon): add showwarning? */
922949
/* XXX(brett.cannon): Reasonable to add formatwarning? */
923950
{NULL, NULL} /* sentinel */
@@ -1070,5 +1097,7 @@ _PyWarnings_Init(void)
10701097
Py_INCREF(_default_action);
10711098
if (PyModule_AddObject(m, "_defaultaction", _default_action) < 0)
10721099
return NULL;
1100+
1101+
_filters_version = 0;
10731102
return m;
10741103
}

0 commit comments

Comments
 (0)