#include "Python.h"

#include "frameobject.h"

#define MODULE_NAME "_warnings"

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 PyObject *_default_action; /* String */

_Py_IDENTIFIER(argv);

_Py_IDENTIFIER(stderr);

static int

check_matched(PyObject *obj, PyObject *arg)

{

PyObject *result;

_Py_IDENTIFIER(match);

int rc;

if (obj == Py_None)

return 1;

result = _PyObject_CallMethodId(obj, &PyId_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 = PyUnicode_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;

}

static PyObject *

get_default_action(void)

{

PyObject *default_action;

default_action = get_warnings_attr("defaultaction");

if (default_action == NULL) {

if (PyErr_Occurred()) {

return NULL;

}

return _default_action;

}

Py_DECREF(_default_action);

_default_action = default_action;

return default_action;

}

/* The item is a borrowed reference. */

static PyObject*

get_filter(PyObject *category, PyObject *text, Py_ssize_t lineno,

PyObject *module, PyObject **item)

{

PyObject *action;

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 (_filters == NULL || !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);

if (good_msg == -1)

return NULL;

good_mod = check_matched(mod, module);

if (good_mod == -1)

return NULL;

is_subclass = PyObject_IsSubclass(category, cat);

if (is_subclass == -1)

return NULL;

ln = PyLong_AsSsize_t(ln_obj);

if (ln == -1 && PyErr_Occurred())

return NULL;

if (good_msg && is_subclass && good_mod && (ln == 0 || lineno == ln))

return action;

}

action = get_default_action();

if (action != NULL)

return action;

PyErr_SetString(PyExc_ValueError,

MODULE_NAME ".defaultaction 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;

int kind;

void *data;

Py_ssize_t len;

len = PyUnicode_GetLength(filename);

if (len < 0)

return NULL;

if (len == 0)

return PyUnicode_FromString("<unknown>");

kind = PyUnicode_KIND(filename);

data = PyUnicode_DATA(filename);

/* if filename.endswith(".py"): */

if (len >= 3 &&

PyUnicode_READ(kind, data, len-3) == '.' &&

PyUnicode_READ(kind, data, len-2) == 'p' &&

PyUnicode_READ(kind, data, len-1) == 'y')

{

module = PyUnicode_Substring(filename, 0, 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 = PyLong_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];

_Py_IDENTIFIER(__name__);

PyOS_snprintf(lineno_str, sizeof(lineno_str), ":%d: ", lineno);

name = _PyObject_GetAttrId(category, &PyId___name__);

if (name == NULL) /* XXX Can an object lack a '__name__' attribute? */

goto error;

f_stderr = _PySys_GetObjectId(&PyId_stderr);

if (f_stderr == NULL) {

fprintf(stderr, "lost sys.stderr\n");

goto error;

}

/* Print "filename:lineno: category: text\n" */

if (PyFile_WriteObject(filename, f_stderr, Py_PRINT_RAW) < 0)

goto error;

if (PyFile_WriteString(lineno_str, f_stderr) < 0)

goto error;

if (PyFile_WriteObject(name, f_stderr, Py_PRINT_RAW) < 0)

goto error;

if (PyFile_WriteString(": ", f_stderr) < 0)

goto error;

if (PyFile_WriteObject(text, f_stderr, Py_PRINT_RAW) < 0)

goto error;

if (PyFile_WriteString("\n", f_stderr) < 0)

goto error;

Py_CLEAR(name);

/* Print " source_line\n" */

if (sourceline) {

int kind;

void *data;

Py_ssize_t i, len;

Py_UCS4 ch;

PyObject *truncated;

if (PyUnicode_READY(sourceline) < 1)

goto error;

kind = PyUnicode_KIND(sourceline);

data = PyUnicode_DATA(sourceline);

len = PyUnicode_GET_LENGTH(sourceline);

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

ch = PyUnicode_READ(kind, data, i);

if (ch != ' ' && ch != '\t' && ch != '\014')

break;

}

truncated = PyUnicode_Substring(sourceline, i, len);

if (truncated == NULL)

goto error;

PyFile_WriteObject(sourceline, f_stderr, Py_PRINT_RAW);

Py_DECREF(truncated);

PyFile_WriteString("\n", f_stderr);

}

else {

_Py_DisplaySourceLine(f_stderr, filename, lineno, 2);

}

error:

Py_XDECREF(name);

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;

PyObject *action;

int rc;

/* module can be None if a warning is emitted late during Python shutdown.

In this case, the Python warnings module was probably unloaded, filters

are no more available to choose as action. It is safer to ignore the

warning and do nothing. */

if (module == Py_None)

Py_RETURN_NONE;

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;

}

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);

if (text == NULL)

goto cleanup;

