Commit bfd249a7 authored by Andreas Zeidler's avatar Andreas Zeidler

Acquisition wrappers now correctly proxy `__iter__`.

parent 1df355d8
...@@ -237,6 +237,8 @@ Zope Changes ...@@ -237,6 +237,8 @@ Zope Changes
Bugs Fixed Bugs Fixed
- Acquisition wrappers now correctly proxy __iter__.
- Launchpad #174705: ensure that the error info object exposed to a - Launchpad #174705: ensure that the error info object exposed to a
'tal:on_error' handler has attributes visible to restricted code. 'tal:on_error' handler has attributes visible to restricted code.
......
...@@ -39,7 +39,7 @@ static PyObject *py__add__, *py__sub__, *py__mul__, *py__div__, ...@@ -39,7 +39,7 @@ static PyObject *py__add__, *py__sub__, *py__mul__, *py__div__,
*py__getitem__, *py__setitem__, *py__delitem__, *py__getitem__, *py__setitem__, *py__delitem__,
*py__getslice__, *py__setslice__, *py__delslice__, *py__contains__, *py__getslice__, *py__setslice__, *py__delslice__, *py__contains__,
*py__len__, *py__of__, *py__call__, *py__repr__, *py__str__, *py__cmp__, *py__len__, *py__of__, *py__call__, *py__repr__, *py__str__, *py__cmp__,
*py__parent__; *py__parent__, *py__iter__;
static PyObject *Acquired=0; static PyObject *Acquired=0;
...@@ -84,6 +84,7 @@ init_py_names(void) ...@@ -84,6 +84,7 @@ init_py_names(void)
INIT_PY_NAME(__str__); INIT_PY_NAME(__str__);
INIT_PY_NAME(__cmp__); INIT_PY_NAME(__cmp__);
INIT_PY_NAME(__parent__); INIT_PY_NAME(__parent__);
INIT_PY_NAME(__iter__);
#undef INIT_PY_NAME #undef INIT_PY_NAME
} }
...@@ -928,6 +929,12 @@ Wrapper_contains(Wrapper *self, PyObject *v) ...@@ -928,6 +929,12 @@ Wrapper_contains(Wrapper *self, PyObject *v)
return c; return c;
} }
static PyObject *
Wrapper_iter(Wrapper *self)
{
return CallMethodO(OBJECT(self), py__iter__, NULL, NULL);
}
static PySequenceMethods Wrapper_as_sequence = { static PySequenceMethods Wrapper_as_sequence = {
(inquiry)Wrapper_length, /*sq_length*/ (inquiry)Wrapper_length, /*sq_length*/
(binaryfunc)Wrapper_add, /*sq_concat*/ (binaryfunc)Wrapper_add, /*sq_concat*/
...@@ -1294,7 +1301,7 @@ static PyExtensionClass Wrappertype = { ...@@ -1294,7 +1301,7 @@ static PyExtensionClass Wrappertype = {
/* tp_clear */ (inquiry)Wrapper_clear, /* tp_clear */ (inquiry)Wrapper_clear,
/* tp_richcompare */ (richcmpfunc)Wrapper_richcompare, /* tp_richcompare */ (richcmpfunc)Wrapper_richcompare,
/* tp_weaklistoffset */ (long)0, /* tp_weaklistoffset */ (long)0,
/* tp_iter */ (getiterfunc)0, (getiterfunc)Wrapper_iter, /*tp_iter*/
/* tp_iternext */ (iternextfunc)0, /* tp_iternext */ (iternextfunc)0,
/* tp_methods */ Wrapper_methods, /* tp_methods */ Wrapper_methods,
/* tp_members */ 0, /* tp_members */ 0,
...@@ -1338,7 +1345,7 @@ static PyExtensionClass XaqWrappertype = { ...@@ -1338,7 +1345,7 @@ static PyExtensionClass XaqWrappertype = {
/* tp_clear */ (inquiry)Wrapper_clear, /* tp_clear */ (inquiry)Wrapper_clear,
/* tp_richcompare */ (richcmpfunc)Wrapper_richcompare, /* tp_richcompare */ (richcmpfunc)Wrapper_richcompare,
/* tp_weaklistoffset */ (long)0, /* tp_weaklistoffset */ (long)0,
/* tp_iter */ (getiterfunc)0, (getiterfunc)Wrapper_iter, /*tp_iter*/
/* tp_iternext */ (iternextfunc)0, /* tp_iternext */ (iternextfunc)0,
/* tp_methods */ Wrapper_methods, /* tp_methods */ Wrapper_methods,
/* tp_members */ 0, /* tp_members */ 0,
......
...@@ -1699,6 +1699,9 @@ def test_proxying(): ...@@ -1699,6 +1699,9 @@ def test_proxying():
... def __contains__(self, key): ... def __contains__(self, key):
... print 'contains', repr(key) ... print 'contains', repr(key)
... return key == 5 ... return key == 5
... def __iter__(self):
... print 'iterating...'
... return iter((42,))
The naked class behaves like this: The naked class behaves like this:
...@@ -1709,6 +1712,9 @@ def test_proxying(): ...@@ -1709,6 +1712,9 @@ def test_proxying():
>>> 5 in c >>> 5 in c
contains 5 contains 5
True True
>>> list(c)
iterating...
[42]
Let's put c in the context of i: Let's put c in the context of i:
...@@ -1723,6 +1729,58 @@ def test_proxying(): ...@@ -1723,6 +1729,58 @@ def test_proxying():
>>> 5 in i.c >>> 5 in i.c
contains 5 contains 5
True True
>>> list(i.c)
iterating...
[42]
Let's let's test the same again with an explicit wrapper:
>>> import Acquisition
>>> class Impl(Acquisition.Explicit):
... pass
>>> class C(Acquisition.Explicit):
... def __getitem__(self, key):
... print 'getitem', key
... if key == 4:
... raise IndexError
... return key
... def __contains__(self, key):
... print 'contains', repr(key)
... return key == 5
... def __iter__(self):
... print 'iterating...'
... return iter((42,))
The naked class behaves like this:
>>> c = C()
>>> 3 in c
contains 3
False
>>> 5 in c
contains 5
True
>>> list(c)
iterating...
[42]
Let's put c in the context of i:
>>> i = Impl()
>>> i.c = c
Now check that __contains__ is properly used:
>>> 3 in i.c # c.__of__(i)
contains 3
False
>>> 5 in i.c
contains 5
True
>>> list(i.c)
iterating...
[42]
""" """
......
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