Improve error message for ChartRequestSchema param show_individual_tr…

…aces_for.
FlexMeasures · Feb 26, 2021 · 204bf90 · 204bf90 · nhoening · Feb 27, 2021
1 parent 99beca7
commit 204bf90
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flexmeasures/ui/views/charts.py b/flexmeasures/ui/views/charts.py
@@ -1,7 +1,7 @@
 from flask_security import roles_accepted
 from flask_json import as_json
 from bokeh.embed import json_item
-from marshmallow import Schema, fields
+from marshmallow import Schema, fields, validate
 from webargs.flaskparser import use_args
 
 from flexmeasures.api.v2_0 import flexmeasures_api as flexmeasures_api_v2_0
@@ -43,7 +43,7 @@ class ChartRequestSchema(Schema):
     resolution = DurationField(required=True)
     show_consumption_as_positive = fields.Bool(missing=True)
     show_individual_traces_for = fields.Str(
-        missing="none", validate=lambda x: x in ("none", "schedules", "power")
+        missing="none", validate=validate.OneOf(["none", "schedules", "power"])
     )
     forecast_horizon = DurationField(missing="PT6H")