/* System module */

/*

Various bits of information used by the interpreter are collected in

module 'sys'.

Function member:

- exit(sts): raise SystemExit

Data members:

- stdin, stdout, stderr: standard file objects

- modules: the table of modules (dictionary)

- path: module search path (list of strings)

- argv: script arguments (list of strings)

- ps1, ps2: optional primary and secondary prompts (strings)

*/

#include "Python.h"

#include "code.h"

#include "frameobject.h"

#include "pycore_pylifecycle.h"

#include "pycore_pymem.h"

#include "pycore_pathconfig.h"

#include "pycore_pystate.h"

#include "pythread.h"

#include "osdefs.h"

#include <locale.h>

#ifdef MS_WINDOWS

#define WIN32_LEAN_AND_MEAN

#include <windows.h>

#endif /* MS_WINDOWS */

#ifdef MS_COREDLL

extern void *PyWin_DLLhModule;

/* A string loaded from the DLL at startup: */

extern const char *PyWin_DLLVersionString;

#endif

/*[clinic input]

module sys

[clinic start generated code]*/

/*[clinic end generated code: output=da39a3ee5e6b4b0d input=3726b388feee8cea]*/

#include "clinic/sysmodule.c.h"

_Py_IDENTIFIER(_);

_Py_IDENTIFIER(__sizeof__);

_Py_IDENTIFIER(_xoptions);

_Py_IDENTIFIER(buffer);

_Py_IDENTIFIER(builtins);

_Py_IDENTIFIER(encoding);

_Py_IDENTIFIER(path);

_Py_IDENTIFIER(stdout);

_Py_IDENTIFIER(stderr);

_Py_IDENTIFIER(warnoptions);

_Py_IDENTIFIER(write);

PyObject *

_PySys_GetObjectId(_Py_Identifier *key)

{

PyObject *sd = _PyInterpreterState_GET_UNSAFE()->sysdict;

if (sd == NULL) {

return NULL;

}

return _PyDict_GetItemId(sd, key);

}

PyObject *

PySys_GetObject(const char *name)

{

PyObject *sd = _PyInterpreterState_GET_UNSAFE()->sysdict;

if (sd == NULL) {

return NULL;

}

return PyDict_GetItemString(sd, name);

}

int

_PySys_SetObjectId(_Py_Identifier *key, PyObject *v)

{

PyObject *sd = _PyInterpreterState_GET_UNSAFE()->sysdict;

if (v == NULL) {

if (_PyDict_GetItemId(sd, key) == NULL) {

return 0;

}

else {

return _PyDict_DelItemId(sd, key);

}

}

else {

return _PyDict_SetItemId(sd, key, v);

}

}

int

PySys_SetObject(const char *name, PyObject *v)

{

PyObject *sd = _PyInterpreterState_GET_UNSAFE()->sysdict;

if (v == NULL) {

if (PyDict_GetItemString(sd, name) == NULL) {

return 0;

}

else {

return PyDict_DelItemString(sd, name);

}

}

else {

return PyDict_SetItemString(sd, name, v);

}

}

static PyObject *

sys_breakpointhook(PyObject *self, PyObject *const *args, Py_ssize_t nargs, PyObject *keywords)

