From baf3464bc94c12155e42c15381a943ecd8ef4c87 Mon Sep 17 00:00:00 2001
From: Kiran Jonnalagadda <kiran@hasgeek.com>
Date: Thu, 22 Apr 2021 17:20:51 +0530
Subject: [PATCH] Add SponsorMembership datasets (for #465)

---
 funnel/models/sponsor_membership.py | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/funnel/models/sponsor_membership.py b/funnel/models/sponsor_membership.py
index feaf3af1a..236e28afc 100644
--- a/funnel/models/sponsor_membership.py
+++ b/funnel/models/sponsor_membership.py
@@ -27,6 +27,35 @@ class SponsorMembership(ReorderMixin, ImmutableProfileMembershipMixin, db.Model)
     __roles__ = {
         'all': {'read': {'urls', 'profile', 'project', 'is_promoted', 'label', 'seq'}}
     }
+    __datasets__ = {
+        'primary': {
+            'urls',
+            'uuid_b58',
+            'offered_roles',
+            'is_promoted',
+            'label',
+            'seq',
+            'profile',
+            'project',
+        },
+        'without_parent': {
+            'urls',
+            'uuid_b58',
+            'offered_roles',
+            'is_promoted',
+            'label',
+            'seq',
+            'profile',
+        },
+        'related': {
+            'urls',
+            'uuid_b58',
+            'offered_roles',
+            'is_promoted',
+            'label',
+            'seq',
+        },
+    }
 
     project_id = immutable(
         db.Column(None, db.ForeignKey('project.id', ondelete='CASCADE'), nullable=False)