Add _PyObject_SupportedAsScript().

Author: ericsnowcurrently

Include/internal/pycore_pythonrun.h

@@ -25,6 +25,7 @@ extern int _PyRun_InteractiveLoopObject(
     PyObject *filename,
     PyCompilerFlags *flags);
 
+extern int _PyObject_SupportedAsScript(PyObject *);
 extern const char* _Py_SourceAsString(
     PyObject *cmd,
     const char *funcname,

Python/crossinterp.c

@@ -851,9 +851,7 @@ get_script_xidata(PyThreadState *tstate, PyObject *obj, int pure,
         PyObject *ref = NULL;
         const char *script = _Py_SourceAsString(obj, "???", "???", &cf, &ref);
         if (script == NULL) {
-            if (!PyBytes_Check(obj) && !PyUnicode_Check(obj)
-                && !PyByteArray_Check(obj) && !PyObject_CheckBuffer(obj))
-            {
+            if (!_PyObject_SupportedAsScript(obj)) {
                 // We discard the raised exception.
                 _PyErr_Format(tstate, PyExc_TypeError,
                               "unsupported script %R", obj);

Python/pythonrun.c

@@ -1524,6 +1524,26 @@ Py_CompileStringExFlags(const char *str, const char *filename_str, int start,
     return co;
 }
 
+int
+_PyObject_SupportedAsScript(PyObject *cmd)
+{
+    if (PyUnicode_Check(cmd)) {
+        return 1;
+    }
+    else if (PyBytes_Check(cmd)) {
+        return 1;
+    }
+    else if (PyByteArray_Check(cmd)) {
+        return 1;
+    }
+    else if (PyObject_CheckBuffer(cmd)) {
+        return 1;
+    }
+    else {
+        return 0;
+    }
+}
+
 const char *
 _Py_SourceAsString(PyObject *cmd, const char *funcname, const char *what, PyCompilerFlags *cf, PyObject **cmd_copy)
 {