{

assert(!PyErr_Occurred());

char *envar = Py_GETENV("PYTHONBREAKPOINT");

if (envar == NULL || strlen(envar) == 0) {

envar = "pdb.set_trace";

}

else if (!strcmp(envar, "0")) {

/* The breakpoint is explicitly no-op'd. */

Py_RETURN_NONE;

}

/* According to POSIX the string returned by getenv() might be invalidated

* or the string content might be overwritten by a subsequent call to

* getenv(). Since importing a module can performs the getenv() calls,

* we need to save a copy of envar. */

envar = _PyMem_RawStrdup(envar);

if (envar == NULL) {

PyErr_NoMemory();

return NULL;

}

const char *last_dot = strrchr(envar, '.');

const char *attrname = NULL;

PyObject *modulepath = NULL;

if (last_dot == NULL) {

/* The breakpoint is a built-in, e.g. PYTHONBREAKPOINT=int */

modulepath = PyUnicode_FromString("builtins");

attrname = envar;

}

else if (last_dot != envar) {

/* Split on the last dot; */

modulepath = PyUnicode_FromStringAndSize(envar, last_dot - envar);

attrname = last_dot + 1;

}

else {

goto warn;

}

if (modulepath == NULL) {

PyMem_RawFree(envar);

return NULL;

}

PyObject *module = PyImport_Import(modulepath);

Py_DECREF(modulepath);

if (module == NULL) {

if (PyErr_ExceptionMatches(PyExc_ImportError)) {

goto warn;

}

PyMem_RawFree(envar);

return NULL;

}

PyObject *hook = PyObject_GetAttrString(module, attrname);

Py_DECREF(module);

if (hook == NULL) {

if (PyErr_ExceptionMatches(PyExc_AttributeError)) {

goto warn;

}

PyMem_RawFree(envar);

return NULL;

}

PyMem_RawFree(envar);

PyObject *retval = _PyObject_FastCallKeywords(hook, args, nargs, keywords);

Py_DECREF(hook);

return retval;

warn:

/* If any of the imports went wrong, then warn and ignore. */

PyErr_Clear();

int status = PyErr_WarnFormat(

PyExc_RuntimeWarning, 0,

"Ignoring unimportable $PYTHONBREAKPOINT: \"%s\"", envar);

PyMem_RawFree(envar);

if (status < 0) {

/* Printing the warning raised an exception. */

return NULL;

}

/* The warning was (probably) issued. */

Py_RETURN_NONE;

}

PyDoc_STRVAR(breakpointhook_doc,

"breakpointhook(*args, **kws)\n"

"\n"

"This hook function is called by built-in breakpoint().\n"

);

/* Write repr(o) to sys.stdout using sys.stdout.encoding and 'backslashreplace'

error handler. If sys.stdout has a buffer attribute, use

sys.stdout.buffer.write(encoded), otherwise redecode the string and use

sys.stdout.write(redecoded).

Helper function for sys_displayhook(). */

static int

sys_displayhook_unencodable(PyObject *outf, PyObject *o)

{

PyObject *stdout_encoding = NULL;

PyObject *encoded, *escaped_str, *repr_str, *buffer, *result;

const char *stdout_encoding_str;

int ret;

stdout_encoding = _PyObject_GetAttrId(outf, &PyId_encoding);

if (stdout_encoding == NULL)

goto error;

stdout_encoding_str = PyUnicode_AsUTF8(stdout_encoding);

if (stdout_encoding_str == NULL)

goto error;

repr_str = PyObject_Repr(o);

if (repr_str == NULL)

goto error;

encoded = PyUnicode_AsEncodedString(repr_str,

stdout_encoding_str,

"backslashreplace");

Py_DECREF(repr_str);

if (encoded == NULL)

goto error;

buffer = _PyObject_GetAttrId(outf, &PyId_buffer);

if (buffer) {

result = _PyObject_CallMethodIdObjArgs(buffer, &PyId_write, encoded, NULL);

Py_DECREF(buffer);

Py_DECREF(encoded);

if (result == NULL)

goto error;

Py_DECREF(result);

}

else {

PyErr_Clear();

escaped_str = PyUnicode_FromEncodedObject(encoded,

stdout_encoding_str,

"strict");

Py_DECREF(encoded);

if (PyFile_WriteObject(escaped_str, outf, Py_PRINT_RAW) != 0) {

Py_DECREF(escaped_str);

goto error;

}

Py_DECREF(escaped_str);

}

ret = 0;

goto finally;

error:

ret = -1;

finally:

Py_XDECREF(stdout_encoding);

return ret;

}

/*[clinic input]

sys.displayhook

object as o: object

/

Print an object to sys.stdout and also save it in builtins._

[clinic start generated code]*/

static PyObject *

sys_displayhook(PyObject *module, PyObject *o)

/*[clinic end generated code: output=347477d006df92ed input=08ba730166d7ef72]*/

