Basket: manage item values for document generation

1 files changed, 7 insertions, 1 deletions

diff --git a/ishtar_common/models.py b/ishtar_common/models.py
index f78ecd511..dbbebae84 100644
--- a/ishtar_common/models.py
+++ b/ishtar_common/models.py
@@ -4601,7 +4601,7 @@ class IshtarUser(FullSearch):
 post_save.connect(cached_label_changed, sender=IshtarUser)
 
 
-class Basket(FullSearch, OwnPerms, TemplateItem):
+class Basket(FullSearch, OwnPerms, ValueGetter, TemplateItem):
     """
     Abstract class for a basket
     Subclass must be defined with an "items" ManyToManyField
@@ -4676,6 +4676,12 @@ class Basket(FullSearch, OwnPerms, TemplateItem):
     def get_write_query_owns(cls, ishtaruser):
         return Q(user=ishtaruser)
 
+    def get_values(self, prefix='', no_values=False):
+        values = super(Basket, self).get_values(prefix=prefix,
+                                                no_values=no_values)
+        values["items"] = [item.get_values() for item in self.items.all()]
+        return values
+
     def duplicate(self, label=None, ishtaruser=None):
         """
         Duplicate the basket. Items in basket are copied but not shared users