Added alias to Marshmallow pre_dump and pre_load

labthings · Jun 29, 2020 · c80e45f · c80e45f
1 parent 6b41db8
commit c80e45f
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/labthings/server/schema.py b/src/labthings/server/schema.py
@@ -1,14 +1,21 @@
 # -*- coding: utf-8 -*-
 from flask import url_for
 from werkzeug.routing import BuildError
-from marshmallow import Schema, pre_dump
+from marshmallow import Schema, pre_load, pre_dump
 from collections.abc import Mapping
 
 from .names import TASK_ENDPOINT, EXTENSION_LIST_ENDPOINT
 from .utilities import view_class_from_endpoint, description_from_view
 from . import fields
 
-__all__ = ["Schema", "FieldSchema", "TaskSchema", "ExtensionSchema"]
+__all__ = [
+    "Schema",
+    "FieldSchema",
+    "TaskSchema",
+    "ExtensionSchema",
+    "pre_load",
+    "pre_dump",
+]
 
 
 class FieldSchema(Schema):