Add error handling in range_count.

This commit is contained in:
Georg Brandl 2010-11-20 22:40:10 +00:00
parent 0b458d52f9
commit 7e5343b882
1 changed files with 4 additions and 1 deletions

View File

@ -337,7 +337,10 @@ static PyObject *
range_count(rangeobject *r, PyObject *ob)
{
if (PyLong_CheckExact(ob) || PyBool_Check(ob)) {
if (range_contains_long(r, ob))
int result = range_contains_long(r, ob);
if (result == -1)
return NULL;
else if (result)
return PyLong_FromLong(1);
else
return PyLong_FromLong(0);