107 lines
3.3 KiB
Python
107 lines
3.3 KiB
Python
from flask import Blueprint, Response, render_template, session, redirect
|
|
from os.path import join
|
|
|
|
from .. import db, config, scanner
|
|
|
|
blueprint = Blueprint('acquisition', __name__)
|
|
|
|
|
|
@blueprint.route('/scan/<id>')
|
|
def scan(id: int):
|
|
"""
|
|
Route to scan an object
|
|
"""
|
|
conn = db.get()
|
|
calibration_id = session.get('calibration_id', None)
|
|
object = db.Object.get_from_id(id, conn)
|
|
|
|
if calibration_id is None:
|
|
raise RuntimeError("Impossible de faire l'acquisition sans étalonnage")
|
|
|
|
return render_template('scan.html', object=object, calibrated=True)
|
|
|
|
|
|
@blueprint.route('/rescan/<id>')
|
|
def scan_existing(id: int):
|
|
conn = db.get()
|
|
calibrated = session.get('calibration_id', None) is not None
|
|
acquisition = db.Acquisition.get_from_id(id, conn)
|
|
object = acquisition.object(conn) if acquisition is not None else None
|
|
return render_template('scan.html', object=object, acquisition=acquisition, calibrated=calibrated)
|
|
|
|
|
|
@blueprint.route('/run/<object_id>')
|
|
def run(object_id: int):
|
|
conn = db.get()
|
|
calibration_id = session.get('calibration_id', None)
|
|
|
|
if calibration_id is None:
|
|
raise RuntimeError("Impossible de faire l'acquisition sans étalonnage")
|
|
|
|
object = db.Object.get_from_id(object_id, conn)
|
|
|
|
if object is None:
|
|
raise RuntimeError(f"Aucun objet d'id {object_id}")
|
|
|
|
with conn:
|
|
acquisition = object.add_acquisition(calibration_id, conn)
|
|
|
|
def generate():
|
|
yield str(acquisition.id)
|
|
length = len(config.LEDS_UUIDS) + 2 # with all_on and all_off
|
|
for index, led_uuid in enumerate(scanner.scan(join(config.OBJECT_DIR, str(object.id), str(acquisition.id)))):
|
|
yield f"{led_uuid},{(index+1)/length}\n"
|
|
|
|
return Response(generate(), mimetype='text/plain')
|
|
|
|
|
|
@blueprint.route('/rerun/<acquisition_id>')
|
|
def rescan(acquisition_id: int):
|
|
"""
|
|
Route to relaunch an acquisition
|
|
"""
|
|
conn = db.get()
|
|
calibration_id = session.get('calibration_id', None)
|
|
|
|
if calibration_id is None:
|
|
raise RuntimeError("Impossible de faire l'acquisition sans étalonnage")
|
|
|
|
acquisition = db.Acquisition.get_from_id(acquisition_id, conn)
|
|
|
|
if acquisition is None:
|
|
raise RuntimeError(f"Aucun acquisition d'id {acquisition_id}")
|
|
|
|
object = acquisition.object(conn)
|
|
|
|
def generate():
|
|
length = len(config.LEDS_UUIDS) + 2 # with all_on and all_off
|
|
for index, led_uuid in enumerate(scanner.scan(join(config.OBJECT_DIR, str(object.id), str(acquisition.id)))):
|
|
yield f"{led_uuid},{(index+1)/length}\n"
|
|
|
|
return Response(generate(), mimetype='text/plain')
|
|
|
|
|
|
@blueprint.route('/validate/<acquisition_id>')
|
|
def validate(acquisition_id: int):
|
|
conn = db.get()
|
|
acquisition = db.Acquisition.get_from_id(acquisition_id, conn)
|
|
|
|
if acquisition is None:
|
|
raise Exception(f"Aucune acquisition d'id {acquisition_id}")
|
|
|
|
object = acquisition.object(conn)
|
|
|
|
acquisition.validated = True
|
|
with conn:
|
|
acquisition.save(conn)
|
|
|
|
return redirect(f'/object/{object.id}')
|
|
|
|
|
|
@blueprint.route("/delete/<acquisition_id>")
|
|
def delete(acquisition_id: int):
|
|
conn = db.get()
|
|
with conn:
|
|
acqusition = db.Acquisition.delete_from_id(acquisition_id, conn)
|
|
return redirect('/object/' + str(acqusition.object_id))
|