category = (PyObject*)message->ob_type;

}

else {

text = message;

message = PyObject_CallFunction(category, "O", message);

if (message == NULL)

goto cleanup;

}

lineno_obj = PyLong_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 (PyUnicode_CompareWithASCIIString(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 (PyUnicode_CompareWithASCIIString(action, "always") != 0) {

if (registry != NULL && registry != Py_None &&

PyDict_SetItem(registry, key, Py_True) < 0)

goto cleanup;

else if (PyUnicode_CompareWithASCIIString(action, "ignore") == 0)

goto return_none;

else if (PyUnicode_CompareWithASCIIString(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 (PyUnicode_CompareWithASCIIString(action, "module") == 0) {

/* registry[(text, category, 0)] = 1 */

if (registry != NULL && registry != Py_None)

rc = update_registry(registry, text, category, 0);

}

else if (PyUnicode_CompareWithASCIIString(action, "default") != 0) {

PyErr_Format(PyExc_RuntimeError,

"Unrecognized action (%R) in warnings.filters:\n %R",

action, item);

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 {

PyObject *res;

if (!PyCallable_Check(show_fxn)) {

PyErr_SetString(PyExc_TypeError,

"warnings.showwarning() must be set to a "

"callable");

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 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 = PyFrame_GetLineNumber(f);

}

*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 = PyUnicode_FromString("<string>");

if (*module == NULL)

goto handle_error;

}

else

Py_INCREF(*module);

/* Setup filename. */

*filename = PyDict_GetItemString(globals, "__file__");

if (*filename != NULL && PyUnicode_Check(*filename)) {

Py_ssize_t len;

int kind;

void *data;

if (PyUnicode_READY(*filename))

goto handle_error;

len = PyUnicode_GetLength(*filename);

kind = PyUnicode_KIND(*filename);

data = PyUnicode_DATA(*filename);

#define ascii_lower(c) ((c <= 127) ? Py_TOLOWER(c) : 0)

/* if filename.lower().endswith((".pyc", ".pyo")): */

if (len >= 4 &&

PyUnicode_READ(kind, data, len-4) == '.' &&

ascii_lower(PyUnicode_READ(kind, data, len-3)) == 'p' &&

ascii_lower(PyUnicode_READ(kind, data, len-2)) == 'y' &&

(ascii_lower(PyUnicode_READ(kind, data, len-1)) == 'c' ||

ascii_lower(PyUnicode_READ(kind, data, len-1)) == 'o'))

{

*filename = PyUnicode_Substring(*filename, 0,

PyUnicode_GET_LENGTH(*filename)-1);

if (*filename == NULL)

goto handle_error;

}

else

Py_INCREF(*filename);

}

else {

*filename = NULL;

if (*module != Py_None && PyUnicode_CompareWithASCIIString(*module, "__main__") == 0) {

PyObject *argv = _PySys_GetObjectId(&PyId_argv);

/* PyList_Check() is needed because sys.argv is set to None during

Python finalization */

if (argv != NULL && PyList_Check(argv) && 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 = PyUnicode_FromString("__main__");

if (*filename == NULL)

goto handle_error;

}

}

else {

/* embedded interpreters don't have sys.argv, see bug #839151 */

*filename = PyUnicode_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 == Py_None)

category = PyExc_UserWarning;

/* Validate category. */

rc = PyObject_IsSubclass(category, PyExc_Warning);

/* category is not a subclass of PyExc_Warning or

PyObject_IsSubclass raised an error */

if (rc == -1 || rc == 0) {

PyErr_Format(PyExc_TypeError,

"category must be a Warning subclass, not '%s'",

Py_TYPE(category)->tp_name);

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, &registry))

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, "OOUi|OOO:warn_explicit",

kwd_list, &message, &category, &filename, &lineno, &module,

&registry, &module_globals))

return NULL;

if (module_globals) {

_Py_IDENTIFIER(get_source);

_Py_IDENTIFIER(splitlines);

PyObject *tmp;

PyObject *loader;

PyObject *module_name;

PyObject *source;

PyObject *source_list;

PyObject *source_line;

PyObject *returned;

if ((tmp = _PyUnicode_FromId(&PyId_get_source)) == NULL)

return NULL;

if ((tmp = _PyUnicode_FromId(&PyId_splitlines)) == NULL)

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_HasAttrId(loader, &PyId_get_source))

goto standard_call;

/* Call get_source() to get the source code. */

source = PyObject_CallMethodObjArgs(loader, PyId_get_source.object,

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,

PyId_splitlines.object,

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. */

static int

warn_unicode(PyObject *category, PyObject *message,

Py_ssize_t stack_level)

{

PyObject *res;

if (category == NULL)

category = PyExc_RuntimeWarning;

res = do_warn(message, category, stack_level);

if (res == NULL)

return -1;

Py_DECREF(res);

return 0;

}

int

PyErr_WarnFormat(PyObject *category, Py_ssize_t stack_level,

const char *format, ...)

{

int ret;

PyObject *message;

va_list vargs;

#ifdef HAVE_STDARG_PROTOTYPES

va_start(vargs, format);

#else

va_start(vargs);

#endif

message = PyUnicode_FromFormatV(format, vargs);

if (message != NULL) {

ret = warn_unicode(category, message, stack_level);

Py_DECREF(message);

}

else

ret = -1;

va_end(vargs);

return ret;

}

int

PyErr_WarnEx(PyObject *category, const char *text, Py_ssize_t stack_level)

{

int ret;

PyObject *message = PyUnicode_FromString(text);

if (message == NULL)

return -1;

ret = warn_unicode(category, message, stack_level);

Py_DECREF(message);

return ret;

}

/* PyErr_Warn is only for backwards compatibility 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_WarnExplicitObject(PyObject *category, PyObject *message,

PyObject *filename, int lineno,

PyObject *module, PyObject *registry)

{

PyObject *res;

if (category == NULL)

category = PyExc_RuntimeWarning;

res = warn_explicit(category, message, filename, lineno,

module, registry, NULL);

if (res == NULL)

return -1;

Py_DECREF(res);

return 0;

}

int

PyErr_WarnExplicit(PyObject *category, const char *text,

const char *filename_str, int lineno,

const char *module_str, PyObject *registry)

{

PyObject *message = PyUnicode_FromString(text);

PyObject *filename = PyUnicode_DecodeFSDefault(filename_str);

PyObject *module = NULL;

int ret = -1;

if (message == NULL || filename == NULL)

goto exit;

if (module_str != NULL) {

module = PyUnicode_FromString(module_str);

if (module == NULL)

goto exit;

}

ret = PyErr_WarnExplicitObject(category, message, filename, lineno,

module, registry);

exit:

Py_XDECREF(message);

Py_XDECREF(module);

Py_XDECREF(filename);

return ret;

}

int

PyErr_WarnExplicitFormat(PyObject *category,

const char *filename_str, int lineno,

const char *module_str, PyObject *registry,

const char *format, ...)

{

PyObject *message;

PyObject *module = NULL;

PyObject *filename = PyUnicode_DecodeFSDefault(filename_str);

int ret = -1;

va_list vargs;

if (filename == NULL)

goto exit;

if (module_str != NULL) {

module = PyUnicode_FromString(module_str);

if (module == NULL)

goto exit;

}

#ifdef HAVE_STDARG_PROTOTYPES

va_start(vargs, format);

#else

va_start(vargs);

#endif

message = PyUnicode_FromFormatV(format, vargs);

if (message != NULL) {

PyObject *res;

res = warn_explicit(category, message, filename, lineno,

module, registry, NULL);

Py_DECREF(message);

if (res != NULL) {

Py_DECREF(res);

ret = 0;

}

}

va_end(vargs);

exit:

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;

static PyObject *always_str = NULL;

PyObject *action_obj = NULL;

PyObject *lineno, *result;

if (!strcmp(action, "ignore")) {

if (ignore_str == NULL) {

ignore_str = PyUnicode_InternFromString("ignore");

if (ignore_str == NULL)

return NULL;

}

action_obj = ignore_str;

}

else if (!strcmp(action, "error")) {

if (error_str == NULL) {

error_str = PyUnicode_InternFromString("error");

if (error_str == NULL)

return NULL;

}

action_obj = error_str;

}

else if (!strcmp(action, "default")) {

if (default_str == NULL) {

default_str = PyUnicode_InternFromString("default");

if (default_str == NULL)

return NULL;

}

action_obj = default_str;

}

else if (!strcmp(action, "always")) {

if (always_str == NULL) {

always_str = PyUnicode_InternFromString("always");

if (always_str == NULL)

return NULL;

}

action_obj = always_str;

}

else {

Py_FatalError("unknown action");

}

/* This assumes the line number is zero for now. */

lineno = PyLong_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(5);

unsigned int pos = 0; /* Post-incremented in each use. */

unsigned int x;

const char *bytes_action, *resource_action;

if (filters == NULL)

return NULL;

PyList_SET_ITEM(filters, pos++,

create_filter(PyExc_DeprecationWarning, "ignore"));

PyList_SET_ITEM(filters, pos++,

create_filter(PyExc_PendingDeprecationWarning, "ignore"));

PyList_SET_ITEM(filters, pos++,

create_filter(PyExc_ImportWarning, "ignore"));

if (Py_BytesWarningFlag > 1)

bytes_action = "error";