47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
import tempfile
|
|
|
|
from flask import jsonify
|
|
from flask import request
|
|
from flask import send_file
|
|
from flask.blueprints import Blueprint
|
|
|
|
from core import logger
|
|
from core.enumerations import EventTypes
|
|
|
|
coreemu = None
|
|
|
|
api = Blueprint("xml_api", __name__)
|
|
|
|
|
|
@api.route("/sessions/<int:session_id>/xml")
|
|
def save_xml(session_id):
|
|
session = coreemu.sessions.get(session_id)
|
|
if not session:
|
|
return jsonify(error="session does not exist"), 404
|
|
|
|
_, temp_path = tempfile.mkstemp()
|
|
session.save_xml(temp_path, "1.0")
|
|
return send_file(
|
|
temp_path,
|
|
as_attachment=True,
|
|
attachment_filename="session-%s.xml" % session_id
|
|
)
|
|
|
|
|
|
@api.route("/sessions/xml", methods=["POST"])
|
|
def open_xml():
|
|
session = coreemu.create_session()
|
|
session.set_state(EventTypes.CONFIGURATION_STATE)
|
|
|
|
logger.info("open xml: %s", request.files)
|
|
_, temp_path = tempfile.mkstemp()
|
|
session_file = request.files["file"]
|
|
session_file.save(temp_path)
|
|
|
|
try:
|
|
session.open_xml(temp_path, start=True)
|
|
return jsonify(id=session.session_id)
|
|
except:
|
|
logger.exception("error opening session file")
|
|
coreemu.delete_session(session.session_id)
|
|
return jsonify(error="error opening session file"), 404
|