mirror of https://github.com/python/cpython
901 lines
26 KiB
C
901 lines
26 KiB
C
#include "Python.h"
|
|
#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 int
|
|
check_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 = PyBytes_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 PyBytes_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 PyBytes_AsString(action);
|
|
|
|
PyErr_SetString(PyExc_ValueError,
|
|
MODULE_NAME "." DEFAULT_ACTION_NAME " not found");
|
|
return NULL;
|
|
}
|
|
|
|
static int
|
|
already_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 PyBytes_FromString("<unknown>");
|
|
|
|
mod_str = PyBytes_AsString(filename);
|
|
if (mod_str == NULL)
|
|
return NULL;
|
|
len = PyBytes_Size(filename);
|
|
if (len < 0)
|
|
return NULL;
|
|
if (len >= 3 &&
|
|
strncmp(mod_str + (len - 3), ".py", 3) == 0) {
|
|
module = PyBytes_FromStringAndSize(mod_str, len-3);
|
|
}
|
|
else {
|
|
module = filename;
|
|
Py_INCREF(module);
|
|
}
|
|
return module;
|
|
}
|
|
|
|
static int
|
|
update_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);
|
|
}
|
|
else
|
|
altkey = PyTuple_Pack(2, text, category);
|
|
|
|
rc = already_warned(registry, altkey, 1);
|
|
Py_XDECREF(zero);
|
|
Py_XDECREF(altkey);
|
|
return rc;
|
|
}
|
|
|
|
static void
|
|
show_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" */
|
|
PyFile_WriteString(" ", f_stderr);
|
|
if (sourceline) {
|
|
char *source_line_str = PyBytes_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, PyBytes_AS_STRING(filename), lineno);
|
|
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;
|
|
|
|
/* Normalize module. */
|
|
if (module == NULL) {
|
|
module = normalize_module(filename);
|
|
if (module == NULL)
|
|
return NULL;
|
|
}
|
|
else
|
|
Py_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);
|
|
}
|
|
|
|
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) {
|
|
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 action
|
|
is "always". */
|
|
rc = 0;
|
|
if (strcmp(action, "always") != 0) {
|
|
if (registry != NULL && 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 = 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)
|
|
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 = PyBytes_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 = PyBytes_AS_STRING(text);
|
|
|
|
if (strcmp(msg, text_char) == 0) {
|
|
/* Prevent infinite recursion by using built-in implementation
|
|
of 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() should
|
|
have at least two default arguments. */
|
|
if ((defaults == NULL) || (PyTuple_Size(defaults) < 2)) {
|
|
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_DECREF(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 int
|
|
setup_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;
|
|
}
|
|
else
|
|
Py_INCREF(*registry);
|
|
|
|
/* Setup module. */
|
|
*module = PyDict_GetItemString(globals, "__name__");
|
|
if (*module == NULL) {
|
|
*module = PyBytes_FromString("<string>");
|
|
if (*module == NULL)
|
|
goto handle_error;
|
|
}
|
|
else
|
|
Py_INCREF(*module);
|
|
|
|
/* Setup filename. */
|
|
*filename = PyDict_GetItemString(globals, "__file__");
|
|
if (*filename != NULL) {
|
|
Py_ssize_t len = PyBytes_Size(*filename);
|
|
const char *file_str = PyBytes_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 = PyBytes_FromStringAndSize(file_str, len-1);
|
|
if (*filename == NULL)
|
|
goto handle_error;
|
|
}
|
|
else
|
|
Py_INCREF(*filename);
|
|
}
|
|
else {
|
|
const char *module_str = PyBytes_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 = PyBytes_FromString("__main__");
|
|
if (*filename == NULL)
|
|
goto handle_error;
|
|
}
|
|
}
|
|
else {
|
|
/* embedded interpreters don't have sys.argv, see bug #839151 */
|
|
*filename = PyBytes_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 a
|
|
dangling 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 = PyBytes_InternFromString("get_source");
|
|
if (!get_source_name)
|
|
return NULL;
|
|
}
|
|
if (splitlines_name == NULL) {
|
|
splitlines_name = PyBytes_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. */
|
|
int
|
|
PyErr_WarnEx(PyObject *category, const char *text, Py_ssize_t stack_level)
|
|
{
|
|
PyObject *res;
|
|
PyObject *message = PyBytes_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_Warn
|
|
|
|
PyAPI_FUNC(int)
|
|
PyErr_Warn(PyObject *category, char *text)
|
|
{
|
|
return PyErr_WarnEx(category, text, 1);
|
|
}
|
|
|
|
/* Warning with explicit origin */
|
|
int
|
|
PyErr_WarnExplicit(PyObject *category, const char *text,
|
|
const char *filename_str, int lineno,
|
|
const char *module_str, PyObject *registry)
|
|
{
|
|
PyObject *res;
|
|
PyObject *message = PyBytes_FromString(text);
|
|
PyObject *filename = PyBytes_FromString(filename_str);
|
|
PyObject *module = NULL;
|
|
int ret = -1;
|
|
|
|
if (message == NULL || filename == NULL)
|
|
goto exit;
|
|
if (module_str != NULL) {
|
|
module = PyBytes_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 = PyBytes_InternFromString("ignore");
|
|
if (ignore_str == NULL)
|
|
return NULL;
|
|
}
|
|
action_obj = ignore_str;
|
|
}
|
|
else if (!strcmp(action, "error")) {
|
|
if (error_str == NULL) {
|
|
error_str = PyBytes_InternFromString("error");
|
|
if (error_str == NULL)
|
|
return NULL;
|
|
}
|
|
action_obj = error_str;
|
|
}
|
|
else if (!strcmp(action, "default")) {
|
|
if (default_str == NULL) {
|
|
default_str = PyBytes_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";
|
|
else
|
|
bytes_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 = PyBytes_InternFromString("default");
|
|
if (default_action == NULL)
|
|
return;
|
|
if (PyModule_AddObject(m, DEFAULT_ACTION_NAME, default_action) < 0)
|
|
return;
|
|
}
|