server.py

from executor import Executor

from enum import StrEnum
from flask import (
    Flask
    , request
    , make_response
    , jsonify
)

import logging

class Errors(StrEnum):
    NO_JSON_IN_BODY = "No JSON payload detected in request body"
    NO_SCRIPT_KEY_IN_JSON = "No `script` key found in payload"
    FAILED_EXECUTION = "Script failed or timed out"

logger = logging.getLogger('script-runner')

app = Flask("Script Runner")

def make_error_response(msg):
    return make_response((
        jsonify({
            'error': (
                msg.value
                if isinstance(msg, Errors)
                else str(msg)
            )
        })
        , 400
    ))

@app.route('/execute', methods=['POST'])
def execute_script():
    try:
        payload = request.get_json(force=True)
    except Exception as e:
        logger.warning(e)
        return make_error_response(Errors.NO_JSON_IN_BODY)

    script = payload.get('script')

    if script is None:
        return make_error_response(Errors.NO_SCRIPT_KEY_IN_JSON)

    try:
        result = Executor.run(script, nsjail=True)
        if error := result.get('error'):
            raise error

        return jsonify(
            result
        ), 200
    except Exception as e:
        logger.warning(e)
        return make_error_response(Errors.FAILED_EXECUTION)

if __name__ == '__main__':
    app.run(host="0.0.0.0", port=8080)