# This is the Flask app.py file that runs a web server and returns "Hello World"
from flask import Flask, request, render_template 
import subprocess
app = Flask(__name__)

doc_root = '/root/origin-33/flask/scripts'

@app.route('/lab/bm',methods = ['GET'])
def bm():
    return render_template('bm.html')

@app.route('/lab/ivm',methods = ['GET'])
def ivm():
    return render_template('ivm.html')
    
@app.route('/lab/bm/run',methods = ['GET'])
def bm_run():
    property = request.args.get('property')
    property = str(property)
    print(property)
    script = f'{doc_root}/mychecker.sh'
    args = ['-c', f'{doc_root}/bm.conf', property]
    output = subprocess.check_output ([script] + args, shell=False)
    output = output.decode('utf-8')
    print (output)
    output = output.splitlines()
    return output

@app.route('/lab/ivm/generate',methods = ['GET'])
def ivm_generate_image():
    property = request.args.get('property')
    property = str(property)
    print(property)
    script = f"{doc_root}/generate.sh"
    args = [property]
    output = subprocess.check_output ([script] + args, shell=False)
    output = output.decode('utf-8')
    print (output)
    output = output.splitlines()
    return output

@app.route('/lab/ivm/replace_image',methods = ['GET'])
def ivm_replace_image():
    property = request.args.get('property')
    property = str(property)
    print(property)
    script = f"{doc_root}/replaceImage.sh"
    args = [property]
    output = subprocess.check_output ([script] + args, shell=False)
    output = output.decode('utf-8')
    print (output)
    output = output.splitlines()
    return output

@app.route('/lab/ivm/replace_video',methods = ['GET'])
def ivm_replace_video():
    property = request.args.get('property')
    property = str(property)
    print(property)
    script = f"{doc_root}/replaceVideo.sh"
    args = [property]
    output = subprocess.check_output ([script] + args, shell=False)
    output = output.decode('utf-8')
    print (output)
    output = output.splitlines()
    return output

if __name__ == '__main__':
    app.run()