1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
""" File contains endpoints for rqlt2"""
import subprocess
import uuid
import os
from flask import current_app
from flask import jsonify
from flask import Blueprint
from flask import request
rqtl2 = Blueprint("rqtl2", __name__)
@rqtl2.route("/compute", methods=["GET"])
def compute():
"""Endpoint for computing QTL analysis using R/QTL2"""
wkdir = current_app.config.get("TMPDIR")
output_file = os.path.join(wkdir, "output.txt")
# this should be computed locally not via files
rscript_cmd = (
f"Rscript ./scripts/rqtl2_wrapper.R "
f"-i /home/kabui/r_playground/meta_grav.json "
f"-d /home/kabui/r_playground "
f"-o /home/kabui/r_playground/rqtl_output.json "
f"--nperm 100 --threshold 1 --cores 0"
)
process = subprocess.Popen(
rscript_cmd, shell=True,
stdout=subprocess.PIPE,
stderr=subprocess.STDOUT
)
# TODO rethink where we write this file
for line in iter(process.stdout.readline, b""):
# dont modify
with open(output_file, "a+") as file_handler:
file_handler.write(line.decode("utf-8"))
process.stdout.close()
process.wait()
if process.returncode == 0:
return jsonify({"msg": "success", "results": "file_here"})
else:
return jsonify({"msg": "fail", "error": "Process failed"})
@rqtl2.route("/stream/<indetifier>", methods=["GET"])
def stream(indetifier):
""" This endpoints streams stdout from a file expects
the indetifier to be the file """
# add seek position to this
output_file = os.path.join(current_app.config.get("TMPDIR"),
f"{indetifier}.txt")
seek_position = int(request.args.get("peak", 0))
with open(output_file) as file_handler:
# rethink how we do the read should this be stream / yield/peak ????
file_handler.seek(seek_position)
return jsonify({"data": file_handler.readlines(),
"pointer": file_handler.tell()})
|
