2523 lines
74 KiB
C
2523 lines
74 KiB
C
/* Built-in functions */
|
|
|
|
#include "Python.h"
|
|
#include "Python-ast.h"
|
|
|
|
#include "node.h"
|
|
#include "code.h"
|
|
|
|
#include "asdl.h"
|
|
#include "ast.h"
|
|
|
|
#include <ctype.h>
|
|
|
|
#ifdef HAVE_LANGINFO_H
|
|
#include <langinfo.h> /* CODESET */
|
|
#endif
|
|
|
|
/* The default encoding used by the platform file system APIs
|
|
Can remain NULL for all platforms that don't have such a concept
|
|
|
|
Don't forget to modify PyUnicode_DecodeFSDefault() if you touch any of the
|
|
values for Py_FileSystemDefaultEncoding!
|
|
*/
|
|
#ifdef HAVE_MBCS
|
|
const char *Py_FileSystemDefaultEncoding = "mbcs";
|
|
int Py_HasFileSystemDefaultEncoding = 1;
|
|
#elif defined(__APPLE__)
|
|
const char *Py_FileSystemDefaultEncoding = "utf-8";
|
|
int Py_HasFileSystemDefaultEncoding = 1;
|
|
#else
|
|
const char *Py_FileSystemDefaultEncoding = NULL; /* set by initfsencoding() */
|
|
int Py_HasFileSystemDefaultEncoding = 0;
|
|
#endif
|
|
|
|
_Py_IDENTIFIER(__builtins__);
|
|
_Py_IDENTIFIER(__dict__);
|
|
_Py_IDENTIFIER(__prepare__);
|
|
_Py_IDENTIFIER(__round__);
|
|
_Py_IDENTIFIER(encoding);
|
|
_Py_IDENTIFIER(errors);
|
|
_Py_IDENTIFIER(fileno);
|
|
_Py_IDENTIFIER(flush);
|
|
_Py_IDENTIFIER(metaclass);
|
|
_Py_IDENTIFIER(sort);
|
|
_Py_IDENTIFIER(stdin);
|
|
_Py_IDENTIFIER(stdout);
|
|
_Py_IDENTIFIER(stderr);
|
|
|
|
static PyObject *
|
|
builtin___build_class__(PyObject *self, PyObject *args, PyObject *kwds)
|
|
{
|
|
PyObject *func, *name, *bases, *mkw, *meta, *winner, *prep, *ns, *cell;
|
|
PyObject *cls = NULL;
|
|
Py_ssize_t nargs;
|
|
int isclass;
|
|
|
|
assert(args != NULL);
|
|
if (!PyTuple_Check(args)) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"__build_class__: args is not a tuple");
|
|
return NULL;
|
|
}
|
|
nargs = PyTuple_GET_SIZE(args);
|
|
if (nargs < 2) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"__build_class__: not enough arguments");
|
|
return NULL;
|
|
}
|
|
func = PyTuple_GET_ITEM(args, 0); /* Better be callable */
|
|
if (!PyFunction_Check(func)) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"__build__class__: func must be a function");
|
|
return NULL;
|
|
}
|
|
name = PyTuple_GET_ITEM(args, 1);
|
|
if (!PyUnicode_Check(name)) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"__build_class__: name is not a string");
|
|
return NULL;
|
|
}
|
|
bases = PyTuple_GetSlice(args, 2, nargs);
|
|
if (bases == NULL)
|
|
return NULL;
|
|
|
|
if (kwds == NULL) {
|
|
meta = NULL;
|
|
mkw = NULL;
|
|
}
|
|
else {
|
|
mkw = PyDict_Copy(kwds); /* Don't modify kwds passed in! */
|
|
if (mkw == NULL) {
|
|
Py_DECREF(bases);
|
|
return NULL;
|
|
}
|
|
meta = _PyDict_GetItemId(mkw, &PyId_metaclass);
|
|
if (meta != NULL) {
|
|
Py_INCREF(meta);
|
|
if (_PyDict_DelItemId(mkw, &PyId_metaclass) < 0) {
|
|
Py_DECREF(meta);
|
|
Py_DECREF(mkw);
|
|
Py_DECREF(bases);
|
|
return NULL;
|
|
}
|
|
/* metaclass is explicitly given, check if it's indeed a class */
|
|
isclass = PyType_Check(meta);
|
|
}
|
|
}
|
|
if (meta == NULL) {
|
|
/* if there are no bases, use type: */
|
|
if (PyTuple_GET_SIZE(bases) == 0) {
|
|
meta = (PyObject *) (&PyType_Type);
|
|
}
|
|
/* else get the type of the first base */
|
|
else {
|
|
PyObject *base0 = PyTuple_GET_ITEM(bases, 0);
|
|
meta = (PyObject *) (base0->ob_type);
|
|
}
|
|
Py_INCREF(meta);
|
|
isclass = 1; /* meta is really a class */
|
|
}
|
|
|
|
if (isclass) {
|
|
/* meta is really a class, so check for a more derived
|
|
metaclass, or possible metaclass conflicts: */
|
|
winner = (PyObject *)_PyType_CalculateMetaclass((PyTypeObject *)meta,
|
|
bases);
|
|
if (winner == NULL) {
|
|
Py_DECREF(meta);
|
|
Py_XDECREF(mkw);
|
|
Py_DECREF(bases);
|
|
return NULL;
|
|
}
|
|
if (winner != meta) {
|
|
Py_DECREF(meta);
|
|
meta = winner;
|
|
Py_INCREF(meta);
|
|
}
|
|
}
|
|
/* else: meta is not a class, so we cannot do the metaclass
|
|
calculation, so we will use the explicitly given object as it is */
|
|
prep = _PyObject_GetAttrId(meta, &PyId___prepare__);
|
|
if (prep == NULL) {
|
|
if (PyErr_ExceptionMatches(PyExc_AttributeError)) {
|
|
PyErr_Clear();
|
|
ns = PyDict_New();
|
|
}
|
|
else {
|
|
Py_DECREF(meta);
|
|
Py_XDECREF(mkw);
|
|
Py_DECREF(bases);
|
|
return NULL;
|
|
}
|
|
}
|
|
else {
|
|
PyObject *pargs = PyTuple_Pack(2, name, bases);
|
|
if (pargs == NULL) {
|
|
Py_DECREF(prep);
|
|
Py_DECREF(meta);
|
|
Py_XDECREF(mkw);
|
|
Py_DECREF(bases);
|
|
return NULL;
|
|
}
|
|
ns = PyEval_CallObjectWithKeywords(prep, pargs, mkw);
|
|
Py_DECREF(pargs);
|
|
Py_DECREF(prep);
|
|
}
|
|
if (ns == NULL) {
|
|
Py_DECREF(meta);
|
|
Py_XDECREF(mkw);
|
|
Py_DECREF(bases);
|
|
return NULL;
|
|
}
|
|
cell = PyEval_EvalCodeEx(PyFunction_GET_CODE(func), PyFunction_GET_GLOBALS(func), ns,
|
|
NULL, 0, NULL, 0, NULL, 0, NULL,
|
|
PyFunction_GET_CLOSURE(func));
|
|
if (cell != NULL) {
|
|
PyObject *margs;
|
|
margs = PyTuple_Pack(3, name, bases, ns);
|
|
if (margs != NULL) {
|
|
cls = PyEval_CallObjectWithKeywords(meta, margs, mkw);
|
|
Py_DECREF(margs);
|
|
}
|
|
if (cls != NULL && PyCell_Check(cell))
|
|
PyCell_Set(cell, cls);
|
|
Py_DECREF(cell);
|
|
}
|
|
Py_DECREF(ns);
|
|
Py_DECREF(meta);
|
|
Py_XDECREF(mkw);
|
|
Py_DECREF(bases);
|
|
return cls;
|
|
}
|
|
|
|
PyDoc_STRVAR(build_class_doc,
|
|
"__build_class__(func, name, *bases, metaclass=None, **kwds) -> class\n\
|
|
\n\
|
|
Internal helper function used by the class statement.");
|
|
|
|
static PyObject *
|
|
builtin___import__(PyObject *self, PyObject *args, PyObject *kwds)
|
|
{
|
|
static char *kwlist[] = {"name", "globals", "locals", "fromlist",
|
|
"level", 0};
|
|
PyObject *name, *globals = NULL, *locals = NULL, *fromlist = NULL;
|
|
int level = 0;
|
|
|
|
if (!PyArg_ParseTupleAndKeywords(args, kwds, "U|OOOi:__import__",
|
|
kwlist, &name, &globals, &locals, &fromlist, &level))
|
|
return NULL;
|
|
return PyImport_ImportModuleLevelObject(name, globals, locals,
|
|
fromlist, level);
|
|
}
|
|
|
|
PyDoc_STRVAR(import_doc,
|
|
"__import__(name, globals=None, locals=None, fromlist=(), level=0) -> module\n\
|
|
\n\
|
|
Import a module. Because this function is meant for use by the Python\n\
|
|
interpreter and not for general use it is better to use\n\
|
|
importlib.import_module() to programmatically import a module.\n\
|
|
\n\
|
|
The globals argument is only used to determine the context;\n\
|
|
they are not modified. The locals argument is unused. The fromlist\n\
|
|
should be a list of names to emulate ``from name import ...'', or an\n\
|
|
empty list to emulate ``import name''.\n\
|
|
When importing a module from a package, note that __import__('A.B', ...)\n\
|
|
returns package A when fromlist is empty, but its submodule B when\n\
|
|
fromlist is not empty. Level is used to determine whether to perform \n\
|
|
absolute or relative imports. 0 is absolute while a positive number\n\
|
|
is the number of parent directories to search relative to the current module.");
|
|
|
|
|
|
static PyObject *
|
|
builtin_abs(PyObject *self, PyObject *v)
|
|
{
|
|
return PyNumber_Absolute(v);
|
|
}
|
|
|
|
PyDoc_STRVAR(abs_doc,
|
|
"abs(number) -> number\n\
|
|
\n\
|
|
Return the absolute value of the argument.");
|
|
|
|
static PyObject *
|
|
builtin_all(PyObject *self, PyObject *v)
|
|
{
|
|
PyObject *it, *item;
|
|
PyObject *(*iternext)(PyObject *);
|
|
int cmp;
|
|
|
|
it = PyObject_GetIter(v);
|
|
if (it == NULL)
|
|
return NULL;
|
|
iternext = *Py_TYPE(it)->tp_iternext;
|
|
|
|
for (;;) {
|
|
item = iternext(it);
|
|
if (item == NULL)
|
|
break;
|
|
cmp = PyObject_IsTrue(item);
|
|
Py_DECREF(item);
|
|
if (cmp < 0) {
|
|
Py_DECREF(it);
|
|
return NULL;
|
|
}
|
|
if (cmp == 0) {
|
|
Py_DECREF(it);
|
|
Py_RETURN_FALSE;
|
|
}
|
|
}
|
|
Py_DECREF(it);
|
|
if (PyErr_Occurred()) {
|
|
if (PyErr_ExceptionMatches(PyExc_StopIteration))
|
|
PyErr_Clear();
|
|
else
|
|
return NULL;
|
|
}
|
|
Py_RETURN_TRUE;
|
|
}
|
|
|
|
PyDoc_STRVAR(all_doc,
|
|
"all(iterable) -> bool\n\
|
|
\n\
|
|
Return True if bool(x) is True for all values x in the iterable.\n\
|
|
If the iterable is empty, return True.");
|
|
|
|
static PyObject *
|
|
builtin_any(PyObject *self, PyObject *v)
|
|
{
|
|
PyObject *it, *item;
|
|
PyObject *(*iternext)(PyObject *);
|
|
int cmp;
|
|
|
|
it = PyObject_GetIter(v);
|
|
if (it == NULL)
|
|
return NULL;
|
|
iternext = *Py_TYPE(it)->tp_iternext;
|
|
|
|
for (;;) {
|
|
item = iternext(it);
|
|
if (item == NULL)
|
|
break;
|
|
cmp = PyObject_IsTrue(item);
|
|
Py_DECREF(item);
|
|
if (cmp < 0) {
|
|
Py_DECREF(it);
|
|
return NULL;
|
|
}
|
|
if (cmp == 1) {
|
|
Py_DECREF(it);
|
|
Py_RETURN_TRUE;
|
|
}
|
|
}
|
|
Py_DECREF(it);
|
|
if (PyErr_Occurred()) {
|
|
if (PyErr_ExceptionMatches(PyExc_StopIteration))
|
|
PyErr_Clear();
|
|
else
|
|
return NULL;
|
|
}
|
|
Py_RETURN_FALSE;
|
|
}
|
|
|
|
PyDoc_STRVAR(any_doc,
|
|
"any(iterable) -> bool\n\
|
|
\n\
|
|
Return True if bool(x) is True for any x in the iterable.\n\
|
|
If the iterable is empty, return False.");
|
|
|
|
static PyObject *
|
|
builtin_ascii(PyObject *self, PyObject *v)
|
|
{
|
|
return PyObject_ASCII(v);
|
|
}
|
|
|
|
PyDoc_STRVAR(ascii_doc,
|
|
"ascii(object) -> string\n\
|
|
\n\
|
|
As repr(), return a string containing a printable representation of an\n\
|
|
object, but escape the non-ASCII characters in the string returned by\n\
|
|
repr() using \\x, \\u or \\U escapes. This generates a string similar\n\
|
|
to that returned by repr() in Python 2.");
|
|
|
|
|
|
static PyObject *
|
|
builtin_bin(PyObject *self, PyObject *v)
|
|
{
|
|
return PyNumber_ToBase(v, 2);
|
|
}
|
|
|
|
PyDoc_STRVAR(bin_doc,
|
|
"bin(number) -> string\n\
|
|
\n\
|
|
Return the binary representation of an integer.\n\
|
|
\n\
|
|
>>> bin(2796202)\n\
|
|
'0b1010101010101010101010'\n\
|
|
");
|
|
|
|
|
|
static PyObject *
|
|
builtin_callable(PyObject *self, PyObject *v)
|
|
{
|
|
return PyBool_FromLong((long)PyCallable_Check(v));
|
|
}
|
|
|
|
PyDoc_STRVAR(callable_doc,
|
|
"callable(object) -> bool\n\
|
|
\n\
|
|
Return whether the object is callable (i.e., some kind of function).\n\
|
|
Note that classes are callable, as are instances of classes with a\n\
|
|
__call__() method.");
|
|
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
PyObject *func;
|
|
PyObject *it;
|
|
} filterobject;
|
|
|
|
static PyObject *
|
|
filter_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
|
|
{
|
|
PyObject *func, *seq;
|
|
PyObject *it;
|
|
filterobject *lz;
|
|
|
|
if (type == &PyFilter_Type && !_PyArg_NoKeywords("filter()", kwds))
|
|
return NULL;
|
|
|
|
if (!PyArg_UnpackTuple(args, "filter", 2, 2, &func, &seq))
|
|
return NULL;
|
|
|
|
/* Get iterator. */
|
|
it = PyObject_GetIter(seq);
|
|
if (it == NULL)
|
|
return NULL;
|
|
|
|
/* create filterobject structure */
|
|
lz = (filterobject *)type->tp_alloc(type, 0);
|
|
if (lz == NULL) {
|
|
Py_DECREF(it);
|
|
return NULL;
|
|
}
|
|
Py_INCREF(func);
|
|
lz->func = func;
|
|
lz->it = it;
|
|
|
|
return (PyObject *)lz;
|
|
}
|
|
|
|
static void
|
|
filter_dealloc(filterobject *lz)
|
|
{
|
|
PyObject_GC_UnTrack(lz);
|
|
Py_XDECREF(lz->func);
|
|
Py_XDECREF(lz->it);
|
|
Py_TYPE(lz)->tp_free(lz);
|
|
}
|
|
|
|
static int
|
|
filter_traverse(filterobject *lz, visitproc visit, void *arg)
|
|
{
|
|
Py_VISIT(lz->it);
|
|
Py_VISIT(lz->func);
|
|
return 0;
|
|
}
|
|
|
|
static PyObject *
|
|
filter_next(filterobject *lz)
|
|
{
|
|
PyObject *item;
|
|
PyObject *it = lz->it;
|
|
long ok;
|
|
PyObject *(*iternext)(PyObject *);
|
|
|
|
iternext = *Py_TYPE(it)->tp_iternext;
|
|
for (;;) {
|
|
item = iternext(it);
|
|
if (item == NULL)
|
|
return NULL;
|
|
|
|
if (lz->func == Py_None || lz->func == (PyObject *)&PyBool_Type) {
|
|
ok = PyObject_IsTrue(item);
|
|
} else {
|
|
PyObject *good;
|
|
good = PyObject_CallFunctionObjArgs(lz->func,
|
|
item, NULL);
|
|
if (good == NULL) {
|
|
Py_DECREF(item);
|
|
return NULL;
|
|
}
|
|
ok = PyObject_IsTrue(good);
|
|
Py_DECREF(good);
|
|
}
|
|
if (ok > 0)
|
|
return item;
|
|
Py_DECREF(item);
|
|
if (ok < 0)
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
static PyObject *
|
|
filter_reduce(filterobject *lz)
|
|
{
|
|
return Py_BuildValue("O(OO)", Py_TYPE(lz), lz->func, lz->it);
|
|
}
|
|
|
|
PyDoc_STRVAR(reduce_doc, "Return state information for pickling.");
|
|
|
|
static PyMethodDef filter_methods[] = {
|
|
{"__reduce__", (PyCFunction)filter_reduce, METH_NOARGS, reduce_doc},
|
|
{NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
PyDoc_STRVAR(filter_doc,
|
|
"filter(function or None, iterable) --> filter object\n\
|
|
\n\
|
|
Return an iterator yielding those items of iterable for which function(item)\n\
|
|
is true. If function is None, return the items that are true.");
|
|
|
|
PyTypeObject PyFilter_Type = {
|
|
PyVarObject_HEAD_INIT(&PyType_Type, 0)
|
|
"filter", /* tp_name */
|
|
sizeof(filterobject), /* tp_basicsize */
|
|
0, /* tp_itemsize */
|
|
/* methods */
|
|
(destructor)filter_dealloc, /* tp_dealloc */
|
|
0, /* tp_print */
|
|
0, /* tp_getattr */
|
|
0, /* tp_setattr */
|
|
0, /* tp_reserved */
|
|
0, /* tp_repr */
|
|
0, /* tp_as_number */
|
|
0, /* tp_as_sequence */
|
|
0, /* tp_as_mapping */
|
|
0, /* tp_hash */
|
|
0, /* tp_call */
|
|
0, /* tp_str */
|
|
PyObject_GenericGetAttr, /* tp_getattro */
|
|
0, /* tp_setattro */
|
|
0, /* tp_as_buffer */
|
|
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC |
|
|
Py_TPFLAGS_BASETYPE, /* tp_flags */
|
|
filter_doc, /* tp_doc */
|
|
(traverseproc)filter_traverse, /* tp_traverse */
|
|
0, /* tp_clear */
|
|
0, /* tp_richcompare */
|
|
0, /* tp_weaklistoffset */
|
|
PyObject_SelfIter, /* tp_iter */
|
|
(iternextfunc)filter_next, /* tp_iternext */
|
|
filter_methods, /* tp_methods */
|
|
0, /* tp_members */
|
|
0, /* tp_getset */
|
|
0, /* tp_base */
|
|
0, /* tp_dict */
|
|
0, /* tp_descr_get */
|
|
0, /* tp_descr_set */
|
|
0, /* tp_dictoffset */
|
|
0, /* tp_init */
|
|
PyType_GenericAlloc, /* tp_alloc */
|
|
filter_new, /* tp_new */
|
|
PyObject_GC_Del, /* tp_free */
|
|
};
|
|
|
|
|
|
static PyObject *
|
|
builtin_format(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *value;
|
|
PyObject *format_spec = NULL;
|
|
|
|
if (!PyArg_ParseTuple(args, "O|U:format", &value, &format_spec))
|
|
return NULL;
|
|
|
|
return PyObject_Format(value, format_spec);
|
|
}
|
|
|
|
PyDoc_STRVAR(format_doc,
|
|
"format(value[, format_spec]) -> string\n\
|
|
\n\
|
|
Returns value.__format__(format_spec)\n\
|
|
format_spec defaults to \"\"");
|
|
|
|
static PyObject *
|
|
builtin_chr(PyObject *self, PyObject *args)
|
|
{
|
|
int x;
|
|
|
|
if (!PyArg_ParseTuple(args, "i:chr", &x))
|
|
return NULL;
|
|
|
|
return PyUnicode_FromOrdinal(x);
|
|
}
|
|
|
|
PyDoc_STRVAR(chr_doc,
|
|
"chr(i) -> Unicode character\n\
|
|
\n\
|
|
Return a Unicode string of one character with ordinal i; 0 <= i <= 0x10ffff.");
|
|
|
|
|
|
static char *
|
|
source_as_string(PyObject *cmd, char *funcname, char *what, PyCompilerFlags *cf)
|
|
{
|
|
char *str;
|
|
Py_ssize_t size;
|
|
|
|
if (PyUnicode_Check(cmd)) {
|
|
cf->cf_flags |= PyCF_IGNORE_COOKIE;
|
|
str = PyUnicode_AsUTF8AndSize(cmd, &size);
|
|
if (str == NULL)
|
|
return NULL;
|
|
}
|
|
else if (!PyObject_CheckReadBuffer(cmd)) {
|
|
PyErr_Format(PyExc_TypeError,
|
|
"%s() arg 1 must be a %s object",
|
|
funcname, what);
|
|
return NULL;
|
|
}
|
|
else if (PyObject_AsReadBuffer(cmd, (const void **)&str, &size) < 0) {
|
|
return NULL;
|
|
}
|
|
|
|
if (strlen(str) != size) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"source code string cannot contain null bytes");
|
|
return NULL;
|
|
}
|
|
return str;
|
|
}
|
|
|
|
static PyObject *
|
|
builtin_compile(PyObject *self, PyObject *args, PyObject *kwds)
|
|
{
|
|
char *str;
|
|
PyObject *filename;
|
|
char *startstr;
|
|
int mode = -1;
|
|
int dont_inherit = 0;
|
|
int supplied_flags = 0;
|
|
int optimize = -1;
|
|
int is_ast;
|
|
PyCompilerFlags cf;
|
|
PyObject *cmd;
|
|
static char *kwlist[] = {"source", "filename", "mode", "flags",
|
|
"dont_inherit", "optimize", NULL};
|
|
int start[] = {Py_file_input, Py_eval_input, Py_single_input};
|
|
PyObject *result;
|
|
|
|
if (!PyArg_ParseTupleAndKeywords(args, kwds, "OO&s|iii:compile", kwlist,
|
|
&cmd,
|
|
PyUnicode_FSDecoder, &filename,
|
|
&startstr, &supplied_flags,
|
|
&dont_inherit, &optimize))
|
|
return NULL;
|
|
|
|
cf.cf_flags = supplied_flags | PyCF_SOURCE_IS_UTF8;
|
|
|
|
if (supplied_flags &
|
|
~(PyCF_MASK | PyCF_MASK_OBSOLETE | PyCF_DONT_IMPLY_DEDENT | PyCF_ONLY_AST))
|
|
{
|
|
PyErr_SetString(PyExc_ValueError,
|
|
"compile(): unrecognised flags");
|
|
goto error;
|
|
}
|
|
/* XXX Warn if (supplied_flags & PyCF_MASK_OBSOLETE) != 0? */
|
|
|
|
if (optimize < -1 || optimize > 2) {
|
|
PyErr_SetString(PyExc_ValueError,
|
|
"compile(): invalid optimize value");
|
|
goto error;
|
|
}
|
|
|
|
if (!dont_inherit) {
|
|
PyEval_MergeCompilerFlags(&cf);
|
|
}
|
|
|
|
if (strcmp(startstr, "exec") == 0)
|
|
mode = 0;
|
|
else if (strcmp(startstr, "eval") == 0)
|
|
mode = 1;
|
|
else if (strcmp(startstr, "single") == 0)
|
|
mode = 2;
|
|
else {
|
|
PyErr_SetString(PyExc_ValueError,
|
|
"compile() arg 3 must be 'exec', 'eval' or 'single'");
|
|
goto error;
|
|
}
|
|
|
|
is_ast = PyAST_Check(cmd);
|
|
if (is_ast == -1)
|
|
goto error;
|
|
if (is_ast) {
|
|
if (supplied_flags & PyCF_ONLY_AST) {
|
|
Py_INCREF(cmd);
|
|
result = cmd;
|
|
}
|
|
else {
|
|
PyArena *arena;
|
|
mod_ty mod;
|
|
|
|
arena = PyArena_New();
|
|
if (arena == NULL)
|
|
goto error;
|
|
mod = PyAST_obj2mod(cmd, arena, mode);
|
|
if (mod == NULL) {
|
|
PyArena_Free(arena);
|
|
goto error;
|
|
}
|
|
if (!PyAST_Validate(mod)) {
|
|
PyArena_Free(arena);
|
|
goto error;
|
|
}
|
|
result = (PyObject*)PyAST_CompileObject(mod, filename,
|
|
&cf, optimize, arena);
|
|
PyArena_Free(arena);
|
|
}
|
|
goto finally;
|
|
}
|
|
|
|
str = source_as_string(cmd, "compile", "string, bytes or AST", &cf);
|
|
if (str == NULL)
|
|
goto error;
|
|
|
|
result = Py_CompileStringObject(str, filename, start[mode], &cf, optimize);
|
|
goto finally;
|
|
|
|
error:
|
|
result = NULL;
|
|
finally:
|
|
Py_DECREF(filename);
|
|
return result;
|
|
}
|
|
|
|
PyDoc_STRVAR(compile_doc,
|
|
"compile(source, filename, mode[, flags[, dont_inherit]]) -> code object\n\
|
|
\n\
|
|
Compile the source string (a Python module, statement or expression)\n\
|
|
into a code object that can be executed by exec() or eval().\n\
|
|
The filename will be used for run-time error messages.\n\
|
|
The mode must be 'exec' to compile a module, 'single' to compile a\n\
|
|
single (interactive) statement, or 'eval' to compile an expression.\n\
|
|
The flags argument, if present, controls which future statements influence\n\
|
|
the compilation of the code.\n\
|
|
The dont_inherit argument, if non-zero, stops the compilation inheriting\n\
|
|
the effects of any future statements in effect in the code calling\n\
|
|
compile; if absent or zero these statements do influence the compilation,\n\
|
|
in addition to any features explicitly specified.");
|
|
|
|
static PyObject *
|
|
builtin_dir(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *arg = NULL;
|
|
|
|
if (!PyArg_UnpackTuple(args, "dir", 0, 1, &arg))
|
|
return NULL;
|
|
return PyObject_Dir(arg);
|
|
}
|
|
|
|
PyDoc_STRVAR(dir_doc,
|
|
"dir([object]) -> list of strings\n"
|
|
"\n"
|
|
"If called without an argument, return the names in the current scope.\n"
|
|
"Else, return an alphabetized list of names comprising (some of) the attributes\n"
|
|
"of the given object, and of attributes reachable from it.\n"
|
|
"If the object supplies a method named __dir__, it will be used; otherwise\n"
|
|
"the default dir() logic is used and returns:\n"
|
|
" for a module object: the module's attributes.\n"
|
|
" for a class object: its attributes, and recursively the attributes\n"
|
|
" of its bases.\n"
|
|
" for any other object: its attributes, its class's attributes, and\n"
|
|
" recursively the attributes of its class's base classes.");
|
|
|
|
static PyObject *
|
|
builtin_divmod(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *v, *w;
|
|
|
|
if (!PyArg_UnpackTuple(args, "divmod", 2, 2, &v, &w))
|
|
return NULL;
|
|
return PyNumber_Divmod(v, w);
|
|
}
|
|
|
|
PyDoc_STRVAR(divmod_doc,
|
|
"divmod(x, y) -> (div, mod)\n\
|
|
\n\
|
|
Return the tuple ((x-x%y)/y, x%y). Invariant: div*y + mod == x.");
|
|
|
|
|
|
static PyObject *
|
|
builtin_eval(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *cmd, *result, *tmp = NULL;
|
|
PyObject *globals = Py_None, *locals = Py_None;
|
|
char *str;
|
|
PyCompilerFlags cf;
|
|
|
|
if (!PyArg_UnpackTuple(args, "eval", 1, 3, &cmd, &globals, &locals))
|
|
return NULL;
|
|
if (locals != Py_None && !PyMapping_Check(locals)) {
|
|
PyErr_SetString(PyExc_TypeError, "locals must be a mapping");
|
|
return NULL;
|
|
}
|
|
if (globals != Py_None && !PyDict_Check(globals)) {
|
|
PyErr_SetString(PyExc_TypeError, PyMapping_Check(globals) ?
|
|
"globals must be a real dict; try eval(expr, {}, mapping)"
|
|
: "globals must be a dict");
|
|
return NULL;
|
|
}
|
|
if (globals == Py_None) {
|
|
globals = PyEval_GetGlobals();
|
|
if (locals == Py_None) {
|
|
locals = PyEval_GetLocals();
|
|
if (locals == NULL)
|
|
return NULL;
|
|
}
|
|
}
|
|
else if (locals == Py_None)
|
|
locals = globals;
|
|
|
|
if (globals == NULL || locals == NULL) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"eval must be given globals and locals "
|
|
"when called without a frame");
|
|
return NULL;
|
|
}
|
|
|
|
if (_PyDict_GetItemId(globals, &PyId___builtins__) == NULL) {
|
|
if (_PyDict_SetItemId(globals, &PyId___builtins__,
|
|
PyEval_GetBuiltins()) != 0)
|
|
return NULL;
|
|
}
|
|
|
|
if (PyCode_Check(cmd)) {
|
|
if (PyCode_GetNumFree((PyCodeObject *)cmd) > 0) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"code object passed to eval() may not contain free variables");
|
|
return NULL;
|
|
}
|
|
return PyEval_EvalCode(cmd, globals, locals);
|
|
}
|
|
|
|
cf.cf_flags = PyCF_SOURCE_IS_UTF8;
|
|
str = source_as_string(cmd, "eval", "string, bytes or code", &cf);
|
|
if (str == NULL)
|
|
return NULL;
|
|
|
|
while (*str == ' ' || *str == '\t')
|
|
str++;
|
|
|
|
(void)PyEval_MergeCompilerFlags(&cf);
|
|
result = PyRun_StringFlags(str, Py_eval_input, globals, locals, &cf);
|
|
Py_XDECREF(tmp);
|
|
return result;
|
|
}
|
|
|
|
PyDoc_STRVAR(eval_doc,
|
|
"eval(source[, globals[, locals]]) -> value\n\
|
|
\n\
|
|
Evaluate the source in the context of globals and locals.\n\
|
|
The source may be a string representing a Python expression\n\
|
|
or a code object as returned by compile().\n\
|
|
The globals must be a dictionary and locals can be any mapping,\n\
|
|
defaulting to the current globals and locals.\n\
|
|
If only globals is given, locals defaults to it.\n");
|
|
|
|
static PyObject *
|
|
builtin_exec(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *v;
|
|
PyObject *prog, *globals = Py_None, *locals = Py_None;
|
|
|
|
if (!PyArg_UnpackTuple(args, "exec", 1, 3, &prog, &globals, &locals))
|
|
return NULL;
|
|
|
|
if (globals == Py_None) {
|
|
globals = PyEval_GetGlobals();
|
|
if (locals == Py_None) {
|
|
locals = PyEval_GetLocals();
|
|
if (locals == NULL)
|
|
return NULL;
|
|
}
|
|
if (!globals || !locals) {
|
|
PyErr_SetString(PyExc_SystemError,
|
|
"globals and locals cannot be NULL");
|
|
return NULL;
|
|
}
|
|
}
|
|
else if (locals == Py_None)
|
|
locals = globals;
|
|
|
|
if (!PyDict_Check(globals)) {
|
|
PyErr_Format(PyExc_TypeError, "exec() arg 2 must be a dict, not %.100s",
|
|
globals->ob_type->tp_name);
|
|
return NULL;
|
|
}
|
|
if (!PyMapping_Check(locals)) {
|
|
PyErr_Format(PyExc_TypeError,
|
|
"arg 3 must be a mapping or None, not %.100s",
|
|
locals->ob_type->tp_name);
|
|
return NULL;
|
|
}
|
|
if (_PyDict_GetItemId(globals, &PyId___builtins__) == NULL) {
|
|
if (_PyDict_SetItemId(globals, &PyId___builtins__,
|
|
PyEval_GetBuiltins()) != 0)
|
|
return NULL;
|
|
}
|
|
|
|
if (PyCode_Check(prog)) {
|
|
if (PyCode_GetNumFree((PyCodeObject *)prog) > 0) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"code object passed to exec() may not "
|
|
"contain free variables");
|
|
return NULL;
|
|
}
|
|
v = PyEval_EvalCode(prog, globals, locals);
|
|
}
|
|
else {
|
|
char *str;
|
|
PyCompilerFlags cf;
|
|
cf.cf_flags = PyCF_SOURCE_IS_UTF8;
|
|
str = source_as_string(prog, "exec",
|
|
"string, bytes or code", &cf);
|
|
if (str == NULL)
|
|
return NULL;
|
|
if (PyEval_MergeCompilerFlags(&cf))
|
|
v = PyRun_StringFlags(str, Py_file_input, globals,
|
|
locals, &cf);
|
|
else
|
|
v = PyRun_String(str, Py_file_input, globals, locals);
|
|
}
|
|
if (v == NULL)
|
|
return NULL;
|
|
Py_DECREF(v);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
PyDoc_STRVAR(exec_doc,
|
|
"exec(object[, globals[, locals]])\n\
|
|
\n\
|
|
Read and execute code from an object, which can be a string or a code\n\
|
|
object.\n\
|
|
The globals and locals are dictionaries, defaulting to the current\n\
|
|
globals and locals. If only globals is given, locals defaults to it.");
|
|
|
|
|
|
static PyObject *
|
|
builtin_getattr(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *v, *result, *dflt = NULL;
|
|
PyObject *name;
|
|
|
|
if (!PyArg_UnpackTuple(args, "getattr", 2, 3, &v, &name, &dflt))
|
|
return NULL;
|
|
|
|
if (!PyUnicode_Check(name)) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"getattr(): attribute name must be string");
|
|
return NULL;
|
|
}
|
|
result = PyObject_GetAttr(v, name);
|
|
if (result == NULL && dflt != NULL &&
|
|
PyErr_ExceptionMatches(PyExc_AttributeError))
|
|
{
|
|
PyErr_Clear();
|
|
Py_INCREF(dflt);
|
|
result = dflt;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
PyDoc_STRVAR(getattr_doc,
|
|
"getattr(object, name[, default]) -> value\n\
|
|
\n\
|
|
Get a named attribute from an object; getattr(x, 'y') is equivalent to x.y.\n\
|
|
When a default argument is given, it is returned when the attribute doesn't\n\
|
|
exist; without it, an exception is raised in that case.");
|
|
|
|
|
|
static PyObject *
|
|
builtin_globals(PyObject *self)
|
|
{
|
|
PyObject *d;
|
|
|
|
d = PyEval_GetGlobals();
|
|
Py_XINCREF(d);
|
|
return d;
|
|
}
|
|
|
|
PyDoc_STRVAR(globals_doc,
|
|
"globals() -> dictionary\n\
|
|
\n\
|
|
Return the dictionary containing the current scope's global variables.");
|
|
|
|
|
|
static PyObject *
|
|
builtin_hasattr(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *v;
|
|
PyObject *name;
|
|
|
|
if (!PyArg_UnpackTuple(args, "hasattr", 2, 2, &v, &name))
|
|
return NULL;
|
|
if (!PyUnicode_Check(name)) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"hasattr(): attribute name must be string");
|
|
return NULL;
|
|
}
|
|
v = PyObject_GetAttr(v, name);
|
|
if (v == NULL) {
|
|
if (PyErr_ExceptionMatches(PyExc_AttributeError)) {
|
|
PyErr_Clear();
|
|
Py_RETURN_FALSE;
|
|
}
|
|
return NULL;
|
|
}
|
|
Py_DECREF(v);
|
|
Py_RETURN_TRUE;
|
|
}
|
|
|
|
PyDoc_STRVAR(hasattr_doc,
|
|
"hasattr(object, name) -> bool\n\
|
|
\n\
|
|
Return whether the object has an attribute with the given name.\n\
|
|
(This is done by calling getattr(object, name) and catching AttributeError.)");
|
|
|
|
|
|
static PyObject *
|
|
builtin_id(PyObject *self, PyObject *v)
|
|
{
|
|
return PyLong_FromVoidPtr(v);
|
|
}
|
|
|
|
PyDoc_STRVAR(id_doc,
|
|
"id(object) -> integer\n\
|
|
\n\
|
|
Return the identity of an object. This is guaranteed to be unique among\n\
|
|
simultaneously existing objects. (Hint: it's the object's memory address.)");
|
|
|
|
|
|
/* map object ************************************************************/
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
PyObject *iters;
|
|
PyObject *func;
|
|
} mapobject;
|
|
|
|
static PyObject *
|
|
map_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
|
|
{
|
|
PyObject *it, *iters, *func;
|
|
mapobject *lz;
|
|
Py_ssize_t numargs, i;
|
|
|
|
if (type == &PyMap_Type && !_PyArg_NoKeywords("map()", kwds))
|
|
return NULL;
|
|
|
|
numargs = PyTuple_Size(args);
|
|
if (numargs < 2) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"map() must have at least two arguments.");
|
|
return NULL;
|
|
}
|
|
|
|
iters = PyTuple_New(numargs-1);
|
|
if (iters == NULL)
|
|
return NULL;
|
|
|
|
for (i=1 ; i<numargs ; i++) {
|
|
/* Get iterator. */
|
|
it = PyObject_GetIter(PyTuple_GET_ITEM(args, i));
|
|
if (it == NULL) {
|
|
Py_DECREF(iters);
|
|
return NULL;
|
|
}
|
|
PyTuple_SET_ITEM(iters, i-1, it);
|
|
}
|
|
|
|
/* create mapobject structure */
|
|
lz = (mapobject *)type->tp_alloc(type, 0);
|
|
if (lz == NULL) {
|
|
Py_DECREF(iters);
|
|
return NULL;
|
|
}
|
|
lz->iters = iters;
|
|
func = PyTuple_GET_ITEM(args, 0);
|
|
Py_INCREF(func);
|
|
lz->func = func;
|
|
|
|
return (PyObject *)lz;
|
|
}
|
|
|
|
static void
|
|
map_dealloc(mapobject *lz)
|
|
{
|
|
PyObject_GC_UnTrack(lz);
|
|
Py_XDECREF(lz->iters);
|
|
Py_XDECREF(lz->func);
|
|
Py_TYPE(lz)->tp_free(lz);
|
|
}
|
|
|
|
static int
|
|
map_traverse(mapobject *lz, visitproc visit, void *arg)
|
|
{
|
|
Py_VISIT(lz->iters);
|
|
Py_VISIT(lz->func);
|
|
return 0;
|
|
}
|
|
|
|
static PyObject *
|
|
map_next(mapobject *lz)
|
|
{
|
|
PyObject *val;
|
|
PyObject *argtuple;
|
|
PyObject *result;
|
|
Py_ssize_t numargs, i;
|
|
|
|
numargs = PyTuple_Size(lz->iters);
|
|
argtuple = PyTuple_New(numargs);
|
|
if (argtuple == NULL)
|
|
return NULL;
|
|
|
|
for (i=0 ; i<numargs ; i++) {
|
|
val = PyIter_Next(PyTuple_GET_ITEM(lz->iters, i));
|
|
if (val == NULL) {
|
|
Py_DECREF(argtuple);
|
|
return NULL;
|
|
}
|
|
PyTuple_SET_ITEM(argtuple, i, val);
|
|
}
|
|
result = PyObject_Call(lz->func, argtuple, NULL);
|
|
Py_DECREF(argtuple);
|
|
return result;
|
|
}
|
|
|
|
static PyObject *
|
|
map_reduce(mapobject *lz)
|
|
{
|
|
Py_ssize_t numargs = PyTuple_GET_SIZE(lz->iters);
|
|
PyObject *args = PyTuple_New(numargs+1);
|
|
Py_ssize_t i;
|
|
if (args == NULL)
|
|
return NULL;
|
|
Py_INCREF(lz->func);
|
|
PyTuple_SET_ITEM(args, 0, lz->func);
|
|
for (i = 0; i<numargs; i++){
|
|
PyObject *it = PyTuple_GET_ITEM(lz->iters, i);
|
|
Py_INCREF(it);
|
|
PyTuple_SET_ITEM(args, i+1, it);
|
|
}
|
|
|
|
return Py_BuildValue("ON", Py_TYPE(lz), args);
|
|
}
|
|
|
|
static PyMethodDef map_methods[] = {
|
|
{"__reduce__", (PyCFunction)map_reduce, METH_NOARGS, reduce_doc},
|
|
{NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
|
|
PyDoc_STRVAR(map_doc,
|
|
"map(func, *iterables) --> map object\n\
|
|
\n\
|
|
Make an iterator that computes the function using arguments from\n\
|
|
each of the iterables. Stops when the shortest iterable is exhausted.");
|
|
|
|
PyTypeObject PyMap_Type = {
|
|
PyVarObject_HEAD_INIT(&PyType_Type, 0)
|
|
"map", /* tp_name */
|
|
sizeof(mapobject), /* tp_basicsize */
|
|
0, /* tp_itemsize */
|
|
/* methods */
|
|
(destructor)map_dealloc, /* tp_dealloc */
|
|
0, /* tp_print */
|
|
0, /* tp_getattr */
|
|
0, /* tp_setattr */
|
|
0, /* tp_reserved */
|
|
0, /* tp_repr */
|
|
0, /* tp_as_number */
|
|
0, /* tp_as_sequence */
|
|
0, /* tp_as_mapping */
|
|
0, /* tp_hash */
|
|
0, /* tp_call */
|
|
0, /* tp_str */
|
|
PyObject_GenericGetAttr, /* tp_getattro */
|
|
0, /* tp_setattro */
|
|
0, /* tp_as_buffer */
|
|
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC |
|
|
Py_TPFLAGS_BASETYPE, /* tp_flags */
|
|
map_doc, /* tp_doc */
|
|
(traverseproc)map_traverse, /* tp_traverse */
|
|
0, /* tp_clear */
|
|
0, /* tp_richcompare */
|
|
0, /* tp_weaklistoffset */
|
|
PyObject_SelfIter, /* tp_iter */
|
|
(iternextfunc)map_next, /* tp_iternext */
|
|
map_methods, /* tp_methods */
|
|
0, /* tp_members */
|
|
0, /* tp_getset */
|
|
0, /* tp_base */
|
|
0, /* tp_dict */
|
|
0, /* tp_descr_get */
|
|
0, /* tp_descr_set */
|
|
0, /* tp_dictoffset */
|
|
0, /* tp_init */
|
|
PyType_GenericAlloc, /* tp_alloc */
|
|
map_new, /* tp_new */
|
|
PyObject_GC_Del, /* tp_free */
|
|
};
|
|
|
|
static PyObject *
|
|
builtin_next(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *it, *res;
|
|
PyObject *def = NULL;
|
|
|
|
if (!PyArg_UnpackTuple(args, "next", 1, 2, &it, &def))
|
|
return NULL;
|
|
if (!PyIter_Check(it)) {
|
|
PyErr_Format(PyExc_TypeError,
|
|
"'%.200s' object is not an iterator",
|
|
it->ob_type->tp_name);
|
|
return NULL;
|
|
}
|
|
|
|
res = (*it->ob_type->tp_iternext)(it);
|
|
if (res != NULL) {
|
|
return res;
|
|
} else if (def != NULL) {
|
|
if (PyErr_Occurred()) {
|
|
if(!PyErr_ExceptionMatches(PyExc_StopIteration))
|
|
return NULL;
|
|
PyErr_Clear();
|
|
}
|
|
Py_INCREF(def);
|
|
return def;
|
|
} else if (PyErr_Occurred()) {
|
|
return NULL;
|
|
} else {
|
|
PyErr_SetNone(PyExc_StopIteration);
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
PyDoc_STRVAR(next_doc,
|
|
"next(iterator[, default])\n\
|
|
\n\
|
|
Return the next item from the iterator. If default is given and the iterator\n\
|
|
is exhausted, it is returned instead of raising StopIteration.");
|
|
|
|
|
|
static PyObject *
|
|
builtin_setattr(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *v;
|
|
PyObject *name;
|
|
PyObject *value;
|
|
|
|
if (!PyArg_UnpackTuple(args, "setattr", 3, 3, &v, &name, &value))
|
|
return NULL;
|
|
if (PyObject_SetAttr(v, name, value) != 0)
|
|
return NULL;
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
PyDoc_STRVAR(setattr_doc,
|
|
"setattr(object, name, value)\n\
|
|
\n\
|
|
Set a named attribute on an object; setattr(x, 'y', v) is equivalent to\n\
|
|
``x.y = v''.");
|
|
|
|
|
|
static PyObject *
|
|
builtin_delattr(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *v;
|
|
PyObject *name;
|
|
|
|
if (!PyArg_UnpackTuple(args, "delattr", 2, 2, &v, &name))
|
|
return NULL;
|
|
if (PyObject_SetAttr(v, name, (PyObject *)NULL) != 0)
|
|
return NULL;
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
PyDoc_STRVAR(delattr_doc,
|
|
"delattr(object, name)\n\
|
|
\n\
|
|
Delete a named attribute on an object; delattr(x, 'y') is equivalent to\n\
|
|
``del x.y''.");
|
|
|
|
|
|
static PyObject *
|
|
builtin_hash(PyObject *self, PyObject *v)
|
|
{
|
|
Py_hash_t x;
|
|
|
|
x = PyObject_Hash(v);
|
|
if (x == -1)
|
|
return NULL;
|
|
return PyLong_FromSsize_t(x);
|
|
}
|
|
|
|
PyDoc_STRVAR(hash_doc,
|
|
"hash(object) -> integer\n\
|
|
\n\
|
|
Return a hash value for the object. Two objects with the same value have\n\
|
|
the same hash value. The reverse is not necessarily true, but likely.");
|
|
|
|
|
|
static PyObject *
|
|
builtin_hex(PyObject *self, PyObject *v)
|
|
{
|
|
return PyNumber_ToBase(v, 16);
|
|
}
|
|
|
|
PyDoc_STRVAR(hex_doc,
|
|
"hex(number) -> string\n\
|
|
\n\
|
|
Return the hexadecimal representation of an integer.\n\
|
|
\n\
|
|
>>> hex(3735928559)\n\
|
|
'0xdeadbeef'\n\
|
|
");
|
|
|
|
|
|
static PyObject *
|
|
builtin_iter(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *v, *w = NULL;
|
|
|
|
if (!PyArg_UnpackTuple(args, "iter", 1, 2, &v, &w))
|
|
return NULL;
|
|
if (w == NULL)
|
|
return PyObject_GetIter(v);
|
|
if (!PyCallable_Check(v)) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"iter(v, w): v must be callable");
|
|
return NULL;
|
|
}
|
|
return PyCallIter_New(v, w);
|
|
}
|
|
|
|
PyDoc_STRVAR(iter_doc,
|
|
"iter(iterable) -> iterator\n\
|
|
iter(callable, sentinel) -> iterator\n\
|
|
\n\
|
|
Get an iterator from an object. In the first form, the argument must\n\
|
|
supply its own iterator, or be a sequence.\n\
|
|
In the second form, the callable is called until it returns the sentinel.");
|
|
|
|
|
|
static PyObject *
|
|
builtin_len(PyObject *self, PyObject *v)
|
|
{
|
|
Py_ssize_t res;
|
|
|
|
res = PyObject_Size(v);
|
|
if (res < 0 && PyErr_Occurred())
|
|
return NULL;
|
|
return PyLong_FromSsize_t(res);
|
|
}
|
|
|
|
PyDoc_STRVAR(len_doc,
|
|
"len(object) -> integer\n\
|
|
\n\
|
|
Return the number of items of a sequence or mapping.");
|
|
|
|
|
|
static PyObject *
|
|
builtin_locals(PyObject *self)
|
|
{
|
|
PyObject *d;
|
|
|
|
d = PyEval_GetLocals();
|
|
Py_XINCREF(d);
|
|
return d;
|
|
}
|
|
|
|
PyDoc_STRVAR(locals_doc,
|
|
"locals() -> dictionary\n\
|
|
\n\
|
|
Update and return a dictionary containing the current scope's local variables.");
|
|
|
|
|
|
static PyObject *
|
|
min_max(PyObject *args, PyObject *kwds, int op)
|
|
{
|
|
PyObject *v, *it, *item, *val, *maxitem, *maxval, *keyfunc=NULL;
|
|
PyObject *emptytuple, *defaultval = NULL;
|
|
static char *kwlist[] = {"key", "default", NULL};
|
|
const char *name = op == Py_LT ? "min" : "max";
|
|
const int positional = PyTuple_Size(args) > 1;
|
|
int ret;
|
|
|
|
if (positional)
|
|
v = args;
|
|
else if (!PyArg_UnpackTuple(args, name, 1, 1, &v))
|
|
return NULL;
|
|
|
|
emptytuple = PyTuple_New(0);
|
|
if (emptytuple == NULL)
|
|
return NULL;
|
|
ret = PyArg_ParseTupleAndKeywords(emptytuple, kwds, "|$OO", kwlist,
|
|
&keyfunc, &defaultval);
|
|
Py_DECREF(emptytuple);
|
|
if (!ret)
|
|
return NULL;
|
|
|
|
if (positional && defaultval != NULL) {
|
|
PyErr_Format(PyExc_TypeError,
|
|
"Cannot specify a default for %s() with multiple "
|
|
"positional arguments", name);
|
|
return NULL;
|
|
}
|
|
|
|
it = PyObject_GetIter(v);
|
|
if (it == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
maxitem = NULL; /* the result */
|
|
maxval = NULL; /* the value associated with the result */
|
|
while (( item = PyIter_Next(it) )) {
|
|
/* get the value from the key function */
|
|
if (keyfunc != NULL) {
|
|
val = PyObject_CallFunctionObjArgs(keyfunc, item, NULL);
|
|
if (val == NULL)
|
|
goto Fail_it_item;
|
|
}
|
|
/* no key function; the value is the item */
|
|
else {
|
|
val = item;
|
|
Py_INCREF(val);
|
|
}
|
|
|
|
/* maximum value and item are unset; set them */
|
|
if (maxval == NULL) {
|
|
maxitem = item;
|
|
maxval = val;
|
|
}
|
|
/* maximum value and item are set; update them as necessary */
|
|
else {
|
|
int cmp = PyObject_RichCompareBool(val, maxval, op);
|
|
if (cmp < 0)
|
|
goto Fail_it_item_and_val;
|
|
else if (cmp > 0) {
|
|
Py_DECREF(maxval);
|
|
Py_DECREF(maxitem);
|
|
maxval = val;
|
|
maxitem = item;
|
|
}
|
|
else {
|
|
Py_DECREF(item);
|
|
Py_DECREF(val);
|
|
}
|
|
}
|
|
}
|
|
if (PyErr_Occurred())
|
|
goto Fail_it;
|
|
if (maxval == NULL) {
|
|
assert(maxitem == NULL);
|
|
if (defaultval != NULL) {
|
|
Py_INCREF(defaultval);
|
|
maxitem = defaultval;
|
|
} else {
|
|
PyErr_Format(PyExc_ValueError,
|
|
"%s() arg is an empty sequence", name);
|
|
}
|
|
}
|
|
else
|
|
Py_DECREF(maxval);
|
|
Py_DECREF(it);
|
|
return maxitem;
|
|
|
|
Fail_it_item_and_val:
|
|
Py_DECREF(val);
|
|
Fail_it_item:
|
|
Py_DECREF(item);
|
|
Fail_it:
|
|
Py_XDECREF(maxval);
|
|
Py_XDECREF(maxitem);
|
|
Py_DECREF(it);
|
|
return NULL;
|
|
}
|
|
|
|
static PyObject *
|
|
builtin_min(PyObject *self, PyObject *args, PyObject *kwds)
|
|
{
|
|
return min_max(args, kwds, Py_LT);
|
|
}
|
|
|
|
PyDoc_STRVAR(min_doc,
|
|
"min(iterable[, key=func]) -> value\n\
|
|
min(a, b, c, ...[, key=func]) -> value\n\
|
|
\n\
|
|
With a single iterable argument, return its smallest item.\n\
|
|
With two or more arguments, return the smallest argument.");
|
|
|
|
|
|
static PyObject *
|
|
builtin_max(PyObject *self, PyObject *args, PyObject *kwds)
|
|
{
|
|
return min_max(args, kwds, Py_GT);
|
|
}
|
|
|
|
PyDoc_STRVAR(max_doc,
|
|
"max(iterable[, key=func]) -> value\n\
|
|
max(a, b, c, ...[, key=func]) -> value\n\
|
|
\n\
|
|
With a single iterable argument, return its largest item.\n\
|
|
With two or more arguments, return the largest argument.");
|
|
|
|
|
|
static PyObject *
|
|
builtin_oct(PyObject *self, PyObject *v)
|
|
{
|
|
return PyNumber_ToBase(v, 8);
|
|
}
|
|
|
|
PyDoc_STRVAR(oct_doc,
|
|
"oct(number) -> string\n\
|
|
\n\
|
|
Return the octal representation of an integer.\n\
|
|
\n\
|
|
>>> oct(342391)\n\
|
|
'0o1234567'\n\
|
|
");
|
|
|
|
|
|
static PyObject *
|
|
builtin_ord(PyObject *self, PyObject* obj)
|
|
{
|
|
long ord;
|
|
Py_ssize_t size;
|
|
|
|
if (PyBytes_Check(obj)) {
|
|
size = PyBytes_GET_SIZE(obj);
|
|
if (size == 1) {
|
|
ord = (long)((unsigned char)*PyBytes_AS_STRING(obj));
|
|
return PyLong_FromLong(ord);
|
|
}
|
|
}
|
|
else if (PyUnicode_Check(obj)) {
|
|
if (PyUnicode_READY(obj) == -1)
|
|
return NULL;
|
|
size = PyUnicode_GET_LENGTH(obj);
|
|
if (size == 1) {
|
|
ord = (long)PyUnicode_READ_CHAR(obj, 0);
|
|
return PyLong_FromLong(ord);
|
|
}
|
|
}
|
|
else if (PyByteArray_Check(obj)) {
|
|
/* XXX Hopefully this is temporary */
|
|
size = PyByteArray_GET_SIZE(obj);
|
|
if (size == 1) {
|
|
ord = (long)((unsigned char)*PyByteArray_AS_STRING(obj));
|
|
return PyLong_FromLong(ord);
|
|
}
|
|
}
|
|
else {
|
|
PyErr_Format(PyExc_TypeError,
|
|
"ord() expected string of length 1, but " \
|
|
"%.200s found", obj->ob_type->tp_name);
|
|
return NULL;
|
|
}
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
"ord() expected a character, "
|
|
"but string of length %zd found",
|
|
size);
|
|
return NULL;
|
|
}
|
|
|
|
PyDoc_VAR(ord_doc) = PyDoc_STR(
|
|
"ord(c) -> integer\n\
|
|
\n\
|
|
Return the integer ordinal of a one-character string."
|
|
);
|
|
|
|
|
|
static PyObject *
|
|
builtin_pow(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *v, *w, *z = Py_None;
|
|
|
|
if (!PyArg_UnpackTuple(args, "pow", 2, 3, &v, &w, &z))
|
|
return NULL;
|
|
return PyNumber_Power(v, w, z);
|
|
}
|
|
|
|
PyDoc_STRVAR(pow_doc,
|
|
"pow(x, y[, z]) -> number\n\
|
|
\n\
|
|
With two arguments, equivalent to x**y. With three arguments,\n\
|
|
equivalent to (x**y) % z, but may be more efficient (e.g. for ints).");
|
|
|
|
|
|
|
|
static PyObject *
|
|
builtin_print(PyObject *self, PyObject *args, PyObject *kwds)
|
|
{
|
|
static char *kwlist[] = {"sep", "end", "file", "flush", 0};
|
|
static PyObject *dummy_args;
|
|
PyObject *sep = NULL, *end = NULL, *file = NULL, *flush = NULL;
|
|
int i, err;
|
|
|
|
if (dummy_args == NULL && !(dummy_args = PyTuple_New(0)))
|
|
return NULL;
|
|
if (!PyArg_ParseTupleAndKeywords(dummy_args, kwds, "|OOOO:print",
|
|
kwlist, &sep, &end, &file, &flush))
|
|
return NULL;
|
|
if (file == NULL || file == Py_None) {
|
|
file = _PySys_GetObjectId(&PyId_stdout);
|
|
if (file == NULL) {
|
|
PyErr_SetString(PyExc_RuntimeError, "lost sys.stdout");
|
|
return NULL;
|
|
}
|
|
|
|
/* sys.stdout may be None when FILE* stdout isn't connected */
|
|
if (file == Py_None)
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
if (sep == Py_None) {
|
|
sep = NULL;
|
|
}
|
|
else if (sep && !PyUnicode_Check(sep)) {
|
|
PyErr_Format(PyExc_TypeError,
|
|
"sep must be None or a string, not %.200s",
|
|
sep->ob_type->tp_name);
|
|
return NULL;
|
|
}
|
|
if (end == Py_None) {
|
|
end = NULL;
|
|
}
|
|
else if (end && !PyUnicode_Check(end)) {
|
|
PyErr_Format(PyExc_TypeError,
|
|
"end must be None or a string, not %.200s",
|
|
end->ob_type->tp_name);
|
|
return NULL;
|
|
}
|
|
|
|
for (i = 0; i < PyTuple_Size(args); i++) {
|
|
if (i > 0) {
|
|
if (sep == NULL)
|
|
err = PyFile_WriteString(" ", file);
|
|
else
|
|
err = PyFile_WriteObject(sep, file,
|
|
Py_PRINT_RAW);
|
|
if (err)
|
|
return NULL;
|
|
}
|
|
err = PyFile_WriteObject(PyTuple_GetItem(args, i), file,
|
|
Py_PRINT_RAW);
|
|
if (err)
|
|
return NULL;
|
|
}
|
|
|
|
if (end == NULL)
|
|
err = PyFile_WriteString("\n", file);
|
|
else
|
|
err = PyFile_WriteObject(end, file, Py_PRINT_RAW);
|
|
if (err)
|
|
return NULL;
|
|
|
|
if (flush != NULL) {
|
|
PyObject *tmp;
|
|
int do_flush = PyObject_IsTrue(flush);
|
|
if (do_flush == -1)
|
|
return NULL;
|
|
else if (do_flush) {
|
|
tmp = _PyObject_CallMethodId(file, &PyId_flush, "");
|
|
if (tmp == NULL)
|
|
return NULL;
|
|
else
|
|
Py_DECREF(tmp);
|
|
}
|
|
}
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
PyDoc_STRVAR(print_doc,
|
|
"print(value, ..., sep=' ', end='\\n', file=sys.stdout, flush=False)\n\
|
|
\n\
|
|
Prints the values to a stream, or to sys.stdout by default.\n\
|
|
Optional keyword arguments:\n\
|
|
file: a file-like object (stream); defaults to the current sys.stdout.\n\
|
|
sep: string inserted between values, default a space.\n\
|
|
end: string appended after the last value, default a newline.\n\
|
|
flush: whether to forcibly flush the stream.");
|
|
|
|
|
|
static PyObject *
|
|
builtin_input(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *promptarg = NULL;
|
|
PyObject *fin = _PySys_GetObjectId(&PyId_stdin);
|
|
PyObject *fout = _PySys_GetObjectId(&PyId_stdout);
|
|
PyObject *ferr = _PySys_GetObjectId(&PyId_stderr);
|
|
PyObject *tmp;
|
|
long fd;
|
|
int tty;
|
|
|
|
/* Parse arguments */
|
|
if (!PyArg_UnpackTuple(args, "input", 0, 1, &promptarg))
|
|
return NULL;
|
|
|
|
/* Check that stdin/out/err are intact */
|
|
if (fin == NULL || fin == Py_None) {
|
|
PyErr_SetString(PyExc_RuntimeError,
|
|
"input(): lost sys.stdin");
|
|
return NULL;
|
|
}
|
|
if (fout == NULL || fout == Py_None) {
|
|
PyErr_SetString(PyExc_RuntimeError,
|
|
"input(): lost sys.stdout");
|
|
return NULL;
|
|
}
|
|
if (ferr == NULL || ferr == Py_None) {
|
|
PyErr_SetString(PyExc_RuntimeError,
|
|
"input(): lost sys.stderr");
|
|
return NULL;
|
|
}
|
|
|
|
/* First of all, flush stderr */
|
|
tmp = _PyObject_CallMethodId(ferr, &PyId_flush, "");
|
|
if (tmp == NULL)
|
|
PyErr_Clear();
|
|
else
|
|
Py_DECREF(tmp);
|
|
|
|
/* We should only use (GNU) readline if Python's sys.stdin and
|
|
sys.stdout are the same as C's stdin and stdout, because we
|
|
need to pass it those. */
|
|
tmp = _PyObject_CallMethodId(fin, &PyId_fileno, "");
|
|
if (tmp == NULL) {
|
|
PyErr_Clear();
|
|
tty = 0;
|
|
}
|
|
else {
|
|
fd = PyLong_AsLong(tmp);
|
|
Py_DECREF(tmp);
|
|
if (fd < 0 && PyErr_Occurred())
|
|
return NULL;
|
|
tty = fd == fileno(stdin) && isatty(fd);
|
|
}
|
|
if (tty) {
|
|
tmp = _PyObject_CallMethodId(fout, &PyId_fileno, "");
|
|
if (tmp == NULL)
|
|
PyErr_Clear();
|
|
else {
|
|
fd = PyLong_AsLong(tmp);
|
|
Py_DECREF(tmp);
|
|
if (fd < 0 && PyErr_Occurred())
|
|
return NULL;
|
|
tty = fd == fileno(stdout) && isatty(fd);
|
|
}
|
|
}
|
|
|
|
/* If we're interactive, use (GNU) readline */
|
|
if (tty) {
|
|
PyObject *po = NULL;
|
|
char *prompt;
|
|
char *s = NULL;
|
|
PyObject *stdin_encoding = NULL, *stdin_errors = NULL;
|
|
PyObject *stdout_encoding = NULL, *stdout_errors = NULL;
|
|
char *stdin_encoding_str, *stdin_errors_str;
|
|
PyObject *result;
|
|
size_t len;
|
|
|
|
stdin_encoding = _PyObject_GetAttrId(fin, &PyId_encoding);
|
|
stdin_errors = _PyObject_GetAttrId(fin, &PyId_errors);
|
|
if (!stdin_encoding || !stdin_errors)
|
|
/* stdin is a text stream, so it must have an
|
|
encoding. */
|
|
goto _readline_errors;
|
|
stdin_encoding_str = _PyUnicode_AsString(stdin_encoding);
|
|
stdin_errors_str = _PyUnicode_AsString(stdin_errors);
|
|
if (!stdin_encoding_str || !stdin_errors_str)
|
|
goto _readline_errors;
|
|
tmp = _PyObject_CallMethodId(fout, &PyId_flush, "");
|
|
if (tmp == NULL)
|
|
PyErr_Clear();
|
|
else
|
|
Py_DECREF(tmp);
|
|
if (promptarg != NULL) {
|
|
/* We have a prompt, encode it as stdout would */
|
|
char *stdout_encoding_str, *stdout_errors_str;
|
|
PyObject *stringpo;
|
|
stdout_encoding = _PyObject_GetAttrId(fout, &PyId_encoding);
|
|
stdout_errors = _PyObject_GetAttrId(fout, &PyId_errors);
|
|
if (!stdout_encoding || !stdout_errors)
|
|
goto _readline_errors;
|
|
stdout_encoding_str = _PyUnicode_AsString(stdout_encoding);
|
|
stdout_errors_str = _PyUnicode_AsString(stdout_errors);
|
|
if (!stdout_encoding_str || !stdout_errors_str)
|
|
goto _readline_errors;
|
|
stringpo = PyObject_Str(promptarg);
|
|
if (stringpo == NULL)
|
|
goto _readline_errors;
|
|
po = PyUnicode_AsEncodedString(stringpo,
|
|
stdout_encoding_str, stdout_errors_str);
|
|
Py_CLEAR(stdout_encoding);
|
|
Py_CLEAR(stdout_errors);
|
|
Py_CLEAR(stringpo);
|
|
if (po == NULL)
|
|
goto _readline_errors;
|
|
prompt = PyBytes_AsString(po);
|
|
if (prompt == NULL)
|
|
goto _readline_errors;
|
|
}
|
|
else {
|
|
po = NULL;
|
|
prompt = "";
|
|
}
|
|
s = PyOS_Readline(stdin, stdout, prompt);
|
|
if (s == NULL) {
|
|
PyErr_CheckSignals();
|
|
if (!PyErr_Occurred())
|
|
PyErr_SetNone(PyExc_KeyboardInterrupt);
|
|
goto _readline_errors;
|
|
}
|
|
|
|
len = strlen(s);
|
|
if (len == 0) {
|
|
PyErr_SetNone(PyExc_EOFError);
|
|
result = NULL;
|
|
}
|
|
else {
|
|
if (len > PY_SSIZE_T_MAX) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"input: input too long");
|
|
result = NULL;
|
|
}
|
|
else {
|
|
len--; /* strip trailing '\n' */
|
|
if (len != 0 && s[len-1] == '\r')
|
|
len--; /* strip trailing '\r' */
|
|
result = PyUnicode_Decode(s, len, stdin_encoding_str,
|
|
stdin_errors_str);
|
|
}
|
|
}
|
|
Py_DECREF(stdin_encoding);
|
|
Py_DECREF(stdin_errors);
|
|
Py_XDECREF(po);
|
|
PyMem_FREE(s);
|
|
return result;
|
|
_readline_errors:
|
|
Py_XDECREF(stdin_encoding);
|
|
Py_XDECREF(stdout_encoding);
|
|
Py_XDECREF(stdin_errors);
|
|
Py_XDECREF(stdout_errors);
|
|
Py_XDECREF(po);
|
|
return NULL;
|
|
}
|
|
|
|
/* Fallback if we're not interactive */
|
|
if (promptarg != NULL) {
|
|
if (PyFile_WriteObject(promptarg, fout, Py_PRINT_RAW) != 0)
|
|
return NULL;
|
|
}
|
|
tmp = _PyObject_CallMethodId(fout, &PyId_flush, "");
|
|
if (tmp == NULL)
|
|
PyErr_Clear();
|
|
else
|
|
Py_DECREF(tmp);
|
|
return PyFile_GetLine(fin, -1);
|
|
}
|
|
|
|
PyDoc_STRVAR(input_doc,
|
|
"input([prompt]) -> string\n\
|
|
\n\
|
|
Read a string from standard input. The trailing newline is stripped.\n\
|
|
If the user hits EOF (Unix: Ctl-D, Windows: Ctl-Z+Return), raise EOFError.\n\
|
|
On Unix, GNU readline is used if enabled. The prompt string, if given,\n\
|
|
is printed without a trailing newline before reading.");
|
|
|
|
|
|
static PyObject *
|
|
builtin_repr(PyObject *self, PyObject *v)
|
|
{
|
|
return PyObject_Repr(v);
|
|
}
|
|
|
|
PyDoc_STRVAR(repr_doc,
|
|
"repr(object) -> string\n\
|
|
\n\
|
|
Return the canonical string representation of the object.\n\
|
|
For most object types, eval(repr(object)) == object.");
|
|
|
|
|
|
static PyObject *
|
|
builtin_round(PyObject *self, PyObject *args, PyObject *kwds)
|
|
{
|
|
PyObject *ndigits = NULL;
|
|
static char *kwlist[] = {"number", "ndigits", 0};
|
|
PyObject *number, *round, *result;
|
|
|
|
if (!PyArg_ParseTupleAndKeywords(args, kwds, "O|O:round",
|
|
kwlist, &number, &ndigits))
|
|
return NULL;
|
|
|
|
if (Py_TYPE(number)->tp_dict == NULL) {
|
|
if (PyType_Ready(Py_TYPE(number)) < 0)
|
|
return NULL;
|
|
}
|
|
|
|
round = _PyObject_LookupSpecial(number, &PyId___round__);
|
|
if (round == NULL) {
|
|
if (!PyErr_Occurred())
|
|
PyErr_Format(PyExc_TypeError,
|
|
"type %.100s doesn't define __round__ method",
|
|
Py_TYPE(number)->tp_name);
|
|
return NULL;
|
|
}
|
|
|
|
if (ndigits == NULL)
|
|
result = PyObject_CallFunctionObjArgs(round, NULL);
|
|
else
|
|
result = PyObject_CallFunctionObjArgs(round, ndigits, NULL);
|
|
Py_DECREF(round);
|
|
return result;
|
|
}
|
|
|
|
PyDoc_STRVAR(round_doc,
|
|
"round(number[, ndigits]) -> number\n\
|
|
\n\
|
|
Round a number to a given precision in decimal digits (default 0 digits).\n\
|
|
This returns an int when called with one argument, otherwise the\n\
|
|
same type as the number. ndigits may be negative.");
|
|
|
|
|
|
static PyObject *
|
|
builtin_sorted(PyObject *self, PyObject *args, PyObject *kwds)
|
|
{
|
|
PyObject *newlist, *v, *seq, *keyfunc=NULL, *newargs;
|
|
PyObject *callable;
|
|
static char *kwlist[] = {"iterable", "key", "reverse", 0};
|
|
int reverse;
|
|
|
|
/* args 1-3 should match listsort in Objects/listobject.c */
|
|
if (!PyArg_ParseTupleAndKeywords(args, kwds, "O|Oi:sorted",
|
|
kwlist, &seq, &keyfunc, &reverse))
|
|
return NULL;
|
|
|
|
newlist = PySequence_List(seq);
|
|
if (newlist == NULL)
|
|
return NULL;
|
|
|
|
callable = _PyObject_GetAttrId(newlist, &PyId_sort);
|
|
if (callable == NULL) {
|
|
Py_DECREF(newlist);
|
|
return NULL;
|
|
}
|
|
|
|
newargs = PyTuple_GetSlice(args, 1, 4);
|
|
if (newargs == NULL) {
|
|
Py_DECREF(newlist);
|
|
Py_DECREF(callable);
|
|
return NULL;
|
|
}
|
|
|
|
v = PyObject_Call(callable, newargs, kwds);
|
|
Py_DECREF(newargs);
|
|
Py_DECREF(callable);
|
|
if (v == NULL) {
|
|
Py_DECREF(newlist);
|
|
return NULL;
|
|
}
|
|
Py_DECREF(v);
|
|
return newlist;
|
|
}
|
|
|
|
PyDoc_STRVAR(sorted_doc,
|
|
"sorted(iterable, key=None, reverse=False) --> new sorted list");
|
|
|
|
static PyObject *
|
|
builtin_vars(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *v = NULL;
|
|
PyObject *d;
|
|
|
|
if (!PyArg_UnpackTuple(args, "vars", 0, 1, &v))
|
|
return NULL;
|
|
if (v == NULL) {
|
|
d = PyEval_GetLocals();
|
|
if (d == NULL)
|
|
return NULL;
|
|
Py_INCREF(d);
|
|
}
|
|
else {
|
|
d = _PyObject_GetAttrId(v, &PyId___dict__);
|
|
if (d == NULL) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"vars() argument must have __dict__ attribute");
|
|
return NULL;
|
|
}
|
|
}
|
|
return d;
|
|
}
|
|
|
|
PyDoc_STRVAR(vars_doc,
|
|
"vars([object]) -> dictionary\n\
|
|
\n\
|
|
Without arguments, equivalent to locals().\n\
|
|
With an argument, equivalent to object.__dict__.");
|
|
|
|
static PyObject*
|
|
builtin_sum(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *seq;
|
|
PyObject *result = NULL;
|
|
PyObject *temp, *item, *iter;
|
|
|
|
if (!PyArg_UnpackTuple(args, "sum", 1, 2, &seq, &result))
|
|
return NULL;
|
|
|
|
iter = PyObject_GetIter(seq);
|
|
if (iter == NULL)
|
|
return NULL;
|
|
|
|
if (result == NULL) {
|
|
result = PyLong_FromLong(0);
|
|
if (result == NULL) {
|
|
Py_DECREF(iter);
|
|
return NULL;
|
|
}
|
|
} else {
|
|
/* reject string values for 'start' parameter */
|
|
if (PyUnicode_Check(result)) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"sum() can't sum strings [use ''.join(seq) instead]");
|
|
Py_DECREF(iter);
|
|
return NULL;
|
|
}
|
|
if (PyBytes_Check(result)) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"sum() can't sum bytes [use b''.join(seq) instead]");
|
|
Py_DECREF(iter);
|
|
return NULL;
|
|
}
|
|
if (PyByteArray_Check(result)) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"sum() can't sum bytearray [use b''.join(seq) instead]");
|
|
Py_DECREF(iter);
|
|
return NULL;
|
|
}
|
|
|
|
Py_INCREF(result);
|
|
}
|
|
|
|
#ifndef SLOW_SUM
|
|
/* Fast addition by keeping temporary sums in C instead of new Python objects.
|
|
Assumes all inputs are the same type. If the assumption fails, default
|
|
to the more general routine.
|
|
*/
|
|
if (PyLong_CheckExact(result)) {
|
|
int overflow;
|
|
long i_result = PyLong_AsLongAndOverflow(result, &overflow);
|
|
/* If this already overflowed, don't even enter the loop. */
|
|
if (overflow == 0) {
|
|
Py_DECREF(result);
|
|
result = NULL;
|
|
}
|
|
while(result == NULL) {
|
|
item = PyIter_Next(iter);
|
|
if (item == NULL) {
|
|
Py_DECREF(iter);
|
|
if (PyErr_Occurred())
|
|
return NULL;
|
|
return PyLong_FromLong(i_result);
|
|
}
|
|
if (PyLong_CheckExact(item)) {
|
|
long b = PyLong_AsLongAndOverflow(item, &overflow);
|
|
long x = i_result + b;
|
|
if (overflow == 0 && ((x^i_result) >= 0 || (x^b) >= 0)) {
|
|
i_result = x;
|
|
Py_DECREF(item);
|
|
continue;
|
|
}
|
|
}
|
|
/* Either overflowed or is not an int. Restore real objects and process normally */
|
|
result = PyLong_FromLong(i_result);
|
|
if (result == NULL) {
|
|
Py_DECREF(item);
|
|
Py_DECREF(iter);
|
|
return NULL;
|
|
}
|
|
temp = PyNumber_Add(result, item);
|
|
Py_DECREF(result);
|
|
Py_DECREF(item);
|
|
result = temp;
|
|
if (result == NULL) {
|
|
Py_DECREF(iter);
|
|
return NULL;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (PyFloat_CheckExact(result)) {
|
|
double f_result = PyFloat_AS_DOUBLE(result);
|
|
Py_DECREF(result);
|
|
result = NULL;
|
|
while(result == NULL) {
|
|
item = PyIter_Next(iter);
|
|
if (item == NULL) {
|
|
Py_DECREF(iter);
|
|
if (PyErr_Occurred())
|
|
return NULL;
|
|
return PyFloat_FromDouble(f_result);
|
|
}
|
|
if (PyFloat_CheckExact(item)) {
|
|
PyFPE_START_PROTECT("add", Py_DECREF(item); Py_DECREF(iter); return 0)
|
|
f_result += PyFloat_AS_DOUBLE(item);
|
|
PyFPE_END_PROTECT(f_result)
|
|
Py_DECREF(item);
|
|
continue;
|
|
}
|
|
if (PyLong_CheckExact(item)) {
|
|
long value;
|
|
int overflow;
|
|
value = PyLong_AsLongAndOverflow(item, &overflow);
|
|
if (!overflow) {
|
|
PyFPE_START_PROTECT("add", Py_DECREF(item); Py_DECREF(iter); return 0)
|
|
f_result += (double)value;
|
|
PyFPE_END_PROTECT(f_result)
|
|
Py_DECREF(item);
|
|
continue;
|
|
}
|
|
}
|
|
result = PyFloat_FromDouble(f_result);
|
|
temp = PyNumber_Add(result, item);
|
|
Py_DECREF(result);
|
|
Py_DECREF(item);
|
|
result = temp;
|
|
if (result == NULL) {
|
|
Py_DECREF(iter);
|
|
return NULL;
|
|
}
|
|
}
|
|
}
|
|
#endif
|
|
|
|
for(;;) {
|
|
item = PyIter_Next(iter);
|
|
if (item == NULL) {
|
|
/* error, or end-of-sequence */
|
|
if (PyErr_Occurred()) {
|
|
Py_DECREF(result);
|
|
result = NULL;
|
|
}
|
|
break;
|
|
}
|
|
/* It's tempting to use PyNumber_InPlaceAdd instead of
|
|
PyNumber_Add here, to avoid quadratic running time
|
|
when doing 'sum(list_of_lists, [])'. However, this
|
|
would produce a change in behaviour: a snippet like
|
|
|
|
empty = []
|
|
sum([[x] for x in range(10)], empty)
|
|
|
|
would change the value of empty. */
|
|
temp = PyNumber_Add(result, item);
|
|
Py_DECREF(result);
|
|
Py_DECREF(item);
|
|
result = temp;
|
|
if (result == NULL)
|
|
break;
|
|
}
|
|
Py_DECREF(iter);
|
|
return result;
|
|
}
|
|
|
|
PyDoc_STRVAR(sum_doc,
|
|
"sum(iterable[, start]) -> value\n\
|
|
\n\
|
|
Return the sum of an iterable of numbers (NOT strings) plus the value\n\
|
|
of parameter 'start' (which defaults to 0). When the iterable is\n\
|
|
empty, return start.");
|
|
|
|
|
|
static PyObject *
|
|
builtin_isinstance(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *inst;
|
|
PyObject *cls;
|
|
int retval;
|
|
|
|
if (!PyArg_UnpackTuple(args, "isinstance", 2, 2, &inst, &cls))
|
|
return NULL;
|
|
|
|
retval = PyObject_IsInstance(inst, cls);
|
|
if (retval < 0)
|
|
return NULL;
|
|
return PyBool_FromLong(retval);
|
|
}
|
|
|
|
PyDoc_STRVAR(isinstance_doc,
|
|
"isinstance(object, class-or-type-or-tuple) -> bool\n\
|
|
\n\
|
|
Return whether an object is an instance of a class or of a subclass thereof.\n\
|
|
With a type as second argument, return whether that is the object's type.\n\
|
|
The form using a tuple, isinstance(x, (A, B, ...)), is a shortcut for\n\
|
|
isinstance(x, A) or isinstance(x, B) or ... (etc.).");
|
|
|
|
|
|
static PyObject *
|
|
builtin_issubclass(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *derived;
|
|
PyObject *cls;
|
|
int retval;
|
|
|
|
if (!PyArg_UnpackTuple(args, "issubclass", 2, 2, &derived, &cls))
|
|
return NULL;
|
|
|
|
retval = PyObject_IsSubclass(derived, cls);
|
|
if (retval < 0)
|
|
return NULL;
|
|
return PyBool_FromLong(retval);
|
|
}
|
|
|
|
PyDoc_STRVAR(issubclass_doc,
|
|
"issubclass(C, B) -> bool\n\
|
|
\n\
|
|
Return whether class C is a subclass (i.e., a derived class) of class B.\n\
|
|
When using a tuple as the second argument issubclass(X, (A, B, ...)),\n\
|
|
is a shortcut for issubclass(X, A) or issubclass(X, B) or ... (etc.).");
|
|
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
Py_ssize_t tuplesize;
|
|
PyObject *ittuple; /* tuple of iterators */
|
|
PyObject *result;
|
|
} zipobject;
|
|
|
|
static PyObject *
|
|
zip_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
|
|
{
|
|
zipobject *lz;
|
|
Py_ssize_t i;
|
|
PyObject *ittuple; /* tuple of iterators */
|
|
PyObject *result;
|
|
Py_ssize_t tuplesize = PySequence_Length(args);
|
|
|
|
if (type == &PyZip_Type && !_PyArg_NoKeywords("zip()", kwds))
|
|
return NULL;
|
|
|
|
/* args must be a tuple */
|
|
assert(PyTuple_Check(args));
|
|
|
|
/* obtain iterators */
|
|
ittuple = PyTuple_New(tuplesize);
|
|
if (ittuple == NULL)
|
|
return NULL;
|
|
for (i=0; i < tuplesize; ++i) {
|
|
PyObject *item = PyTuple_GET_ITEM(args, i);
|
|
PyObject *it = PyObject_GetIter(item);
|
|
if (it == NULL) {
|
|
if (PyErr_ExceptionMatches(PyExc_TypeError))
|
|
PyErr_Format(PyExc_TypeError,
|
|
"zip argument #%zd must support iteration",
|
|
i+1);
|
|
Py_DECREF(ittuple);
|
|
return NULL;
|
|
}
|
|
PyTuple_SET_ITEM(ittuple, i, it);
|
|
}
|
|
|
|
/* create a result holder */
|
|
result = PyTuple_New(tuplesize);
|
|
if (result == NULL) {
|
|
Py_DECREF(ittuple);
|
|
return NULL;
|
|
}
|
|
for (i=0 ; i < tuplesize ; i++) {
|
|
Py_INCREF(Py_None);
|
|
PyTuple_SET_ITEM(result, i, Py_None);
|
|
}
|
|
|
|
/* create zipobject structure */
|
|
lz = (zipobject *)type->tp_alloc(type, 0);
|
|
if (lz == NULL) {
|
|
Py_DECREF(ittuple);
|
|
Py_DECREF(result);
|
|
return NULL;
|
|
}
|
|
lz->ittuple = ittuple;
|
|
lz->tuplesize = tuplesize;
|
|
lz->result = result;
|
|
|
|
return (PyObject *)lz;
|
|
}
|
|
|
|
static void
|
|
zip_dealloc(zipobject *lz)
|
|
{
|
|
PyObject_GC_UnTrack(lz);
|
|
Py_XDECREF(lz->ittuple);
|
|
Py_XDECREF(lz->result);
|
|
Py_TYPE(lz)->tp_free(lz);
|
|
}
|
|
|
|
static int
|
|
zip_traverse(zipobject *lz, visitproc visit, void *arg)
|
|
{
|
|
Py_VISIT(lz->ittuple);
|
|
Py_VISIT(lz->result);
|
|
return 0;
|
|
}
|
|
|
|
static PyObject *
|
|
zip_next(zipobject *lz)
|
|
{
|
|
Py_ssize_t i;
|
|
Py_ssize_t tuplesize = lz->tuplesize;
|
|
PyObject *result = lz->result;
|
|
PyObject *it;
|
|
PyObject *item;
|
|
PyObject *olditem;
|
|
|
|
if (tuplesize == 0)
|
|
return NULL;
|
|
if (Py_REFCNT(result) == 1) {
|
|
Py_INCREF(result);
|
|
for (i=0 ; i < tuplesize ; i++) {
|
|
it = PyTuple_GET_ITEM(lz->ittuple, i);
|
|
item = (*Py_TYPE(it)->tp_iternext)(it);
|
|
if (item == NULL) {
|
|
Py_DECREF(result);
|
|
return NULL;
|
|
}
|
|
olditem = PyTuple_GET_ITEM(result, i);
|
|
PyTuple_SET_ITEM(result, i, item);
|
|
Py_DECREF(olditem);
|
|
}
|
|
} else {
|
|
result = PyTuple_New(tuplesize);
|
|
if (result == NULL)
|
|
return NULL;
|
|
for (i=0 ; i < tuplesize ; i++) {
|
|
it = PyTuple_GET_ITEM(lz->ittuple, i);
|
|
item = (*Py_TYPE(it)->tp_iternext)(it);
|
|
if (item == NULL) {
|
|
Py_DECREF(result);
|
|
return NULL;
|
|
}
|
|
PyTuple_SET_ITEM(result, i, item);
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
static PyObject *
|
|
zip_reduce(zipobject *lz)
|
|
{
|
|
/* Just recreate the zip with the internal iterator tuple */
|
|
return Py_BuildValue("OO", Py_TYPE(lz), lz->ittuple);
|
|
}
|
|
|
|
static PyMethodDef zip_methods[] = {
|
|
{"__reduce__", (PyCFunction)zip_reduce, METH_NOARGS, reduce_doc},
|
|
{NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
PyDoc_STRVAR(zip_doc,
|
|
"zip(iter1 [,iter2 [...]]) --> zip object\n\
|
|
\n\
|
|
Return a zip object whose .__next__() method returns a tuple where\n\
|
|
the i-th element comes from the i-th iterable argument. The .__next__()\n\
|
|
method continues until the shortest iterable in the argument sequence\n\
|
|
is exhausted and then it raises StopIteration.");
|
|
|
|
PyTypeObject PyZip_Type = {
|
|
PyVarObject_HEAD_INIT(&PyType_Type, 0)
|
|
"zip", /* tp_name */
|
|
sizeof(zipobject), /* tp_basicsize */
|
|
0, /* tp_itemsize */
|
|
/* methods */
|
|
(destructor)zip_dealloc, /* tp_dealloc */
|
|
0, /* tp_print */
|
|
0, /* tp_getattr */
|
|
0, /* tp_setattr */
|
|
0, /* tp_reserved */
|
|
0, /* tp_repr */
|
|
0, /* tp_as_number */
|
|
0, /* tp_as_sequence */
|
|
0, /* tp_as_mapping */
|
|
0, /* tp_hash */
|
|
0, /* tp_call */
|
|
0, /* tp_str */
|
|
PyObject_GenericGetAttr, /* tp_getattro */
|
|
0, /* tp_setattro */
|
|
0, /* tp_as_buffer */
|
|
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC |
|
|
Py_TPFLAGS_BASETYPE, /* tp_flags */
|
|
zip_doc, /* tp_doc */
|
|
(traverseproc)zip_traverse, /* tp_traverse */
|
|
0, /* tp_clear */
|
|
0, /* tp_richcompare */
|
|
0, /* tp_weaklistoffset */
|
|
PyObject_SelfIter, /* tp_iter */
|
|
(iternextfunc)zip_next, /* tp_iternext */
|
|
zip_methods, /* tp_methods */
|
|
0, /* tp_members */
|
|
0, /* tp_getset */
|
|
0, /* tp_base */
|
|
0, /* tp_dict */
|
|
0, /* tp_descr_get */
|
|
0, /* tp_descr_set */
|
|
0, /* tp_dictoffset */
|
|
0, /* tp_init */
|
|
PyType_GenericAlloc, /* tp_alloc */
|
|
zip_new, /* tp_new */
|
|
PyObject_GC_Del, /* tp_free */
|
|
};
|
|
|
|
|
|
static PyMethodDef builtin_methods[] = {
|
|
{"__build_class__", (PyCFunction)builtin___build_class__,
|
|
METH_VARARGS | METH_KEYWORDS, build_class_doc},
|
|
{"__import__", (PyCFunction)builtin___import__, METH_VARARGS | METH_KEYWORDS, import_doc},
|
|
{"abs", builtin_abs, METH_O, abs_doc},
|
|
{"all", builtin_all, METH_O, all_doc},
|
|
{"any", builtin_any, METH_O, any_doc},
|
|
{"ascii", builtin_ascii, METH_O, ascii_doc},
|
|
{"bin", builtin_bin, METH_O, bin_doc},
|
|
{"callable", builtin_callable, METH_O, callable_doc},
|
|
{"chr", builtin_chr, METH_VARARGS, chr_doc},
|
|
{"compile", (PyCFunction)builtin_compile, METH_VARARGS | METH_KEYWORDS, compile_doc},
|
|
{"delattr", builtin_delattr, METH_VARARGS, delattr_doc},
|
|
{"dir", builtin_dir, METH_VARARGS, dir_doc},
|
|
{"divmod", builtin_divmod, METH_VARARGS, divmod_doc},
|
|
{"eval", builtin_eval, METH_VARARGS, eval_doc},
|
|
{"exec", builtin_exec, METH_VARARGS, exec_doc},
|
|
{"format", builtin_format, METH_VARARGS, format_doc},
|
|
{"getattr", builtin_getattr, METH_VARARGS, getattr_doc},
|
|
{"globals", (PyCFunction)builtin_globals, METH_NOARGS, globals_doc},
|
|
{"hasattr", builtin_hasattr, METH_VARARGS, hasattr_doc},
|
|
{"hash", builtin_hash, METH_O, hash_doc},
|
|
{"hex", builtin_hex, METH_O, hex_doc},
|
|
{"id", builtin_id, METH_O, id_doc},
|
|
{"input", builtin_input, METH_VARARGS, input_doc},
|
|
{"isinstance", builtin_isinstance, METH_VARARGS, isinstance_doc},
|
|
{"issubclass", builtin_issubclass, METH_VARARGS, issubclass_doc},
|
|
{"iter", builtin_iter, METH_VARARGS, iter_doc},
|
|
{"len", builtin_len, METH_O, len_doc},
|
|
{"locals", (PyCFunction)builtin_locals, METH_NOARGS, locals_doc},
|
|
{"max", (PyCFunction)builtin_max, METH_VARARGS | METH_KEYWORDS, max_doc},
|
|
{"min", (PyCFunction)builtin_min, METH_VARARGS | METH_KEYWORDS, min_doc},
|
|
{"next", (PyCFunction)builtin_next, METH_VARARGS, next_doc},
|
|
{"oct", builtin_oct, METH_O, oct_doc},
|
|
{"ord", builtin_ord, METH_O, ord_doc},
|
|
{"pow", builtin_pow, METH_VARARGS, pow_doc},
|
|
{"print", (PyCFunction)builtin_print, METH_VARARGS | METH_KEYWORDS, print_doc},
|
|
{"repr", builtin_repr, METH_O, repr_doc},
|
|
{"round", (PyCFunction)builtin_round, METH_VARARGS | METH_KEYWORDS, round_doc},
|
|
{"setattr", builtin_setattr, METH_VARARGS, setattr_doc},
|
|
{"sorted", (PyCFunction)builtin_sorted, METH_VARARGS | METH_KEYWORDS, sorted_doc},
|
|
{"sum", builtin_sum, METH_VARARGS, sum_doc},
|
|
{"vars", builtin_vars, METH_VARARGS, vars_doc},
|
|
{NULL, NULL},
|
|
};
|
|
|
|
PyDoc_STRVAR(builtin_doc,
|
|
"Built-in functions, exceptions, and other objects.\n\
|
|
\n\
|
|
Noteworthy: None is the `nil' object; Ellipsis represents `...' in slices.");
|
|
|
|
static struct PyModuleDef builtinsmodule = {
|
|
PyModuleDef_HEAD_INIT,
|
|
"builtins",
|
|
builtin_doc,
|
|
-1, /* multiple "initialization" just copies the module dict. */
|
|
builtin_methods,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
NULL
|
|
};
|
|
|
|
|
|
PyObject *
|
|
_PyBuiltin_Init(void)
|
|
{
|
|
PyObject *mod, *dict, *debug;
|
|
|
|
if (PyType_Ready(&PyFilter_Type) < 0 ||
|
|
PyType_Ready(&PyMap_Type) < 0 ||
|
|
PyType_Ready(&PyZip_Type) < 0)
|
|
return NULL;
|
|
|
|
mod = PyModule_Create(&builtinsmodule);
|
|
if (mod == NULL)
|
|
return NULL;
|
|
dict = PyModule_GetDict(mod);
|
|
|
|
#ifdef Py_TRACE_REFS
|
|
/* "builtins" exposes a number of statically allocated objects
|
|
* that, before this code was added in 2.3, never showed up in
|
|
* the list of "all objects" maintained by Py_TRACE_REFS. As a
|
|
* result, programs leaking references to None and False (etc)
|
|
* couldn't be diagnosed by examining sys.getobjects(0).
|
|
*/
|
|
#define ADD_TO_ALL(OBJECT) _Py_AddToAllObjects((PyObject *)(OBJECT), 0)
|
|
#else
|
|
#define ADD_TO_ALL(OBJECT) (void)0
|
|
#endif
|
|
|
|
#define SETBUILTIN(NAME, OBJECT) \
|
|
if (PyDict_SetItemString(dict, NAME, (PyObject *)OBJECT) < 0) \
|
|
return NULL; \
|
|
ADD_TO_ALL(OBJECT)
|
|
|
|
SETBUILTIN("None", Py_None);
|
|
SETBUILTIN("Ellipsis", Py_Ellipsis);
|
|
SETBUILTIN("NotImplemented", Py_NotImplemented);
|
|
SETBUILTIN("False", Py_False);
|
|
SETBUILTIN("True", Py_True);
|
|
SETBUILTIN("bool", &PyBool_Type);
|
|
SETBUILTIN("memoryview", &PyMemoryView_Type);
|
|
SETBUILTIN("bytearray", &PyByteArray_Type);
|
|
SETBUILTIN("bytes", &PyBytes_Type);
|
|
SETBUILTIN("classmethod", &PyClassMethod_Type);
|
|
SETBUILTIN("complex", &PyComplex_Type);
|
|
SETBUILTIN("dict", &PyDict_Type);
|
|
SETBUILTIN("enumerate", &PyEnum_Type);
|
|
SETBUILTIN("filter", &PyFilter_Type);
|
|
SETBUILTIN("float", &PyFloat_Type);
|
|
SETBUILTIN("frozenset", &PyFrozenSet_Type);
|
|
SETBUILTIN("property", &PyProperty_Type);
|
|
SETBUILTIN("int", &PyLong_Type);
|
|
SETBUILTIN("list", &PyList_Type);
|
|
SETBUILTIN("map", &PyMap_Type);
|
|
SETBUILTIN("object", &PyBaseObject_Type);
|
|
SETBUILTIN("range", &PyRange_Type);
|
|
SETBUILTIN("reversed", &PyReversed_Type);
|
|
SETBUILTIN("set", &PySet_Type);
|
|
SETBUILTIN("slice", &PySlice_Type);
|
|
SETBUILTIN("staticmethod", &PyStaticMethod_Type);
|
|
SETBUILTIN("str", &PyUnicode_Type);
|
|
SETBUILTIN("super", &PySuper_Type);
|
|
SETBUILTIN("tuple", &PyTuple_Type);
|
|
SETBUILTIN("type", &PyType_Type);
|
|
SETBUILTIN("zip", &PyZip_Type);
|
|
debug = PyBool_FromLong(Py_OptimizeFlag == 0);
|
|
if (PyDict_SetItemString(dict, "__debug__", debug) < 0) {
|
|
Py_XDECREF(debug);
|
|
return NULL;
|
|
}
|
|
Py_XDECREF(debug);
|
|
|
|
return mod;
|
|
#undef ADD_TO_ALL
|
|
#undef SETBUILTIN
|
|
}
|