From 5da854fe51ed76a236a44f82e6ee553a0ad9c51d Mon Sep 17 00:00:00 2001 From: "Michael W. Hudson" Date: Tue, 5 Nov 2002 17:38:05 +0000 Subject: [PATCH] This is Alex Martelli's patch [ 633870 ] allow any seq assignment to a list slice plus a very silly little test case of my own. --- Lib/test/test_types.py | 4 ++++ Objects/listobject.c | 21 ++++++++++++--------- 2 files changed, 16 insertions(+), 9 deletions(-) diff --git a/Lib/test/test_types.py b/Lib/test/test_types.py index a38eb7f7d4d..9777e838f66 100644 --- a/Lib/test/test_types.py +++ b/Lib/test/test_types.py @@ -297,6 +297,10 @@ a *= 0 if a != []: raise TestFailed, "list inplace repeat" +a = [] +a[:] = tuple(range(10)) +if a != range(10): + raise TestFailed, "assigning tuple to slice" print '6.5.3a Additional list operations' a = [0,1,2,3,4] diff --git a/Objects/listobject.c b/Objects/listobject.c index 45b0951da3e..c28bfb485a7 100644 --- a/Objects/listobject.c +++ b/Objects/listobject.c @@ -448,14 +448,22 @@ list_ass_slice(PyListObject *a, int ilow, int ihigh, PyObject *v) list. :-( */ PyObject **recycle, **p; PyObject **item; + PyObject *v_as_SF = NULL; /* PySequence_Fast(v) */ int n; /* Size of replacement list */ int d; /* Change in size */ int k; /* Loop index */ #define b ((PyListObject *)v) if (v == NULL) n = 0; - else if (PyList_Check(v)) { - n = b->ob_size; + else { + char msg[256]; + sprintf(msg, "must assign sequence (not \"%.200s\") to slice", + v->ob_type->tp_name); + v_as_SF = PySequence_Fast(v, msg); + if(v_as_SF == NULL) + return -1; + n = PySequence_Fast_GET_SIZE(v_as_SF); + if (a == b) { /* Special case "a[i:j] = a" -- copy b first */ int ret; @@ -465,12 +473,6 @@ list_ass_slice(PyListObject *a, int ilow, int ihigh, PyObject *v) return ret; } } - else { - PyErr_Format(PyExc_TypeError, - "must assign list (not \"%.200s\") to slice", - v->ob_type->tp_name); - return -1; - } if (ilow < 0) ilow = 0; else if (ilow > a->ob_size) @@ -512,7 +514,7 @@ list_ass_slice(PyListObject *a, int ilow, int ihigh, PyObject *v) a->ob_size += d; } for (k = 0; k < n; k++, ilow++) { - PyObject *w = b->ob_item[k]; + PyObject *w = PySequence_Fast_GET_ITEM(v_as_SF, k); Py_XINCREF(w); item[ilow] = w; } @@ -525,6 +527,7 @@ list_ass_slice(PyListObject *a, int ilow, int ihigh, PyObject *v) PyMem_FREE(a->ob_item); a->ob_item = NULL; } + Py_XDECREF(v_as_SF); return 0; #undef b }