{

PyObject *outf;

PyObject *builtins;

static PyObject *newline = NULL;

int err;

builtins = _PyImport_GetModuleId(&PyId_builtins);

if (builtins == NULL) {

PyErr_SetString(PyExc_RuntimeError, "lost builtins module");

return NULL;

}

Py_DECREF(builtins);

/* Print value except if None */

/* After printing, also assign to '_' */

/* Before, set '_' to None to avoid recursion */

if (o == Py_None) {

Py_RETURN_NONE;

}

if (_PyObject_SetAttrId(builtins, &PyId__, Py_None) != 0)

return NULL;

outf = _PySys_GetObjectId(&PyId_stdout);

if (outf == NULL || outf == Py_None) {

PyErr_SetString(PyExc_RuntimeError, "lost sys.stdout");

return NULL;

}

if (PyFile_WriteObject(o, outf, 0) != 0) {

if (PyErr_ExceptionMatches(PyExc_UnicodeEncodeError)) {

/* repr(o) is not encodable to sys.stdout.encoding with

* sys.stdout.errors error handler (which is probably 'strict') */

PyErr_Clear();

err = sys_displayhook_unencodable(outf, o);

if (err)

return NULL;

}

else {

return NULL;

}

}

if (newline == NULL) {

newline = PyUnicode_FromString("\n");

if (newline == NULL)

return NULL;

}

if (PyFile_WriteObject(newline, outf, Py_PRINT_RAW) != 0)

return NULL;

if (_PyObject_SetAttrId(builtins, &PyId__, o) != 0)

return NULL;

Py_RETURN_NONE;

}

/*[clinic input]

sys.excepthook

exctype: object

value: object

traceback: object

/

Handle an exception by displaying it with a traceback on sys.stderr.

[clinic start generated code]*/

static PyObject *

sys_excepthook_impl(PyObject *module, PyObject *exctype, PyObject *value,

PyObject *traceback)

/*[clinic end generated code: output=18d99fdda21b6b5e input=ecf606fa826f19d9]*/

{

PyErr_Display(exctype, value, traceback);

Py_RETURN_NONE;

}

/*[clinic input]

sys.exc_info

Return current exception information: (type, value, traceback).

Return information about the most recent exception caught by an except

clause in the current stack frame or in an older stack frame.

[clinic start generated code]*/

static PyObject *

sys_exc_info_impl(PyObject *module)

/*[clinic end generated code: output=3afd0940cf3a4d30 input=b5c5bf077788a3e5]*/

{

_PyErr_StackItem *err_info = _PyErr_GetTopmostException(_PyThreadState_GET());

return Py_BuildValue(

"(OOO)",

err_info->exc_type != NULL ? err_info->exc_type : Py_None,

err_info->exc_value != NULL ? err_info->exc_value : Py_None,

err_info->exc_traceback != NULL ?

err_info->exc_traceback : Py_None);

}

/*[clinic input]

sys.exit

status: object = NULL

/

Exit the interpreter by raising SystemExit(status).

If the status is omitted or None, it defaults to zero (i.e., success).

If the status is an integer, it will be used as the system exit status.

If it is another kind of object, it will be printed and the system

exit status will be one (i.e., failure).

[clinic start generated code]*/

static PyObject *

sys_exit_impl(PyObject *module, PyObject *status)

/*[clinic end generated code: output=13870986c1ab2ec0 input=a737351f86685e9c]*/

{

/* Raise SystemExit so callers may catch it or clean up. */

PyErr_SetObject(PyExc_SystemExit, status);

return NULL;

}

/*[clinic input]

sys.getdefaultencoding

Return the current default encoding used by the Unicode implementation.

[clinic start generated code]*/

static PyObject *

sys_getdefaultencoding_impl(PyObject *module)

/*[clinic end generated code: output=256d19dfcc0711e6 input=d416856ddbef6909]*/

{

return PyUnicode_FromString(PyUnicode_GetDefaultEncoding());

}

/*[clinic input]

sys.getfilesystemencoding

Return the encoding used to convert Unicode filenames to OS filenames.

[clinic start generated code]*/

static PyObject *

sys_getfilesystemencoding_impl(PyObject *module)

