From 0df002c45b0bf2bf23bb8c0a395e86f907a94d81 Mon Sep 17 00:00:00 2001 From: Guido van Rossum Date: Sun, 27 Aug 2000 19:21:52 +0000 Subject: [PATCH] Add three new APIs: PyRun_AnyFileEx(), PyRun_SimpleFileEx(), PyRun_FileEx(). These are the same as their non-Ex counterparts but have an extra argument, a flag telling them to close the file when done. Then this is used by Py_Main() and execfile() to close the file after it is parsed but before it is executed. Adding APIs seems strange given the feature freeze but it's the only way I see to close the bug report without incompatible changes. [ Bug #110616 ] source file stays open after parsing is done (PR#209) --- Include/pythonrun.h | 4 ++++ Modules/main.c | 7 +++---- Python/bltinmodule.c | 5 +---- Python/pythonrun.c | 40 +++++++++++++++++++++++++++++++++------- 4 files changed, 41 insertions(+), 15 deletions(-) diff --git a/Include/pythonrun.h b/Include/pythonrun.h index 05287021aa7..bbf2903a397 100644 --- a/Include/pythonrun.h +++ b/Include/pythonrun.h @@ -29,9 +29,11 @@ DL_IMPORT(PyThreadState *) Py_NewInterpreter(void); DL_IMPORT(void) Py_EndInterpreter(PyThreadState *); DL_IMPORT(int) PyRun_AnyFile(FILE *, char *); +DL_IMPORT(int) PyRun_AnyFileEx(FILE *, char *, int); DL_IMPORT(int) PyRun_SimpleString(char *); DL_IMPORT(int) PyRun_SimpleFile(FILE *, char *); +DL_IMPORT(int) PyRun_SimpleFileEx(FILE *, char *, int); DL_IMPORT(int) PyRun_InteractiveOne(FILE *, char *); DL_IMPORT(int) PyRun_InteractiveLoop(FILE *, char *); @@ -40,6 +42,8 @@ DL_IMPORT(struct _node *) PyParser_SimpleParseFile(FILE *, char *, int); DL_IMPORT(PyObject *) PyRun_String(char *, int, PyObject *, PyObject *); DL_IMPORT(PyObject *) PyRun_File(FILE *, char *, int, PyObject *, PyObject *); +DL_IMPORT(PyObject *) PyRun_FileEx(FILE *, char *, int, + PyObject *, PyObject *, int); DL_IMPORT(PyObject *) Py_CompileString(char *, char *, int); diff --git a/Modules/main.c b/Modules/main.c index 34e99b9de6b..9afe80b3b56 100644 --- a/Modules/main.c +++ b/Modules/main.c @@ -268,11 +268,10 @@ Py_Main(int argc, char **argv) } } } - sts = PyRun_AnyFile( + sts = PyRun_AnyFileEx( fp, - filename == NULL ? "" : filename) != 0; - if (filename != NULL) - fclose(fp); + filename == NULL ? "" : filename, + filename != NULL) != 0; } if (inspect && stdin_is_interactive && diff --git a/Python/bltinmodule.c b/Python/bltinmodule.c index ea1269ed9ea..c2a74996446 100644 --- a/Python/bltinmodule.c +++ b/Python/bltinmodule.c @@ -815,10 +815,7 @@ builtin_execfile(PyObject *self, PyObject *args) PyErr_SetFromErrno(PyExc_IOError); return NULL; } - res = PyRun_File(fp, filename, Py_file_input, globals, locals); - Py_BEGIN_ALLOW_THREADS - fclose(fp); - Py_END_ALLOW_THREADS + res = PyRun_FileEx(fp, filename, Py_file_input, globals, locals, 1); return res; } diff --git a/Python/pythonrun.c b/Python/pythonrun.c index e951ccd1b7a..56b3ba82993 100644 --- a/Python/pythonrun.c +++ b/Python/pythonrun.c @@ -449,13 +449,23 @@ initsite(void) int PyRun_AnyFile(FILE *fp, char *filename) +{ + return PyRun_AnyFileEx(fp, filename, 0); +} + +int +PyRun_AnyFileEx(FILE *fp, char *filename, int closeit) { if (filename == NULL) filename = "???"; - if (Py_FdIsInteractive(fp, filename)) - return PyRun_InteractiveLoop(fp, filename); + if (Py_FdIsInteractive(fp, filename)) { + int err = PyRun_InteractiveLoop(fp, filename); + if (closeit) + fclose(fp); + return err; + } else - return PyRun_SimpleFile(fp, filename); + return PyRun_SimpleFileEx(fp, filename, closeit); } int @@ -541,6 +551,12 @@ PyRun_InteractiveOne(FILE *fp, char *filename) int PyRun_SimpleFile(FILE *fp, char *filename) +{ + return PyRun_SimpleFileEx(fp, filename, 0); +} + +int +PyRun_SimpleFileEx(FILE *fp, char *filename, int closeit) { PyObject *m, *d, *v; char *ext; @@ -558,7 +574,8 @@ PyRun_SimpleFile(FILE *fp, char *filename) #endif /* macintosh */ ) { /* Try to run a pyc file. First, re-open in binary */ - /* Don't close, done in main: fclose(fp); */ + if (closeit) + fclose(fp); if( (fp = fopen(filename, "rb")) == NULL ) { fprintf(stderr, "python: Can't reopen .pyc file\n"); return -1; @@ -568,7 +585,7 @@ PyRun_SimpleFile(FILE *fp, char *filename) Py_OptimizeFlag = 1; v = run_pyc_file(fp, filename, d, d); } else { - v = PyRun_File(fp, filename, Py_file_input, d, d); + v = PyRun_FileEx(fp, filename, Py_file_input, d, d, closeit); } if (v == NULL) { PyErr_Print(); @@ -845,8 +862,17 @@ PyObject * PyRun_File(FILE *fp, char *filename, int start, PyObject *globals, PyObject *locals) { - return run_err_node(PyParser_SimpleParseFile(fp, filename, start), - filename, globals, locals); + PyRun_FileEx(fp, filename, start, globals, locals, 0); +} + +PyObject * +PyRun_FileEx(FILE *fp, char *filename, int start, PyObject *globals, + PyObject *locals, int closeit) +{ + node *n = PyParser_SimpleParseFile(fp, filename, start); + if (closeit) + fclose(fp); + return run_err_node(n, filename, globals, locals); } static PyObject *