self.assertEqual(exc.name, orig.name)
self.assertEqual(exc.path, orig.path)
+ def test_repr(self):
+ exc = ImportError()
+ self.assertEqual(repr(exc), "ImportError()")
+
+ exc = ImportError('test')
+ self.assertEqual(repr(exc), "ImportError('test')")
+
+ exc = ImportError('test', 'case')
+ self.assertEqual(repr(exc), "ImportError('test', 'case')")
+
+ exc = ImportError(name='somemodule')
+ self.assertEqual(repr(exc), "ImportError(name='somemodule')")
+
+ exc = ImportError('test', name='somemodule')
+ self.assertEqual(repr(exc), "ImportError('test', name='somemodule')")
+
+ exc = ImportError(path='somepath')
+ self.assertEqual(repr(exc), "ImportError(path='somepath')")
+
+ exc = ImportError('test', path='somepath')
+ self.assertEqual(repr(exc), "ImportError('test', path='somepath')")
+
+ exc = ImportError(name='somename', path='somepath')
+ self.assertEqual(repr(exc),
+ "ImportError(name='somename', path='somepath')")
+
+ exc = ImportError('test', name='somename', path='somepath')
+ self.assertEqual(repr(exc),
+ "ImportError('test', name='somename', path='somepath')")
+
+ exc = ModuleNotFoundError('test', name='somename', path='somepath')
+ self.assertEqual(repr(exc),
+ "ModuleNotFoundError('test', name='somename', path='somepath')")
+
+ def test_ModuleNotFoundError_repr_with_failed_import(self):
+ with self.assertRaises(ModuleNotFoundError) as cm:
+ import does_not_exist # type: ignore[import] # noqa: F401
+
+ self.assertEqual(cm.exception.name, "does_not_exist")
+ self.assertIsNone(cm.exception.path)
+
+ self.assertEqual(repr(cm.exception),
+ "ModuleNotFoundError(\"No module named 'does_not_exist'\", name='does_not_exist')")
+
def run_script(source):
if isinstance(source, str):
return res;
}
+static PyObject *
+ImportError_repr(PyObject *self)
+{
+ int hasargs = PyTuple_GET_SIZE(((PyBaseExceptionObject *)self)->args) != 0;
+ PyImportErrorObject *exc = PyImportErrorObject_CAST(self);
+ if (exc->name == NULL && exc->path == NULL) {
+ return BaseException_repr(self);
+ }
+ PyUnicodeWriter *writer = PyUnicodeWriter_Create(0);
+ if (writer == NULL) {
+ goto error;
+ }
+ PyObject *r = BaseException_repr(self);
+ if (r == NULL) {
+ goto error;
+ }
+ if (PyUnicodeWriter_WriteSubstring(
+ writer, r, 0, PyUnicode_GET_LENGTH(r) - 1) < 0)
+ {
+ Py_DECREF(r);
+ goto error;
+ }
+ Py_DECREF(r);
+ if (exc->name) {
+ if (hasargs) {
+ if (PyUnicodeWriter_WriteASCII(writer, ", ", 2) < 0) {
+ goto error;
+ }
+ }
+ if (PyUnicodeWriter_Format(writer, "name=%R", exc->name) < 0) {
+ goto error;
+ }
+ hasargs = 1;
+ }
+ if (exc->path) {
+ if (hasargs) {
+ if (PyUnicodeWriter_WriteASCII(writer, ", ", 2) < 0) {
+ goto error;
+ }
+ }
+ if (PyUnicodeWriter_Format(writer, "path=%R", exc->path) < 0) {
+ goto error;
+ }
+ }
+
+ if (PyUnicodeWriter_WriteChar(writer, ')') < 0) {
+ goto error;
+ }
+
+ return PyUnicodeWriter_Finish(writer);
+
+error:
+ PyUnicodeWriter_Discard(writer);
+ return NULL;
+}
+
static PyMemberDef ImportError_members[] = {
{"msg", _Py_T_OBJECT, offsetof(PyImportErrorObject, msg), 0,
PyDoc_STR("exception message")},
{NULL}
};
-ComplexExtendsException(PyExc_Exception, ImportError,
- ImportError, 0 /* new */,
- ImportError_methods, ImportError_members,
- 0 /* getset */, ImportError_str,
- "Import can't find module, or can't find name in "
- "module.");
+static PyTypeObject _PyExc_ImportError = {
+ PyVarObject_HEAD_INIT(NULL, 0)
+ .tp_name = "ImportError",
+ .tp_basicsize = sizeof(PyImportErrorObject),
+ .tp_dealloc = ImportError_dealloc,
+ .tp_repr = ImportError_repr,
+ .tp_str = ImportError_str,
+ .tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC,
+ .tp_doc = PyDoc_STR(
+ "Import can't find module, "
+ "or can't find name in module."),
+ .tp_traverse = ImportError_traverse,
+ .tp_clear = ImportError_clear,
+ .tp_methods = ImportError_methods,
+ .tp_members = ImportError_members,
+ .tp_base = &_PyExc_Exception,
+ .tp_dictoffset = offsetof(PyImportErrorObject, dict),
+ .tp_init = ImportError_init,
+};
+PyObject *PyExc_ImportError = (PyObject *)&_PyExc_ImportError;
/*
* ModuleNotFoundError extends ImportError