/*[clinic end generated code: output=1dc4bdbe9be44aa7 input=8475f8649b8c7d8c]*/

{

PyInterpreterState *interp = _PyInterpreterState_GET_UNSAFE();

const _PyCoreConfig *config = &interp->core_config;

return PyUnicode_FromString(config->filesystem_encoding);

}

/*[clinic input]

sys.getfilesystemencodeerrors

Return the error mode used Unicode to OS filename conversion.

[clinic start generated code]*/

static PyObject *

sys_getfilesystemencodeerrors_impl(PyObject *module)

/*[clinic end generated code: output=ba77b36bbf7c96f5 input=22a1e8365566f1e5]*/

{

PyInterpreterState *interp = _PyInterpreterState_GET_UNSAFE();

const _PyCoreConfig *config = &interp->core_config;

return PyUnicode_FromString(config->filesystem_errors);

}

/*[clinic input]

sys.intern

string as s: unicode

/

``Intern'' the given string.

This enters the string in the (global) table of interned strings whose

purpose is to speed up dictionary lookups. Return the string itself or

the previously interned string object with the same value.

[clinic start generated code]*/

static PyObject *

sys_intern_impl(PyObject *module, PyObject *s)

/*[clinic end generated code: output=be680c24f5c9e5d6 input=849483c006924e2f]*/

{

if (PyUnicode_CheckExact(s)) {

Py_INCREF(s);

PyUnicode_InternInPlace(&s);

return s;

}

else {

PyErr_Format(PyExc_TypeError,

"can't intern %.400s", s->ob_type->tp_name);

return NULL;

}

}

/*

* Cached interned string objects used for calling the profile and

* trace functions. Initialized by trace_init().

*/

static PyObject *whatstrings[8] = {NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL};

static int

trace_init(void)

{

static const char * const whatnames[8] = {

"call", "exception", "line", "return",

"c_call", "c_exception", "c_return",

"opcode"

};

PyObject *name;

int i;

for (i = 0; i < 8; ++i) {

if (whatstrings[i] == NULL) {

name = PyUnicode_InternFromString(whatnames[i]);

if (name == NULL)

return -1;

whatstrings[i] = name;

}

}

return 0;

}

static PyObject *

call_trampoline(PyObject* callback,

PyFrameObject *frame, int what, PyObject *arg)

{

PyObject *result;

PyObject *stack[3];

if (PyFrame_FastToLocalsWithError(frame) < 0) {

return NULL;

}

stack[0] = (PyObject *)frame;

stack[1] = whatstrings[what];

stack[2] = (arg != NULL) ? arg : Py_None;

/* call the Python-level function */

result = _PyObject_FastCall(callback, stack, 3);

PyFrame_LocalsToFast(frame, 1);

if (result == NULL) {

PyTraceBack_Here(frame);

}

return result;

}

static int

profile_trampoline(PyObject *self, PyFrameObject *frame,

int what, PyObject *arg)

{

PyObject *result;

if (arg == NULL)

arg = Py_None;

result = call_trampoline(self, frame, what, arg);

if (result == NULL) {

PyEval_SetProfile(NULL, NULL);

return -1;

}

Py_DECREF(result);

return 0;

}

static int

trace_trampoline(PyObject *self, PyFrameObject *frame,

int what, PyObject *arg)

{

PyObject *callback;

PyObject *result;

if (what == PyTrace_CALL)

callback = self;

else

callback = frame->f_trace;

if (callback == NULL)

return 0;

result = call_trampoline(callback, frame, what, arg);

if (result == NULL) {

PyEval_SetTrace(NULL, NULL);

Py_CLEAR(frame->f_trace);

return -1;

}

if (result != Py_None) {

Py_XSETREF(frame->f_trace, result);

}

else {

Py_DECREF(result);

}

return 0;

}

static PyObject *

sys_settrace(PyObject *self, PyObject *args)

{

if (trace_init() == -1)

return NULL;

if (args == Py_None)

PyEval_SetTrace(NULL, NULL);

else

PyEval_SetTrace(trace_trampoline, args);

Py_RETURN_NONE;

}

