summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--fatcat/models.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/fatcat/models.py b/fatcat/models.py
index 24faee91..de913147 100644
--- a/fatcat/models.py
+++ b/fatcat/models.py
@@ -14,7 +14,7 @@ possible refactors:
import json
import hashlib
-from marshmallow import post_dump
+from marshmallow import post_dump, pre_load
from fatcat import db, ma
@@ -287,7 +287,7 @@ class ExtraJsonSchema(ma.ModelSchema):
raw = data.pop('json')
data.update(json.loads(raw))
- @post_dump(pass_many=False)
+ @pre_load(pass_many=False)
def flatten(self, data):
raw = json.dumps(data, indent=None)
for k in list(data.keys()):
@@ -342,6 +342,7 @@ class WorkEditSchema(ma.ModelSchema):
class Meta:
model = WorkEdit
+work_rev_schema = WorkRevSchema()
work_schema = WorkSchema()
work_edit_schema = WorkEditSchema()