106 lines
3.1 KiB
Python
106 lines
3.1 KiB
Python
|
from flask import jsonify
|
||
|
|
from flask import request
|
||
|
|
from flask.blueprints import Blueprint
|
||
|
|
|
||
|
|
import core_utils
|
||
|
|
|
||
|
|
coreemu = None
|
||
|
|
|
||
|
|
api = Blueprint("emane_api", __name__)
|
||
|
|
|
||
|
|
|
||
|
|
@api.route("/sessions/<int:session_id>/emane/config", methods=["PUT"])
|
||
|
|
def set_emane_config(session_id):
|
||
|
|
session = core_utils.get_session(coreemu, session_id)
|
||
|
|
data = request.get_json() or {}
|
||
|
|
values = data["values"]
|
||
|
|
config = {x["name"]: x["value"] for x in values}
|
||
|
|
session.emane.set_configs(config)
|
||
|
|
return jsonify()
|
||
|
|
|
||
|
|
|
||
|
|
@api.route("/sessions/<int:session_id>/emane/model/config", methods=["PUT"])
|
||
|
|
def set_emane_model_config(session_id):
|
||
|
|
session = core_utils.get_session(coreemu, session_id)
|
||
|
|
data = request.get_json() or {}
|
||
|
|
model_name = data["name"]
|
||
|
|
node_id = data.get("node")
|
||
|
|
values = data["values"]
|
||
|
|
config = {x["name"]: x["value"] for x in values}
|
||
|
|
session.emane.set_model_config(node_id, model_name, config)
|
||
|
|
return jsonify()
|
||
|
|
|
||
|
|
|
||
|
|
@api.route("/sessions/<int:session_id>/emane/config")
|
||
|
|
def get_emane_config(session_id):
|
||
|
|
session = core_utils.get_session(coreemu, session_id)
|
||
|
|
|
||
|
|
config = session.emane.get_configs()
|
||
|
|
|
||
|
|
config_options = []
|
||
|
|
for configuration in session.emane.emane_config.configurations():
|
||
|
|
value = config[configuration.id]
|
||
|
|
config_options.append({
|
||
|
|
"label": configuration.label,
|
||
|
|
"name": configuration.id,
|
||
|
|
"value": value,
|
||
|
|
"type": configuration.type.value,
|
||
|
|
"select": configuration.options
|
||
|
|
})
|
||
|
|
|
||
|
|
response = []
|
||
|
|
for config_group in session.emane.emane_config.config_groups():
|
||
|
|
start = config_group.start - 1
|
||
|
|
stop = config_group.stop
|
||
|
|
response.append({
|
||
|
|
"name": config_group.name,
|
||
|
|
"options": config_options[start: stop]
|
||
|
|
})
|
||
|
|
|
||
|
|
return jsonify(groups=response)
|
||
|
|
|
||
|
|
|
||
|
|
@api.route("/sessions/<int:session_id>/emane/model/config")
|
||
|
|
def get_emane_model_config(session_id):
|
||
|
|
session = core_utils.get_session(coreemu, session_id)
|
||
|
|
node_id = core_utils.get_node_id(request.args.get("node"))
|
||
|
|
|
||
|
|
model_name = request.args["name"]
|
||
|
|
model = session.emane.models[model_name]
|
||
|
|
config = session.emane.get_model_config(node_id, model_name)
|
||
|
|
|
||
|
|
config_options = []
|
||
|
|
for configuration in model.configurations():
|
||
|
|
value = config[configuration.id]
|
||
|
|
config_options.append({
|
||
|
|
"label": configuration.label,
|
||
|
|
"name": configuration.id,
|
||
|
|
"value": value,
|
||
|
|
"type": configuration.type.value,
|
||
|
|
"select": configuration.options
|
||
|
|
})
|
||
|
|
|
||
|
|
response = []
|
||
|
|
for config_group in model.config_groups():
|
||
|
|
start = config_group.start - 1
|
||
|
|
stop = config_group.stop
|
||
|
|
response.append({
|
||
|
|
"name": config_group.name,
|
||
|
|
"options": config_options[start: stop]
|
||
|
|
})
|
||
|
|
|
||
|
|
return jsonify(groups=response)
|
||
|
|
|
||
|
|
|
||
|
|
@api.route("/sessions/<int:session_id>/emane/models")
|
||
|
|
def get_emane_models(session_id):
|
||
|
|
session = core_utils.get_session(coreemu, session_id)
|
||
|
|
|
||
|
|
models = []
|
||
|
|
for model in session.emane.models.keys():
|
||
|
|
if len(model.split("_")) != 2:
|
||
|
|
continue
|
||
|
|
models.append(model)
|
||
|
|
|
||
|
|
return jsonify(models=models)
|