Added OneOf to stage type schema

This commit is contained in:
Joel Collins 2020-10-13 14:59:59 +01:00
parent ba9c77d5c3
commit 4521913f1b

View file

@ -7,14 +7,19 @@ import logging
class StageTypeProperty(PropertyView): class StageTypeProperty(PropertyView):
"""The type of the stage""" """The type of the stage"""
schema = fields.String(
missing=None,
example="SangaStage",
OneOf=["SangaStage", "SangaDeltaStage"],
description="The translation stage geometry",
)
def get(self): def get(self):
""" """
Get the stage geometry. Get the stage geometry.
""" """
microscope = find_component("org.openflexure.microscope") microscope = find_component("org.openflexure.microscope")
return microscope.configuration['stage']['type'] return microscope.configuration["stage"]["type"]
schema = fields.String(missing=None, example="SangaStage", description="The stage geometry [SangaStage, SangaDeltaStage]")
def put(self, stage_type): def put(self, stage_type):
""" """
@ -23,4 +28,3 @@ class StageTypeProperty(PropertyView):
microscope = find_component("org.openflexure.microscope") microscope = find_component("org.openflexure.microscope")
microscope.set_stage(stage_type=stage_type) microscope.set_stage(stage_type=stage_type)
return microscope.configuration["stage"]["type"] return microscope.configuration["stage"]["type"]