app.py

from flask import Flask, send_from_directory, request, jsonify
from Translator import TranslationWalk, InvalidTranslationWalk
from Context import Context, ContextFromJSON, ContextFromJSON5
from typing import Union
from simulator_errors import Errors

# import traceback
# import benedict
app = Flask(__name__)


@app.route('/basic')
def basic():
    return send_from_directory('gui/modules', 'basic.html')


@app.route('/')
@app.route('/advanced')
def advanced():
    return send_from_directory('gui/modules', 'advanced.html')


@app.route('/modules/<path:filename>')
def send_module(filename):
    return send_from_directory('gui/modules', filename)


def h2i(s: Union[str, None, int]) -> int:
    if not s:
        return None
    if type(s) == int:
        return s
    return int(s, 16)


def _clean(data: dict) -> dict:
    # d = benedict.benedict(data)
    d = data
    d['memory_size'] = h2i(d.get('memory_size'))
    d['lower_bound'] = h2i(d.get('lower_bound')) or 0
    d['satp.ppn'] = h2i(d.get('satp.ppn'))
    d['test_cases'] = [d]
    return d


@app.route('/api', methods=['POST'])
def simapi():
    data = request.get_json()
    # print('API called,', data)
    mgr = ContextFromJSON(_clean(data))
    d = mgr.jsonify_color()
    # print(d)
    return jsonify(d)


@app.route('/api/json5', methods=['POST'])
def json5api():
    try:
        data = request.get_json()
        data = data['code']
        mgr = ContextFromJSON5(data)
        d = mgr.jsonify_color()
        return jsonify(d)
    except Errors.InvalidConstraints:
        return jsonify({'error': "Couldn't satisfy the provided constraints"})
    except Errors.UnexpectedLeaf:
        return jsonify({'error': "Constraints caused an unexpected leaf"})
    except Errors.LeafMarkedAsPointer:
        return jsonify({'error': "Constraints given caused a leaf to be used as a pointer"})
    except ValueError:
        return jsonify({'error': 'JSON5 syntax error'})
    except:
        return jsonify({'error': 'Unknown runtime error'})


if __name__ == "__main__":
    app.run("0.0.0.0", port=8080, debug=True)