PyDoc_STRVAR(settrace_doc,

"settrace(function)\n\

\n\

Set the global debug tracing function. It will be called on each\n\

function call. See the debugger chapter in the library manual."

);

/*[clinic input]

sys.gettrace

Return the global debug tracing function set with sys.settrace.

See the debugger chapter in the library manual.

[clinic start generated code]*/

static PyObject *

sys_gettrace_impl(PyObject *module)

/*[clinic end generated code: output=e97e3a4d8c971b6e input=373b51bb2147f4d8]*/

{

PyThreadState *tstate = _PyThreadState_GET();

PyObject *temp = tstate->c_traceobj;

if (temp == NULL)

temp = Py_None;

Py_INCREF(temp);

return temp;

}

static PyObject *

sys_setprofile(PyObject *self, PyObject *args)

{

if (trace_init() == -1)

return NULL;

if (args == Py_None)

PyEval_SetProfile(NULL, NULL);

else

PyEval_SetProfile(profile_trampoline, args);

Py_RETURN_NONE;

}

PyDoc_STRVAR(setprofile_doc,

"setprofile(function)\n\

\n\

Set the profiling function. It will be called on each function call\n\

and return. See the profiler chapter in the library manual."

);

/*[clinic input]

sys.getprofile

Return the profiling function set with sys.setprofile.

See the profiler chapter in the library manual.

[clinic start generated code]*/

static PyObject *

sys_getprofile_impl(PyObject *module)

/*[clinic end generated code: output=579b96b373448188 input=1b3209d89a32965d]*/

{

PyThreadState *tstate = _PyThreadState_GET();

PyObject *temp = tstate->c_profileobj;

if (temp == NULL)

temp = Py_None;

Py_INCREF(temp);

return temp;

}

/*[clinic input]

sys.setcheckinterval

n: int

/

Set the async event check interval to n instructions.

This tells the Python interpreter to check for asynchronous events

every n instructions.

This also affects how often thread switches occur.

[clinic start generated code]*/

static PyObject *

sys_setcheckinterval_impl(PyObject *module, int n)

/*[clinic end generated code: output=3f686cef07e6e178 input=7a35b17bf22a6227]*/

{

if (PyErr_WarnEx(PyExc_DeprecationWarning,

"sys.getcheckinterval() and sys.setcheckinterval() "

"are deprecated. Use sys.setswitchinterval() "

"instead.", 1) < 0)

return NULL;

PyInterpreterState *interp = _PyInterpreterState_Get();

interp->check_interval = n;

Py_RETURN_NONE;

}

/*[clinic input]

sys.getcheckinterval

Return the current check interval; see sys.setcheckinterval().

[clinic start generated code]*/

static PyObject *

sys_getcheckinterval_impl(PyObject *module)

/*[clinic end generated code: output=1b5060bf2b23a47c input=4b6589cbcca1db4e]*/

{

if (PyErr_WarnEx(PyExc_DeprecationWarning,

"sys.getcheckinterval() and sys.setcheckinterval() "

"are deprecated. Use sys.getswitchinterval() "

"instead.", 1) < 0)

return NULL;

PyInterpreterState *interp = _PyInterpreterState_Get();

return PyLong_FromLong(interp->check_interval);

}

/*[clinic input]

sys.setswitchinterval

interval: double

/

Set the ideal thread switching delay inside the Python interpreter.

The actual frequency of switching threads can be lower if the

interpreter executes long sequences of uninterruptible code

(this is implementation-specific and workload-dependent).

The parameter must represent the desired switching delay in seconds

A typical value is 0.005 (5 milliseconds).

[clinic start generated code]*/

static PyObject *

sys_setswitchinterval_impl(PyObject *module, double interval)

/*[clinic end generated code: output=65a19629e5153983 input=561b477134df91d9]*/

{

if (interval <= 0.0) {

PyErr_SetString(PyExc_ValueError,

"switch interval must be strictly positive");

return NULL;

}

_PyEval_SetSwitchInterval((unsigned long) (1e6 * interval));

Py_RETURN_NONE;

}

