Function executePythonFile

python/executor.py:909–963

(tree, runType="COMMAND_LINE", eventData=None)

def executePythonFile(tree, runType="COMMAND_LINE", eventData=None) -> Tuple[dict, dict]:
    global capture
    global response

    if tree["type"] == "PYTHON_FILE":
        statements = getStatementsFromBlock(tree["childSets"]["body"], True)

        if runType == "COMMAND_LINE" or runType == "SCHEDULE":
            pass
        elif runType == "HTTP_REQUEST":
            # we need to parse our http scenario event into serverless_wsgi so
            if not eventData:
                raise Exception("Need an event to run a HTTP request")

            extra = ast.parse(f"""
import serverless_wsgi

flask_app = None

try:
    flask_app = app
except NameError:
    print("Please call your Flask app 'app'")

if flask_app:
    {SPLOOT_SET_RESPONSE_FUNC}(serverless_wsgi.handle_request(app, {SPLOOT_HANDLER_ARG}, {{}}))
            """)

            statements.extend(extra.body)
        else:
            raise NotImplementedError("This run type is not implemented: " + runType)

        mods = ast.Module(body=statements, type_ignores=[])
        code = compile(ast.fix_missing_locations(mods), "main.py", mode="exec")
        # Uncomment to print generated Python code
        # print(ast.unparse(ast.fix_missing_locations(mods)))
        # print(ast.dump(mods))

        capture = SplootCapture()
        response = {}

        def set_response(r):
            global response
            response = r

        try:
            exec(code, {SPLOOT_KEY: capture, '__name__': '__main__', SPLOOT_HANDLER_ARG: eventData, SPLOOT_SET_RESPONSE_FUNC: set_response})
        except EOFError as e:
            # This is because we don't have inputs in a rerun.
            capture.logException(e)
        except BaseException as e:
            capture.logException(e)
            traceback.print_exc()

        return (capture.toDict(), response)


def wrapStdout(write):