Attach HyperlinkRelated to Marshmallow objects

h/t @singingwolfboy for catching this See https://github.com/marshmallow-code/flask-marshmallow/pull/39/files#r61445304
marshmallow-code · Apr 28, 2016 · edfd569 · edfd569
1 parent c36fdf9
commit edfd569
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/flask_marshmallow/__init__.py b/flask_marshmallow/__init__.py
@@ -109,6 +109,7 @@ def __init__(self, app=None):
         self.Schema = Schema
         if has_sqla:
             self.ModelSchema = sqla.ModelSchema
+            self.HyperlinkRelated = sqla.HyperlinkRelated
         _attach_fields(self)
         if app is not None:
             self.init_app(app)

diff --git a/tests/test_sqla.py b/tests/test_sqla.py
@@ -118,7 +118,7 @@ def test_hyperlink_related_field(self, extma, models, db, extapp):
         class BookSchema(extma.ModelSchema):
             class Meta:
                 model = models.Book
-            author = HyperlinkRelated('author')
+            author = extma.HyperlinkRelated('author')
 
         book_schema = BookSchema()