Commit 0207e769 authored by Serhiy Storchaka's avatar Serhiy Storchaka

Issue #26822: itemgetter, attrgetter and methodcaller objects no longer

silently ignore keyword arguments.
parent bca06ccb
...@@ -371,6 +371,9 @@ class OperatorTestCase(unittest.TestCase): ...@@ -371,6 +371,9 @@ class OperatorTestCase(unittest.TestCase):
a.name = 'arthur' a.name = 'arthur'
f = operator.attrgetter('name') f = operator.attrgetter('name')
self.assertEqual(f(a), 'arthur') self.assertEqual(f(a), 'arthur')
self.assertRaises(TypeError, f)
self.assertRaises(TypeError, f, a, 'dent')
self.assertRaises(TypeError, f, a, surname='dent')
f = operator.attrgetter('rank') f = operator.attrgetter('rank')
self.assertRaises(AttributeError, f, a) self.assertRaises(AttributeError, f, a)
f = operator.attrgetter(2) f = operator.attrgetter(2)
...@@ -414,6 +417,9 @@ class OperatorTestCase(unittest.TestCase): ...@@ -414,6 +417,9 @@ class OperatorTestCase(unittest.TestCase):
a = 'ABCDE' a = 'ABCDE'
f = operator.itemgetter(2) f = operator.itemgetter(2)
self.assertEqual(f(a), 'C') self.assertEqual(f(a), 'C')
self.assertRaises(TypeError, f)
self.assertRaises(TypeError, f, a, 3)
self.assertRaises(TypeError, f, a, size=3)
f = operator.itemgetter(10) f = operator.itemgetter(10)
self.assertRaises(IndexError, f, a) self.assertRaises(IndexError, f, a)
...@@ -456,6 +462,9 @@ class OperatorTestCase(unittest.TestCase): ...@@ -456,6 +462,9 @@ class OperatorTestCase(unittest.TestCase):
self.assertRaises(IndexError, f, a) self.assertRaises(IndexError, f, a)
f = operator.methodcaller('foo', 1, 2) f = operator.methodcaller('foo', 1, 2)
self.assertEqual(f(a), 3) self.assertEqual(f(a), 3)
self.assertRaises(TypeError, f)
self.assertRaises(TypeError, f, a, 3)
self.assertRaises(TypeError, f, a, spam=3)
f = operator.methodcaller('bar') f = operator.methodcaller('bar')
self.assertEqual(f(a), 42) self.assertEqual(f(a), 42)
self.assertRaises(TypeError, f, a, a) self.assertRaises(TypeError, f, a, a)
......
...@@ -77,6 +77,9 @@ Core and Builtins ...@@ -77,6 +77,9 @@ Core and Builtins
Library Library
------- -------
- Issue #26822: itemgetter, attrgetter and methodcaller objects no longer
silently ignore keyword arguments.
- Issue #26657: Fix directory traversal vulnerability with SimpleHTTPServer - Issue #26657: Fix directory traversal vulnerability with SimpleHTTPServer
on Windows. This fixes a regression that was introduced in 2.7.7. Based on Windows. This fixes a regression that was introduced in 2.7.7. Based
on patch by Philipp Hagemeister. on patch by Philipp Hagemeister.
......
...@@ -511,6 +511,8 @@ itemgetter_call(itemgetterobject *ig, PyObject *args, PyObject *kw) ...@@ -511,6 +511,8 @@ itemgetter_call(itemgetterobject *ig, PyObject *args, PyObject *kw)
PyObject *obj, *result; PyObject *obj, *result;
Py_ssize_t i, nitems=ig->nitems; Py_ssize_t i, nitems=ig->nitems;
if (!_PyArg_NoKeywords("itemgetter", kw))
return NULL;
if (!PyArg_UnpackTuple(args, "itemgetter", 1, 1, &obj)) if (!PyArg_UnpackTuple(args, "itemgetter", 1, 1, &obj))
return NULL; return NULL;
if (nitems == 1) if (nitems == 1)
...@@ -691,6 +693,8 @@ attrgetter_call(attrgetterobject *ag, PyObject *args, PyObject *kw) ...@@ -691,6 +693,8 @@ attrgetter_call(attrgetterobject *ag, PyObject *args, PyObject *kw)
PyObject *obj, *result; PyObject *obj, *result;
Py_ssize_t i, nattrs=ag->nattrs; Py_ssize_t i, nattrs=ag->nattrs;
if (!_PyArg_NoKeywords("attrgetter", kw))
return NULL;
if (!PyArg_UnpackTuple(args, "attrgetter", 1, 1, &obj)) if (!PyArg_UnpackTuple(args, "attrgetter", 1, 1, &obj))
return NULL; return NULL;
if (ag->nattrs == 1) if (ag->nattrs == 1)
...@@ -838,6 +842,8 @@ methodcaller_call(methodcallerobject *mc, PyObject *args, PyObject *kw) ...@@ -838,6 +842,8 @@ methodcaller_call(methodcallerobject *mc, PyObject *args, PyObject *kw)
{ {
PyObject *method, *obj, *result; PyObject *method, *obj, *result;
if (!_PyArg_NoKeywords("methodcaller", kw))
return NULL;
if (!PyArg_UnpackTuple(args, "methodcaller", 1, 1, &obj)) if (!PyArg_UnpackTuple(args, "methodcaller", 1, 1, &obj))
return NULL; return NULL;
method = PyObject_GetAttr(obj, mc->name); method = PyObject_GetAttr(obj, mc->name);
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment