rahix/openflexure-microscope-server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use @use_args

Browse source
This commit is contained in:
jtc42 2020-01-06 17:48:16 +00:00
parent 39cba66743
commit 8b472eab1b
1 changed files with 34 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

67
openflexure_microscope/api/default_extensions/scan.py
View file

@ -12,7 +12,11 @@ from openflexure_microscope.common.flask_labthings.find import (
) )
from openflexure_microscope.common.flask_labthings.extensions import BaseExtension from openflexure_microscope.common.flask_labthings.extensions import BaseExtension
from openflexure_microscope.common.flask_labthings.views.tasks import TaskSchema from openflexure_microscope.common.flask_labthings.views.tasks import TaskSchema
from openflexure_microscope.common.flask_labthings.decorators import marshal_with from openflexure_microscope.common.flask_labthings.decorators import (
marshal_with,
use_args,
)
from openflexure_microscope.common.flask_labthings import fields
from openflexure_microscope.devel import ( from openflexure_microscope.devel import (
JsonResponse, JsonResponse,
@ -346,30 +350,31 @@ def stack(
class TileScanAPI(Resource): class TileScanAPI(Resource):
@use_args(
{
"filename": fields.String(),
"temporary": fields.Boolean(missing=False),
"step_size": fields.List(fields.Integer, missing=[2000, 1500, 100]),
"grid": fields.List(fields.Integer, missing=[3, 3, 5]),
"style": fields.String(missing="raster"),
"autofocus_dz": fields.Integer(missing=50),
"fast_autofocus": fields.Boolean(missing=False),
"use_video_port": fields.Boolean(missing=False),
"bayer": fields.Boolean(missing=False),
"metadata": fields.Dict(missing={}),
"tags": fields.List(fields.String, missing=[]),
"resize": fields.Dict(missing=None), # TODO: Validate keys
}
)
@marshal_with(TaskSchema()) @marshal_with(TaskSchema())
def post(self): def post(self, args):
payload = JsonResponse(request) payload = JsonResponse(request)
microscope = find_device("openflexure_microscope") microscope = find_device("openflexure_microscope")
if not microscope: if not microscope:
abort(503, "No microscope connected. Unable to autofocus.") abort(503, "No microscope connected. Unable to autofocus.")
# Get params resize = args.get("resize", None)
filename = payload.param("filename")
temporary = payload.param("temporary", default=False, convert=bool)
step_size = payload.param("step_size", default=[2000, 1500, 100], convert=list)
step_size = [int(i) for i in step_size]
grid = payload.param("grid", default=[3, 3, 5], convert=list)
grid = [int(i) for i in grid]
style = payload.param("style", default="raster", convert=str)
autofocus_dz = payload.param("autofocus_dz", default=50, convert=int)
fast_autofocus = payload.param("fast_autofocus", default=False, convert=bool)
use_video_port = payload.param("use_video_port", default=True, convert=bool)
resize = payload.param("size", default=None)
if resize: if resize:
if ("width" in resize) and ("height" in resize): if ("width" in resize) and ("height" in resize):
resize = ( resize = (
@ -379,25 +384,21 @@ class TileScanAPI(Resource):
else: else:
abort(404) abort(404)
bayer = payload.param("bayer", default=False, convert=bool)
metadata = payload.param("metadata", default={}, convert=dict)
tags = payload.param("tags", default=[], convert=list)
logging.info("Running tile scan...") logging.info("Running tile scan...")
task = taskify(tile)( task = taskify(tile)(
microscope, microscope,
basename=filename, basename=args.get("filename"),
temporary=temporary, temporary=args.get("temporary"),
step_size=step_size, step_size=args.get("step_size"),
grid=grid, grid=args.get("grid"),
style=style, style=args.get("style"),
autofocus_dz=autofocus_dz, autofocus_dz=args.get("autofocus_dz"),
use_video_port=use_video_port, use_video_port=args.get("use_video_port"),
resize=resize, resize=resize,
bayer=bayer, bayer=args.get("bayer"),
fast_autofocus=fast_autofocus, fast_autofocus=args.get("fast_autofocus"),
metadata=metadata, metadata=args.get("metadata"),
tags=tags, tags=args.get("tags"),
) )
# return a handle on the scan task # return a handle on the scan task