add fields from schema.only to schema.dump_fields

Author: pysalt

flask_rest_jsonapi/schema.py

@@ -1,4 +1,5 @@
 """Helpers to deal with marshmallow schemas"""
+from collections import OrderedDict
 
 from marshmallow import class_registry
 from marshmallow.base import SchemaABC
@@ -58,6 +59,8 @@ def compute_schema(schema_cls, default_kwargs, qs, include):
         if schema.only is not None and 'id' not in schema.only:
             schema.only += ('id',)
 
+        schema.dump_fields = OrderedDict(**{name: val for name, val in schema.fields.items() if name in schema.only})
+
     # manage compound documents
     if include:
         for include_path in include: