Mangle __parameters in __annotations__ dict properly. Issue #20625.

This commit is contained in:
Yury Selivanov 2014-02-18 12:49:41 -05:00
parent b59e4425d5
commit 34ce99f66d
5 changed files with 37 additions and 1 deletions

View File

@ -1781,6 +1781,10 @@ Changes in the Python API
perpetually alive (for example, database connections kept in thread-local
storage). (:issue:`17094`.)
* Parameter names in ``__annotations__`` dict are now mangled properly,
similarly to ``__kwdefaults__``. (Contributed by Yury Selivanov in
:issue:`20625`).
Changes in the C API
--------------------

View File

@ -314,6 +314,13 @@ class GrammarTests(unittest.TestCase):
self.assertEqual(f.__annotations__,
{'b': 1, 'c': 2, 'e': 3, 'g': 6, 'h': 7, 'j': 9,
'k': 11, 'return': 12})
# Check for issue #20625 -- annotations mangling
class Spam:
def f(self, *, __kw:1):
pass
class Ham(Spam): pass
self.assertEquals(Spam.f.__annotations__, {'_Spam__kw': 1})
self.assertEquals(Ham.f.__annotations__, {'_Spam__kw': 1})
# Check for SF Bug #1697248 - mixing decorators and a return annotation
def null(x): return x
@null

View File

@ -2390,6 +2390,22 @@ class TestSignatureObject(unittest.TestCase):
self.assertEqual(sig.return_annotation, 42)
self.assertEqual(sig, inspect.signature(test))
def test_signature_on_mangled_parameters(self):
class Spam:
def foo(self, __p1:1=2, *, __p2:2=3):
pass
class Ham(Spam):
pass
self.assertEqual(self.signature(Spam.foo),
((('self', ..., ..., "positional_or_keyword"),
('_Spam__p1', 2, 1, "positional_or_keyword"),
('_Spam__p2', 3, 2, "keyword_only")),
...))
self.assertEqual(self.signature(Spam.foo),
self.signature(Ham.foo))
class TestParameterObject(unittest.TestCase):
def test_signature_parameter_kinds(self):

View File

@ -22,6 +22,9 @@ Core and Builtins
- Issue #20595: Make getargs.c C89 compliant.
- Issue #20625: Parameter names in __annotations__ were not mangled properly.
Discovered by Jonas Wielicki, patch by Yury Selivanov.
Library
-------

View File

@ -1533,8 +1533,14 @@ compiler_visit_argannotation(struct compiler *c, identifier id,
{
if (annotation) {
VISIT(c, expr, annotation);
if (PyList_Append(names, id))
PyObject *mangled = _Py_Mangle(c->u->u_private, id);
if (!mangled)
return -1;
if (PyList_Append(names, mangled) < 0) {
Py_DECREF(mangled);
return -1;
}
Py_DECREF(mangled);
}
return 0;
}