/*[clinic input]

sys.getswitchinterval -> double

Return the current thread switch interval; see sys.setswitchinterval().

[clinic start generated code]*/

static double

sys_getswitchinterval_impl(PyObject *module)

/*[clinic end generated code: output=a38c277c85b5096d input=bdf9d39c0ebbbb6f]*/

{

return 1e-6 * _PyEval_GetSwitchInterval();

}

/*[clinic input]

sys.setrecursionlimit

limit as new_limit: int

/

Set the maximum depth of the Python interpreter stack to n.

This limit prevents infinite recursion from causing an overflow of the C

stack and crashing Python. The highest possible limit is platform-

dependent.

[clinic start generated code]*/

static PyObject *

sys_setrecursionlimit_impl(PyObject *module, int new_limit)

/*[clinic end generated code: output=35e1c64754800ace input=b0f7a23393924af3]*/

{

int mark;

PyThreadState *tstate;

if (new_limit < 1) {

PyErr_SetString(PyExc_ValueError,

"recursion limit must be greater or equal than 1");

return NULL;

}

/* Issue #25274: When the recursion depth hits the recursion limit in

_Py_CheckRecursiveCall(), the overflowed flag of the thread state is

set to 1 and a RecursionError is raised. The overflowed flag is reset

to 0 when the recursion depth goes below the low-water mark: see

Py_LeaveRecursiveCall().

Reject too low new limit if the current recursion depth is higher than

the new low-water mark. Otherwise it may not be possible anymore to

reset the overflowed flag to 0. */

mark = _Py_RecursionLimitLowerWaterMark(new_limit);

tstate = _PyThreadState_GET();

if (tstate->recursion_depth >= mark) {

PyErr_Format(PyExc_RecursionError,

"cannot set the recursion limit to %i at "

"the recursion depth %i: the limit is too low",

new_limit, tstate->recursion_depth);

return NULL;

}

Py_SetRecursionLimit(new_limit);

Py_RETURN_NONE;

}

/*[clinic input]

sys.set_coroutine_origin_tracking_depth

depth: int

Enable or disable origin tracking for coroutine objects in this thread.

Coroutine objects will track 'depth' frames of traceback information

about where they came from, available in their cr_origin attribute.

Set a depth of 0 to disable.

[clinic start generated code]*/

static PyObject *

sys_set_coroutine_origin_tracking_depth_impl(PyObject *module, int depth)

/*[clinic end generated code: output=0a2123c1cc6759c5 input=a1d0a05f89d2c426]*/

{

if (depth < 0) {

PyErr_SetString(PyExc_ValueError, "depth must be >= 0");

return NULL;

}

_PyEval_SetCoroutineOriginTrackingDepth(depth);

Py_RETURN_NONE;

}

/*[clinic input]

sys.get_coroutine_origin_tracking_depth -> int

Check status of origin tracking for coroutine objects in this thread.

[clinic start generated code]*/

static int

sys_get_coroutine_origin_tracking_depth_impl(PyObject *module)

/*[clinic end generated code: output=3699f7be95a3afb8 input=335266a71205b61a]*/

{

return _PyEval_GetCoroutineOriginTrackingDepth();

}

/*[clinic input]

sys.set_coroutine_wrapper

wrapper: object

/

Set a wrapper for coroutine objects.

[clinic start generated code]*/

static PyObject *

sys_set_coroutine_wrapper(PyObject *module, PyObject *wrapper)

/*[clinic end generated code: output=9c7db52d65f6b188 input=df6ac09a06afef34]*/

{

if (PyErr_WarnEx(PyExc_DeprecationWarning,

"set_coroutine_wrapper is deprecated", 1) < 0) {

return NULL;

}

if (wrapper != Py_None) {

if (!PyCallable_Check(wrapper)) {

PyErr_Format(PyExc_TypeError,

"callable expected, got %.50s",

Py_TYPE(wrapper)->tp_name);

return NULL;

}

_PyEval_SetCoroutineWrapper(wrapper);

}

else {

_PyEval_SetCoroutineWrapper(NULL);

}

Py_RETURN_NONE;

}

