Added re-indexing of data sets when they are un-shared.

refinery-platform · Jun 29, 2015 · d00bb44 · d00bb44
1 parent 4859a13
commit d00bb44
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/refinery/core/models.py b/refinery/core/models.py
@@ -497,6 +497,15 @@ def share(self, group, readonly=True):
         logger.info("Re-index / update data set: %s", self)
         DataSetIndex().update_object(self, using='core')
 
+    def unshare(self, group):
+        super(DataSet, self).unshare(group)
+        # This might be a hack but I couldn't find an easier solution to about
+        # the import loop. I found this solution here
+        # http://stackoverflow.com/a/7199514/981933
+        from core.search_indexes import DataSetIndex
+        logger.info("Re-index / update data set: %s", self)
+        DataSetIndex().update_object(self, using='core')
+
 
 class InvestigationLink(models.Model):
     data_set = models.ForeignKey(DataSet)