static int igraphmodule_i_filehandle_init_pypy_2(igraphmodule_filehandle_t* handle,
PyObject* object, char* mode) {
int fp;
PyObject* fpobj;
char* fname;
if (object == 0 ||
(!PyBaseString_Check(object) && !PyFile_Check(object))) {
PyErr_SetString(PyExc_TypeError, "string or file handle expected");
return 1;
}
handle->need_close = 0;
if (PyBaseString_Check(object)) {
/* We have received a string; we need to open the file denoted by this
* string now and mark that we opened the file ourselves (so we need
* to close it when igraphmodule_filehandle_destroy is invoked). */
handle->object = PyFile_FromString(PyString_AsString(object), mode);
if (handle->object == 0) {
/* Could not open the file; just return an error code because an
* exception was raised already */
return 1;
}
/* Remember that we need to close the file ourselves */
handle->need_close = 1;
} else {
/* This is probably a file-like object; store a reference for it and
* we will handle it later */
handle->object = object;
Py_INCREF(handle->object);
}
/* PyPy does not have PyFile_AsFile, so we will try to access the file
* descriptor instead by calling its fileno() method and then opening the
* file handle with fdopen */
fpobj = PyObject_CallMethod(handle->object, "fileno", 0);
if (fpobj == 0 || !PyInt_Check(fpobj)) {
if (fpobj != 0) {
Py_DECREF(fpobj);
}
igraphmodule_filehandle_destroy(handle);
/* This already called Py_DECREF(handle->object), no need to call it.
* Also, an exception was raised by PyObject_CallMethod so no need to
* raise one ourselves */
return 1;
}
fp = (int)PyInt_AsLong(fpobj);
Py_DECREF(fpobj);
handle->fp = fdopen(fp, mode);
if (handle->fp == 0) {
igraphmodule_filehandle_destroy(handle);
/* This already called Py_DECREF(handle->object), no need to call it */
PyErr_SetString(PyExc_RuntimeError, "fdopen() failed unexpectedly");
return 1;
}
return 0;
}
static int igraphmodule_i_filehandle_init_cpython_3(igraphmodule_filehandle_t* handle,
PyObject* object, char* mode) {
int fp;
if (object == 0 || PyLong_Check(object)) {
PyErr_SetString(PyExc_TypeError, "string or file-like object expected");
return 1;
}
handle->need_close = 0;
handle->object = 0;
if (PyBaseString_Check(object)) {
/* We have received a string; we need to open the file denoted by this
* string now and mark that we opened the file ourselves (so we need
* to close it when igraphmodule_filehandle_destroy is invoked). */
handle->object = PyFile_FromObject(object, mode);
if (handle->object == 0) {
/* Could not open the file; just return an error code because an
* exception was raised already */
return 1;
}
/* Remember that we need to close the file ourselves */
handle->need_close = 1;
} else {
/* This is probably a file-like object; store a reference for it and
* we will handle it later */
handle->object = object;
Py_INCREF(handle->object);
}
/* At this stage, handle->object is something we can handle.
* We have to call PyObject_AsFileDescriptor instead
* and then fdopen() it to get the corresponding FILE* object.
*/
fp = PyObject_AsFileDescriptor(handle->object);
if (fp == -1) {
igraphmodule_filehandle_destroy(handle);
/* This already called Py_DECREF(handle->object), no need to call it */
return 1;
}
handle->fp = fdopen(fp, mode);
if (handle->fp == 0) {
igraphmodule_filehandle_destroy(handle);
/* This already called Py_DECREF(handle->object), no need to call it */
PyErr_SetString(PyExc_RuntimeError, "fdopen() failed unexpectedly");
return 1;
}
return 0;
}
static int igraphmodule_i_filehandle_init_cpython_2(igraphmodule_filehandle_t* handle,
PyObject* object, char* mode) {
FILE* fp;
PyObject* fileno_method;
PyObject* fileno_result;
int fileno = -1;
if (object == 0) {
PyErr_SetString(PyExc_TypeError, "trying to convert a null object "
"to a file handle");
return 1;
}
handle->object = 0;
handle->need_close = 0;
if (PyBaseString_Check(object)) {
/* We have received a string; we need to open the file denoted by this
* string now and mark that we opened the file ourselves (so we need
* to close it when igraphmodule_filehandle_destroy is invoked). */
handle->object = PyFile_FromString(PyString_AsString(object), mode);
if (handle->object == 0) {
/* Could not open the file; just return an error code because an
* exception was raised already */
return 1;
}
/* Remember that we need to close the file ourselves */
handle->need_close = 1;
/* Get a FILE* object from the file */
fp = PyFile_AsFile(handle->object);
} else if (PyFile_Check(object)) {
/* This is a file-like object; store a reference for it and
* we will handle it later */
handle->object = object;
Py_INCREF(handle->object);
/* Get a FILE* object from the file */
fp = PyFile_AsFile(handle->object);
} else {
/* Check whether the object has a fileno() method. If so, we convert
* that to a file descriptor and then fdopen() it */
fileno_method = PyObject_GetAttrString(object, "fileno");
if (fileno_method != 0) {
if (PyCallable_Check(fileno_method)) {
fileno_result = PyObject_CallObject(fileno_method, 0);
Py_DECREF(fileno_method);
if (fileno_result != 0) {
if (PyInt_Check(fileno_result)) {
fileno = (int)PyInt_AsLong(fileno_result);
Py_DECREF(fileno_result);
} else {
Py_DECREF(fileno_result);
PyErr_SetString(PyExc_TypeError,
"fileno() method of file-like object should return "
"an integer");
return 1;
}
} else {
/* Exception set already by PyObject_CallObject() */
return 1;
}
} else {
Py_DECREF(fileno_method);
PyErr_SetString(PyExc_TypeError,
"fileno() attribute of file-like object must be callable");
return 1;
}
} else {
PyErr_SetString(PyExc_TypeError, "expected filename or file-like object");
return 1;
}
if (fileno > 0) {
fp = fdopen(fileno, mode);
} else {
PyErr_SetString(PyExc_ValueError, "fileno() method returned invalid "
"file descriptor");
return 1;
}
}
handle->fp = fp;
if (handle->fp == 0) {
igraphmodule_filehandle_destroy(handle);
/* This already called Py_DECREF(handle->object), no need to call it */
PyErr_SetString(PyExc_RuntimeError, "PyFile_AsFile() failed unexpectedly");
return 1;
}
return 0;
}
