Changeset 12194 for main/waeup.ikoba/trunk/src/waeup/ikoba/customers/documents.py

Timestamp:

11 Dec 2014, 08:12:31 (10 years ago)

Author:

Henrik Bettermann

Message:

Fix handle_product_remove. The attribute is meanwhile called product_object.

Add handle_document_removed with test.

File:

• main/waeup.ikoba/trunk/src/waeup/ikoba/customers/documents.py (modified) (2 diffs)

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

@@ -25 +25 @@
 from zope.component.interfaces import IFactory
 from zope.interface import implementedBy
+from zope.event import notify
 
 from waeup.ikoba.image import IkobaImageFile
@@ -321 +322 @@
             file_obj.filename, file_obj.data)
 
+
+@grok.subscribe(ICustomerDocument, grok.IObjectRemovedEvent)
+def handle_document_removed(document, event):
+    """If a document is deleted, we make sure that also referrers to
+    customer contract objects are removed.
+    """
+    docid = document.document_id
+
+    # Find all customer contracts that refer to given document...
+    try:
+        contracts = document.customer['contracts'].values()
+    except AttributeError:
+        # customer not available. This might happen during tests.
+        return
+    for contract in contracts:
+        # Remove that referrer...
+        #contract.product = None
+        for key, value in contract.__dict__.items():
+            if key.endswith('_object') and \
+                getattr(value, 'document_id', None) == docid:
+                setattr(contract, key, None)
+                notify(grok.ObjectModifiedEvent(contract))
+                contract.customer.__parent__.logger.info(
+                    'ObjectRemovedEvent - %s - %s - removed: %s' % (
+                        contract.customer.customer_id,
+                        contract.contract_id,
+                        document.document_id))
+    return
+