mirror of
				https://github.com/python/cpython.git
				synced 2025-10-25 10:44:55 +00:00 
			
		
		
		
	 1a8501c648
			
		
	
	
		1a8501c648
		
	
	
	
	
		
			
			svn+ssh://pythondev@svn.python.org/python/trunk ........ r66748 | christian.heimes | 2008-10-02 21:47:50 +0200 (Thu, 02 Oct 2008) | 1 line Fixed a couple more C99 comments and one occurence of inline. ........ + another // comment in bytesobject
		
			
				
	
	
		
			903 lines
		
	
	
	
		
			25 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			903 lines
		
	
	
	
		
			25 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 = 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;
 | |
| }
 | |
| 
 | |
| 
 | |
| /* 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 = PyLong_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 _PyUnicode_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 _PyUnicode_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 PyUnicode_FromString("<unknown>");
 | |
| 
 | |
|     mod_str = _PyUnicode_AsString(filename);
 | |
|     if (mod_str == NULL)
 | |
| 	    return NULL;
 | |
|     len = PyUnicode_GetSize(filename);
 | |
|     if (len < 0)
 | |
|         return NULL;
 | |
|     if (len >= 3 &&
 | |
| 	strncmp(mod_str + (len - 3), ".py", 3) == 0) {
 | |
|         module = PyUnicode_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 = 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];
 | |
| 
 | |
|     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 = _PyUnicode_AsString(sourceline);
 | |
| 	if (source_line_str == NULL)
 | |
| 		return;
 | |
|         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
 | |
|         if (_Py_DisplaySourceLine(f_stderr, _PyUnicode_AsString(filename),
 | |
|                               lineno, 2) < 0)
 | |
| 		return;
 | |
|     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;
 | |
|     }
 | |
|     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);
 | |
|         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 (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 && 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 = _PyUnicode_AsString(to_str);
 | |
| 		if (err_str == NULL)
 | |
| 			goto cleanup;
 | |
| 	    }
 | |
|             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 {
 | |
|             PyObject *res;
 | |
| 
 | |
|             if (!PyMethod_Check(show_fxn) && !PyFunction_Check(show_fxn)) {
 | |
|                 PyErr_SetString(PyExc_TypeError,
 | |
|                                 "warnings.showwarning() must be set to a "
 | |
|                                 "function or method");
 | |
|                 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 = 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 = PyUnicode_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 = PyUnicode_GetSize(*filename);
 | |
|         const char *file_str = _PyUnicode_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 = PyUnicode_FromStringAndSize(file_str, len-1);
 | |
| 	        if (*filename == NULL)
 | |
| 		        goto handle_error;
 | |
| 	    }
 | |
| 	    else
 | |
|             Py_INCREF(*filename);
 | |
|     }
 | |
|     else {
 | |
|         const char *module_str = _PyUnicode_AsString(*module);
 | |
| 	if (module_str == NULL)
 | |
| 		goto handle_error;
 | |
|         if (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 = 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 = 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 = PyUnicode_InternFromString("get_source");
 | |
|             if (!get_source_name)
 | |
|                 return NULL;
 | |
|         }
 | |
|         if (splitlines_name == NULL) {
 | |
|             splitlines_name = PyUnicode_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 = PyUnicode_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 = PyUnicode_FromString(text);
 | |
|     PyObject *filename = PyUnicode_FromString(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;
 | |
|     }
 | |
| 
 | |
|     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 = 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 {
 | |
|         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(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;
 | |
| }
 | |
| 
 | |
| static struct PyModuleDef warningsmodule = {
 | |
| 	PyModuleDef_HEAD_INIT,
 | |
| 	MODULE_NAME,
 | |
| 	warnings__doc__,
 | |
| 	0,
 | |
| 	warnings_functions,
 | |
| 	NULL,
 | |
| 	NULL,
 | |
| 	NULL,
 | |
| 	NULL
 | |
| };
 | |
| 
 | |
| 
 | |
| PyMODINIT_FUNC
 | |
| _PyWarnings_Init(void)
 | |
| {
 | |
|     PyObject *m, *default_action;
 | |
| 
 | |
|     m = PyModule_Create(&warningsmodule);
 | |
|     if (m == NULL)
 | |
|         return NULL;
 | |
| 
 | |
|     _filters = init_filters();
 | |
|     if (_filters == NULL)
 | |
|         return NULL;
 | |
|     Py_INCREF(_filters);
 | |
|     if (PyModule_AddObject(m, "filters", _filters) < 0)
 | |
|         return NULL;
 | |
| 
 | |
|     _once_registry = PyDict_New();
 | |
|     if (_once_registry == NULL)
 | |
|         return NULL;
 | |
|     Py_INCREF(_once_registry);
 | |
|     if (PyModule_AddObject(m, "once_registry", _once_registry) < 0)
 | |
|         return NULL;
 | |
| 
 | |
|     default_action = PyUnicode_InternFromString("default");
 | |
|     if (default_action == NULL)
 | |
|         return NULL;
 | |
|     if (PyModule_AddObject(m, DEFAULT_ACTION_NAME, default_action) < 0)
 | |
|         return NULL;
 | |
|     return m;
 | |
| }
 |