new support for remove from collection

hivesolutions · Apr 1, 2016 · bcb8260 · bcb8260
1 parent b194c03
commit bcb8260
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/src/budy/models/product.py b/src/budy/models/product.py
@@ -301,17 +301,31 @@ def simple_csv_url(cls, absolute = False):
         )
 
     @appier.operation(
-        name = "Apply Collection",
+        name = "Add Collection",
         parameters = (
             ("Collection", "collection", str),
         )
     )
-    def apply_collection_s(self, collection):
+    def add_collection_s(self, collection):
         from . import collection as _collection
         collection = _collection.Collection.get(name = collection)
+        if collection in self.collections: return
         self.collections.append(collection)
         self.save()
 
+    @appier.operation(
+        name = "Remove Collection",
+        parameters = (
+            ("Collection", "collection", str),
+        )
+    )
+    def remove_collection_s(self, collection):
+        from . import collection as _collection
+        collection = _collection.Collection.get(name = collection)
+        if not collection in self.collections: return
+        self.collections.remove(collection)
+        self.save()
+
     def get_price(
         self,
         currency = None,