Changeset 12100 for main/waeup.ikoba

Timestamp:

1 Dec 2014, 05:51:58 (10 years ago)

Author:

Henrik Bettermann

Message:

We don't need a contains method since there are usually only a few products in the products container. Thus we can use the the BasicContextualSourceFactory? instead of the SmartBasicContextualSourceFactory?.

Add source factory for customer documents.

File:

• main/waeup.ikoba/trunk/src/waeup/ikoba/customers/vocabularies.py (modified) (3 diffs)

main/waeup.ikoba/trunk/src/waeup/ikoba/customers/vocabularies.py

@@ -26 +26 @@
 from zc.sourcefactory.basic import BasicSourceFactory
 from zc.sourcefactory.contextual import BasicContextualSourceFactory
-from waeup.ikoba.interfaces import SimpleIkobaVocabulary
+from waeup.ikoba.interfaces import SimpleIkobaVocabulary, SUBMITTED, VERIFIED
 from waeup.ikoba.interfaces import MessageFactory as _
 from waeup.ikoba.utils.helpers import get_sorted_preferred
 from waeup.ikoba.utils.countries import COUNTRIES
-from waeup.ikoba.sourcefactory import SmartBasicContextualSourceFactory
-
 
 #: a tuple of tuples (<COUNTRY-NAME>, <ISO-CODE>) with Nigeria first.
@@ -120 +118 @@
 
 
-class ConCatProductSource(SmartBasicContextualSourceFactory):
-    """An contract category product delivers all products
+class ConCatProductSource(BasicContextualSourceFactory):
+    """A contract category product source delivers all products
     which belong to a certain contract_category.
     """
-    def contains(self, context, value):
-        concat = getattr(context, 'contract_category', None)
-        if value.contract_category == concat:
-            return True
-        return False
 
     def getValues(self, context):
@@ -144 +137 @@
     def getTitle(self, context, value):
         return "%s - %s" % (value.product_id, value.title)
+
+
+class CustomerDocumentSource(BasicContextualSourceFactory):
+    """A customer document source delivers all submitted and verfified documents
+    of the context customer.
+    """
+
+    def getValues(self, context):
+        documents = context.customer['documents'].values()
+        resultlist = [
+            value for value in documents if value.state in (SUBMITTED, VERIFIED)]
+        return resultlist
+
+    def getToken(self, context, value):
+        return value.document_id
+
+    def getTitle(self, context, value):
+        return "%s - %s" % (value.document_id, value.title)