#include "Python.h"#include "code.h" /* For DeprecationWarning about adding 'line'. */#include "frameobject.h"#define MODULE_NAME "_warnings"#define DEFAULT_ACTION_NAME "default_action"PyDoc_STRVAR(warnings__doc__,MODULE_NAME " provides basic warning filtering support.\n""It is a helper module to speed up interpreter start-up.");/* Both 'filters' and 'onceregistry' can be set in warnings.py;get_warnings_attr() will reset these variables accordingly. */static PyObject *_filters; /* List */static PyObject *_once_registry; /* Dict */static intcheck_matched(PyObject *obj, PyObject *arg){PyObject *result;int rc;if (obj == Py_None)return 1;result = PyObject_CallMethod(obj, "match", "O", arg);if (result == NULL)return -1;rc = PyObject_IsTrue(result);Py_DECREF(result);return rc;}/*Returns a new reference.A NULL return value can mean false or an error.*/static PyObject *get_warnings_attr(const char *attr){static PyObject *warnings_str = NULL;PyObject *all_modules;PyObject *warnings_module;int result;if (warnings_str == NULL) {warnings_str = PyString_InternFromString("warnings");if (warnings_str == NULL)return NULL;}all_modules = PyImport_GetModuleDict();result = PyDict_Contains(all_modules, warnings_str);if (result == -1 || result == 0)return NULL;warnings_module = PyDict_GetItem(all_modules, warnings_str);if (!PyObject_HasAttrString(warnings_module, attr))return NULL;return PyObject_GetAttrString(warnings_module, attr);}static PyObject *get_once_registry(void){PyObject *registry;registry = get_warnings_attr("onceregistry");if (registry == NULL) {if (PyErr_Occurred())return NULL;return _once_registry;}Py_DECREF(_once_registry);_once_registry = registry;return registry;}/* The item is a borrowed reference. */static const char *get_filter(PyObject *category, PyObject *text, Py_ssize_t lineno,PyObject *module, PyObject **item){PyObject *action, *m, *d;Py_ssize_t i;PyObject *warnings_filters;warnings_filters = get_warnings_attr("filters");if (warnings_filters == NULL) {if (PyErr_Occurred())return NULL;}else {Py_DECREF(_filters);_filters = warnings_filters;}if (!PyList_Check(_filters)) {PyErr_SetString(PyExc_ValueError,MODULE_NAME ".filters must be a list");return NULL;}/* _filters could change while we are iterating over it. */for (i = 0; i < PyList_GET_SIZE(_filters); i++) {PyObject *tmp_item, *action, *msg, *cat, *mod, *ln_obj;Py_ssize_t ln;int is_subclass, good_msg, good_mod;tmp_item = *item = PyList_GET_ITEM(_filters, i);if (PyTuple_Size(tmp_item) != 5) {PyErr_Format(PyExc_ValueError,MODULE_NAME ".filters item %zd isn't a 5-tuple", i);return NULL;}/* Python code: action, msg, cat, mod, ln = item */action = PyTuple_GET_ITEM(tmp_item, 0);msg = PyTuple_GET_ITEM(tmp_item, 1);cat = PyTuple_GET_ITEM(tmp_item, 2);mod = PyTuple_GET_ITEM(tmp_item, 3);ln_obj = PyTuple_GET_ITEM(tmp_item, 4);good_msg = check_matched(msg, text);good_mod = check_matched(mod, module);is_subclass = PyObject_IsSubclass(category, cat);ln = PyInt_AsSsize_t(ln_obj);if (good_msg == -1 || good_mod == -1 || is_subclass == -1 ||(ln == -1 && PyErr_Occurred()))return NULL;if (good_msg && is_subclass && good_mod && (ln == 0 || lineno == ln))return PyString_AsString(action);}m = PyImport_ImportModule(MODULE_NAME);if (m == NULL)return NULL;d = PyModule_GetDict(m);Py_DECREF(m);if (d == NULL)return NULL;action = PyDict_GetItemString(d, DEFAULT_ACTION_NAME);if (action != NULL)return PyString_AsString(action);PyErr_SetString(PyExc_ValueError,MODULE_NAME "." DEFAULT_ACTION_NAME " not found");return NULL;}static intalready_warned(PyObject *registry, PyObject *key, int should_set){PyObject *already_warned;if (key == NULL)return -1;already_warned = PyDict_GetItem(registry, key);if (already_warned != NULL) {int rc = PyObject_IsTrue(already_warned);if (rc != 0)return rc;}/* This warning wasn't found in the registry, set it. */if (should_set)return PyDict_SetItem(registry, key, Py_True);return 0;}/* New reference. */static PyObject *normalize_module(PyObject *filename){PyObject *module;const char *mod_str;Py_ssize_t len;int rc = PyObject_IsTrue(filename);if (rc == -1)return NULL;else if (rc == 0)return PyString_FromString("<unknown>");mod_str = PyString_AsString(filename);if (mod_str == NULL)return NULL;len = PyString_Size(filename);if (len < 0)return NULL;if (len >= 3 &&strncmp(mod_str + (len - 3), ".py", 3) == 0) {module = PyString_FromStringAndSize(mod_str, len-3);}else {module = filename;Py_INCREF(module);}return module;}static intupdate_registry(PyObject *registry, PyObject *text, PyObject *category,int add_zero){PyObject *altkey, *zero = NULL;int rc;if (add_zero) {zero = PyInt_FromLong(0);if (zero == NULL)return -1;altkey = PyTuple_Pack(3, text, category, zero);}elsealtkey = PyTuple_Pack(2, text, category);rc = already_warned(registry, altkey, 1);Py_XDECREF(zero);Py_XDECREF(altkey);return rc;}static voidshow_warning(PyObject *filename, int lineno, PyObject *text, PyObject*category, PyObject *sourceline){PyObject *f_stderr;PyObject *name;char lineno_str[128];PyOS_snprintf(lineno_str, sizeof(lineno_str), ":%d: ", lineno);name = PyObject_GetAttrString(category, "__name__");if (name == NULL) /* XXX Can an object lack a '__name__' attribute? */return;f_stderr = PySys_GetObject("stderr");if (f_stderr == NULL) {fprintf(stderr, "lost sys.stderr\n");Py_DECREF(name);return;}/* Print "filename:lineno: category: text\n" */PyFile_WriteObject(filename, f_stderr, Py_PRINT_RAW);PyFile_WriteString(lineno_str, f_stderr);PyFile_WriteObject(name, f_stderr, Py_PRINT_RAW);PyFile_WriteString(": ", f_stderr);PyFile_WriteObject(text, f_stderr, Py_PRINT_RAW);PyFile_WriteString("\n", f_stderr);Py_XDECREF(name);/* Print " source_line\n" */if (sourceline) {char *source_line_str = PyString_AS_STRING(sourceline);while (*source_line_str == ' ' || *source_line_str == '\t' ||*source_line_str == '014円')source_line_str++;PyFile_WriteString(source_line_str, f_stderr);PyFile_WriteString("\n", f_stderr);}else_Py_DisplaySourceLine(f_stderr, PyString_AS_STRING(filename),lineno, 2);PyErr_Clear();}static PyObject *warn_explicit(PyObject *category, PyObject *message,PyObject *filename, int lineno,PyObject *module, PyObject *registry, PyObject *sourceline){PyObject *key = NULL, *text = NULL, *result = NULL, *lineno_obj = NULL;PyObject *item = Py_None;const char *action;int rc;if (registry && !PyDict_Check(registry) && (registry != Py_None)) {PyErr_SetString(PyExc_TypeError, "'registry' must be a dict");return NULL;}/* Normalize module. */if (module == NULL) {module = normalize_module(filename);if (module == NULL)return NULL;}elsePy_INCREF(module);/* Normalize message. */Py_INCREF(message); /* DECREF'ed in cleanup. */rc = PyObject_IsInstance(message, PyExc_Warning);if (rc == -1) {goto cleanup;}if (rc == 1) {text = PyObject_Str(message);category = (PyObject*)message->ob_type;}else {text = message;message = PyObject_CallFunction(category, "O", message);if (message == NULL)goto cleanup;}lineno_obj = PyInt_FromLong(lineno);if (lineno_obj == NULL)goto cleanup;/* Create key. */key = PyTuple_Pack(3, text, category, lineno_obj);if (key == NULL)goto cleanup;if ((registry != NULL) && (registry != Py_None)) {rc = already_warned(registry, key, 0);if (rc == -1)goto cleanup;else if (rc == 1)goto return_none;/* Else this warning hasn't been generated before. */}action = get_filter(category, text, lineno, module, &item);if (action == NULL)goto cleanup;if (strcmp(action, "error") == 0) {PyErr_SetObject(category, message);goto cleanup;}/* Store in the registry that we've been here, *except* when the actionis "always". */rc = 0;if (strcmp(action, "always") != 0) {if (registry != NULL && registry != Py_None &&PyDict_SetItem(registry, key, Py_True) < 0)goto cleanup;else if (strcmp(action, "ignore") == 0)goto return_none;else if (strcmp(action, "once") == 0) {if (registry == NULL || registry == Py_None) {registry = get_once_registry();if (registry == NULL)goto cleanup;}/* _once_registry[(text, category)] = 1 */rc = update_registry(registry, text, category, 0);}else if (strcmp(action, "module") == 0) {/* registry[(text, category, 0)] = 1 */if (registry != NULL && registry != Py_None)rc = update_registry(registry, text, category, 0);}else if (strcmp(action, "default") != 0) {PyObject *to_str = PyObject_Str(item);const char *err_str = "???";if (to_str != NULL)err_str = PyString_AS_STRING(to_str);PyErr_Format(PyExc_RuntimeError,"Unrecognized action (%s) in warnings.filters:\n %s",action, err_str);Py_XDECREF(to_str);goto cleanup;}}if (rc == 1) /* Already warned for this module. */goto return_none;if (rc == 0) {PyObject *show_fxn = get_warnings_attr("showwarning");if (show_fxn == NULL) {if (PyErr_Occurred())goto cleanup;show_warning(filename, lineno, text, category, sourceline);}else {const char *msg = "functions overriding warnings.showwarning() ""must support the 'line' argument";const char *text_char = PyString_AS_STRING(text);if (strcmp(msg, text_char) == 0) {/* Prevent infinite recursion by using built-in implementationof showwarning(). */show_warning(filename, lineno, text, category, sourceline);}else {PyObject *check_fxn;PyObject *defaults;PyObject *res;if (PyMethod_Check(show_fxn))check_fxn = PyMethod_Function(show_fxn);else if (PyFunction_Check(show_fxn))check_fxn = show_fxn;else {PyErr_SetString(PyExc_TypeError,"warnings.showwarning() must be set to a ""function or method");Py_DECREF(show_fxn);goto cleanup;}defaults = PyFunction_GetDefaults(check_fxn);/* A proper implementation of warnings.showwarning() shouldhave at least two default arguments. */if ((defaults == NULL) || (PyTuple_Size(defaults) < 2)) {PyCodeObject *code = (PyCodeObject *)PyFunction_GetCode(check_fxn);if (!(code->co_flags & CO_VARARGS)) {if (PyErr_WarnEx(PyExc_DeprecationWarning, msg, 1) <0) {Py_DECREF(show_fxn);goto cleanup;}}}res = PyObject_CallFunctionObjArgs(show_fxn, message, category,filename, lineno_obj,NULL);Py_DECREF(show_fxn);Py_XDECREF(res);if (res == NULL)goto cleanup;}}}else /* if (rc == -1) */goto cleanup;return_none:result = Py_None;Py_INCREF(result);cleanup:Py_XDECREF(key);Py_XDECREF(text);Py_XDECREF(lineno_obj);Py_DECREF(module);Py_XDECREF(message);return result; /* Py_None or NULL. */}/* filename, module, and registry are new refs, globals is borrowed *//* Returns 0 on error (no new refs), 1 on success */static intsetup_context(Py_ssize_t stack_level, PyObject **filename, int *lineno,PyObject **module, PyObject **registry){PyObject *globals;/* Setup globals and lineno. */PyFrameObject *f = PyThreadState_GET()->frame;while (--stack_level > 0 && f != NULL)f = f->f_back;if (f == NULL) {globals = PyThreadState_Get()->interp->sysdict;*lineno = 1;}else {globals = f->f_globals;*lineno = PyCode_Addr2Line(f->f_code, f->f_lasti);}*module = NULL;/* Setup registry. */assert(globals != NULL);assert(PyDict_Check(globals));*registry = PyDict_GetItemString(globals, "__warningregistry__");if (*registry == NULL) {int rc;*registry = PyDict_New();if (*registry == NULL)return 0;rc = PyDict_SetItemString(globals, "__warningregistry__", *registry);if (rc < 0)goto handle_error;}elsePy_INCREF(*registry);/* Setup module. */*module = PyDict_GetItemString(globals, "__name__");if (*module == NULL) {*module = PyString_FromString("<string>");if (*module == NULL)goto handle_error;}elsePy_INCREF(*module);/* Setup filename. */*filename = PyDict_GetItemString(globals, "__file__");if (*filename != NULL) {Py_ssize_t len = PyString_Size(*filename);const char *file_str = PyString_AsString(*filename);if (file_str == NULL || (len < 0 && PyErr_Occurred()))goto handle_error;/* if filename.lower().endswith((".pyc", ".pyo")): */if (len >= 4 &&file_str[len-4] == '.' &&tolower(file_str[len-3]) == 'p' &&tolower(file_str[len-2]) == 'y' &&(tolower(file_str[len-1]) == 'c' ||tolower(file_str[len-1]) == 'o')){*filename = PyString_FromStringAndSize(file_str, len-1);if (*filename == NULL)goto handle_error;}elsePy_INCREF(*filename);}else {const char *module_str = PyString_AsString(*module);if (module_str && strcmp(module_str, "__main__") == 0) {PyObject *argv = PySys_GetObject("argv");if (argv != NULL && PyList_Size(argv) > 0) {int is_true;*filename = PyList_GetItem(argv, 0);Py_INCREF(*filename);/* If sys.argv[0] is false, then use '__main__'. */is_true = PyObject_IsTrue(*filename);if (is_true < 0) {Py_DECREF(*filename);goto handle_error;}else if (!is_true) {Py_DECREF(*filename);*filename = PyString_FromString("__main__");if (*filename == NULL)goto handle_error;}}else {/* embedded interpreters don't have sys.argv, see bug #839151 */*filename = PyString_FromString("__main__");if (*filename == NULL)goto handle_error;}}if (*filename == NULL) {*filename = *module;Py_INCREF(*filename);}}return 1;handle_error:/* filename not XDECREF'ed here as there is no way to jump here with adangling reference. */Py_XDECREF(*registry);Py_XDECREF(*module);return 0;}static PyObject *get_category(PyObject *message, PyObject *category){int rc;/* Get category. */rc = PyObject_IsInstance(message, PyExc_Warning);if (rc == -1)return NULL;if (rc == 1)category = (PyObject*)message->ob_type;else if (category == NULL)category = PyExc_UserWarning;/* Validate category. */rc = PyObject_IsSubclass(category, PyExc_Warning);if (rc == -1)return NULL;if (rc == 0) {PyErr_SetString(PyExc_ValueError,"category is not a subclass of Warning");return NULL;}return category;}static PyObject *do_warn(PyObject *message, PyObject *category, Py_ssize_t stack_level){PyObject *filename, *module, *registry, *res;int lineno;if (!setup_context(stack_level, &filename, &lineno, &module, ®istry))return NULL;res = warn_explicit(category, message, filename, lineno, module, registry,NULL);Py_DECREF(filename);Py_DECREF(registry);Py_DECREF(module);return res;}static PyObject *warnings_warn(PyObject *self, PyObject *args, PyObject *kwds){static char *kw_list[] = { "message", "category", "stacklevel", 0 };PyObject *message, *category = NULL;Py_ssize_t stack_level = 1;if (!PyArg_ParseTupleAndKeywords(args, kwds, "O|On:warn", kw_list,&message, &category, &stack_level))return NULL;category = get_category(message, category);if (category == NULL)return NULL;return do_warn(message, category, stack_level);}static PyObject *warnings_warn_explicit(PyObject *self, PyObject *args, PyObject *kwds){static char *kwd_list[] = {"message", "category", "filename", "lineno","module", "registry", "module_globals", 0};PyObject *message;PyObject *category;PyObject *filename;int lineno;PyObject *module = NULL;PyObject *registry = NULL;PyObject *module_globals = NULL;if (!PyArg_ParseTupleAndKeywords(args, kwds, "OOOi|OOO:warn_explicit",kwd_list, &message, &category, &filename, &lineno, &module,®istry, &module_globals))return NULL;if (module_globals) {static PyObject *get_source_name = NULL;static PyObject *splitlines_name = NULL;PyObject *loader;PyObject *module_name;PyObject *source;PyObject *source_list;PyObject *source_line;PyObject *returned;if (get_source_name == NULL) {get_source_name = PyString_InternFromString("get_source");if (!get_source_name)return NULL;}if (splitlines_name == NULL) {splitlines_name = PyString_InternFromString("splitlines");if (!splitlines_name)return NULL;}/* Check/get the requisite pieces needed for the loader. */loader = PyDict_GetItemString(module_globals, "__loader__");module_name = PyDict_GetItemString(module_globals, "__name__");if (loader == NULL || module_name == NULL)goto standard_call;/* Make sure the loader implements the optional get_source() method. */if (!PyObject_HasAttrString(loader, "get_source"))goto standard_call;/* Call get_source() to get the source code. */source = PyObject_CallMethodObjArgs(loader, get_source_name,module_name, NULL);if (!source)return NULL;else if (source == Py_None) {Py_DECREF(Py_None);goto standard_call;}/* Split the source into lines. */source_list = PyObject_CallMethodObjArgs(source, splitlines_name,NULL);Py_DECREF(source);if (!source_list)return NULL;/* Get the source line. */source_line = PyList_GetItem(source_list, lineno-1);if (!source_line) {Py_DECREF(source_list);return NULL;}/* Handle the warning. */returned = warn_explicit(category, message, filename, lineno, module,registry, source_line);Py_DECREF(source_list);return returned;}standard_call:return warn_explicit(category, message, filename, lineno, module,registry, NULL);}/* Function to issue a warning message; may raise an exception. */intPyErr_WarnEx(PyObject *category, const char *text, Py_ssize_t stack_level){PyObject *res;PyObject *message = PyString_FromString(text);if (message == NULL)return -1;if (category == NULL)category = PyExc_RuntimeWarning;res = do_warn(message, category, stack_level);Py_DECREF(message);if (res == NULL)return -1;Py_DECREF(res);return 0;}/* PyErr_Warn is only for backwards compatability and will be removed.Use PyErr_WarnEx instead. */#undef PyErr_WarnPyAPI_FUNC(int)PyErr_Warn(PyObject *category, char *text){return PyErr_WarnEx(category, text, 1);}/* Warning with explicit origin */intPyErr_WarnExplicit(PyObject *category, const char *text,const char *filename_str, int lineno,const char *module_str, PyObject *registry){PyObject *res;PyObject *message = PyString_FromString(text);PyObject *filename = PyString_FromString(filename_str);PyObject *module = NULL;int ret = -1;if (message == NULL || filename == NULL)goto exit;if (module_str != NULL) {module = PyString_FromString(module_str);if (module == NULL)goto exit;}if (category == NULL)category = PyExc_RuntimeWarning;res = warn_explicit(category, message, filename, lineno, module, registry,NULL);if (res == NULL)goto exit;Py_DECREF(res);ret = 0;exit:Py_XDECREF(message);Py_XDECREF(module);Py_XDECREF(filename);return ret;}PyDoc_STRVAR(warn_doc,"Issue a warning, or maybe ignore it or raise an exception.");PyDoc_STRVAR(warn_explicit_doc,"Low-level inferface to warnings functionality.");static PyMethodDef warnings_functions[] = {{"warn", (PyCFunction)warnings_warn, METH_VARARGS | METH_KEYWORDS,warn_doc},{"warn_explicit", (PyCFunction)warnings_warn_explicit,METH_VARARGS | METH_KEYWORDS, warn_explicit_doc},/* XXX(brett.cannon): add showwarning? *//* XXX(brett.cannon): Reasonable to add formatwarning? */{NULL, NULL} /* sentinel */};static PyObject *create_filter(PyObject *category, const char *action){static PyObject *ignore_str = NULL;static PyObject *error_str = NULL;static PyObject *default_str = NULL;PyObject *action_obj = NULL;PyObject *lineno, *result;if (!strcmp(action, "ignore")) {if (ignore_str == NULL) {ignore_str = PyString_InternFromString("ignore");if (ignore_str == NULL)return NULL;}action_obj = ignore_str;}else if (!strcmp(action, "error")) {if (error_str == NULL) {error_str = PyString_InternFromString("error");if (error_str == NULL)return NULL;}action_obj = error_str;}else if (!strcmp(action, "default")) {if (default_str == NULL) {default_str = PyString_InternFromString("default");if (default_str == NULL)return NULL;}action_obj = default_str;}else {Py_FatalError("unknown action");}/* This assumes the line number is zero for now. */lineno = PyInt_FromLong(0);if (lineno == NULL)return NULL;result = PyTuple_Pack(5, action_obj, Py_None, category, Py_None, lineno);Py_DECREF(lineno);return result;}static PyObject *init_filters(void){PyObject *filters = PyList_New(3);const char *bytes_action;if (filters == NULL)return NULL;PyList_SET_ITEM(filters, 0,create_filter(PyExc_PendingDeprecationWarning, "ignore"));PyList_SET_ITEM(filters, 1, create_filter(PyExc_ImportWarning, "ignore"));if (Py_BytesWarningFlag > 1)bytes_action = "error";else if (Py_BytesWarningFlag)bytes_action = "default";elsebytes_action = "ignore";PyList_SET_ITEM(filters, 2, create_filter(PyExc_BytesWarning,bytes_action));if (PyList_GET_ITEM(filters, 0) == NULL ||PyList_GET_ITEM(filters, 1) == NULL ||PyList_GET_ITEM(filters, 2) == NULL) {Py_DECREF(filters);return NULL;}return filters;}PyMODINIT_FUNC_PyWarnings_Init(void){PyObject *m, *default_action;m = Py_InitModule3(MODULE_NAME, warnings_functions, warnings__doc__);if (m == NULL)return;_filters = init_filters();if (_filters == NULL)return;Py_INCREF(_filters);if (PyModule_AddObject(m, "filters", _filters) < 0)return;_once_registry = PyDict_New();if (_once_registry == NULL)return;Py_INCREF(_once_registry);if (PyModule_AddObject(m, "once_registry", _once_registry) < 0)return;default_action = PyString_InternFromString("default");if (default_action == NULL)return;if (PyModule_AddObject(m, DEFAULT_ACTION_NAME, default_action) < 0)return;}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。