/*[clinic input]

sys.get_coroutine_wrapper

Return the wrapper for coroutines set by sys.set_coroutine_wrapper.

[clinic start generated code]*/

static PyObject *

sys_get_coroutine_wrapper_impl(PyObject *module)

/*[clinic end generated code: output=b74a7e4b14fe898e input=ef0351fb9ece0bb4]*/

{

if (PyErr_WarnEx(PyExc_DeprecationWarning,

"get_coroutine_wrapper is deprecated", 1) < 0) {

return NULL;

}

PyObject *wrapper = _PyEval_GetCoroutineWrapper();

if (wrapper == NULL) {

wrapper = Py_None;

}

Py_INCREF(wrapper);

return wrapper;

}

static PyTypeObject AsyncGenHooksType;

PyDoc_STRVAR(asyncgen_hooks_doc,

"asyncgen_hooks\n\

\n\

A struct sequence providing information about asynhronous\n\

generators hooks. The attributes are read only.");

static PyStructSequence_Field asyncgen_hooks_fields[] = {

{"firstiter", "Hook to intercept first iteration"},

{"finalizer", "Hook to intercept finalization"},

{0}

};

static PyStructSequence_Desc asyncgen_hooks_desc = {

"asyncgen_hooks", /* name */

asyncgen_hooks_doc, /* doc */

asyncgen_hooks_fields , /* fields */

2

};

static PyObject *

sys_set_asyncgen_hooks(PyObject *self, PyObject *args, PyObject *kw)

{

static char *keywords[] = {"firstiter", "finalizer", NULL};

PyObject *firstiter = NULL;

PyObject *finalizer = NULL;

if (!PyArg_ParseTupleAndKeywords(

args, kw, "|OO", keywords,

&firstiter, &finalizer)) {

return NULL;

}

if (finalizer && finalizer != Py_None) {

if (!PyCallable_Check(finalizer)) {

PyErr_Format(PyExc_TypeError,

"callable finalizer expected, got %.50s",

Py_TYPE(finalizer)->tp_name);

return NULL;

}

_PyEval_SetAsyncGenFinalizer(finalizer);

}

else if (finalizer == Py_None) {

_PyEval_SetAsyncGenFinalizer(NULL);

}

if (firstiter && firstiter != Py_None) {

if (!PyCallable_Check(firstiter)) {

PyErr_Format(PyExc_TypeError,

"callable firstiter expected, got %.50s",

Py_TYPE(firstiter)->tp_name);

return NULL;

}

_PyEval_SetAsyncGenFirstiter(firstiter);

}

else if (firstiter == Py_None) {

_PyEval_SetAsyncGenFirstiter(NULL);

}

Py_RETURN_NONE;

}

PyDoc_STRVAR(set_asyncgen_hooks_doc,

"set_asyncgen_hooks(* [, firstiter] [, finalizer])\n\

\n\

Set a finalizer for async generators objects."

);

/*[clinic input]

sys.get_asyncgen_hooks

Return the installed asynchronous generators hooks.

This returns a namedtuple of the form (firstiter, finalizer).

[clinic start generated code]*/

static PyObject *

sys_get_asyncgen_hooks_impl(PyObject *module)

/*[clinic end generated code: output=53a253707146f6cf input=3676b9ea62b14625]*/

{

PyObject *res;

PyObject *firstiter = _PyEval_GetAsyncGenFirstiter();

PyObject *finalizer = _PyEval_GetAsyncGenFinalizer();

res = PyStructSequence_New(&AsyncGenHooksType);

if (res == NULL) {

return NULL;

}

if (firstiter == NULL) {

firstiter = Py_None;

}

if (finalizer == NULL) {

finalizer = Py_None;

}

Py_INCREF(firstiter);

PyStructSequence_SET_ITEM(res, 0, firstiter);

Py_INCREF(finalizer);

PyStructSequence_SET_ITEM(res, 1, finalizer);

return res;

}

static PyTypeObject Hash_InfoType;

PyDoc_STRVAR(hash_info_doc,

"hash_info\n\

\n\

A struct sequence providing parameters used for computing\n\

hashes. The attributes are read only.");