# HG changeset patch
# User Sverre Rabbelier <srabbelier@gmail.com>
# Date 1236378703 0
# Node ID 699f65fbf08ad41a255c197c446f406ddd879d49
# Parent  a8da5a62023677d3a79a12043dd27c4da22d3837
Add a document ref cleaner

Patch by: Sverre Rabbelier

diff -r a8da5a620236 -r 699f65fbf08a app/soc/logic/cleaning.py
--- a/app/soc/logic/cleaning.py	Fri Mar 06 22:31:19 2009 +0000
+++ b/app/soc/logic/cleaning.py	Fri Mar 06 22:31:43 2009 +0000
@@ -34,6 +34,7 @@
 
 from soc.logic import rights as rights_logic
 from soc.logic import validate
+from soc.logic.models import document as document_logic
 from soc.logic.models.site import logic as site_logic
 from soc.logic.models.user import logic as user_logic
 from soc.models import document as document_model
@@ -48,6 +49,10 @@
 DEF_ORGANZIATION_NOT_ACTIVE_MSG = ugettext(
     "This organization is not active or doesn't exist.")
 
+DEF_NO_SUCH_DOCUMENT_MSG = ugettext(
+    "There is no such document with that link_id under this entity.")
+
+
 def check_field_is_empty(field_name):
   """Returns decorator that bypasses cleaning for empty fields.
   """
@@ -328,6 +333,43 @@
   return wrapped
 
 
+def clean_refs(params, fields):
+  """Cleans all references to make sure they are valid.
+  """
+
+  logic = params['logic']
+
+  def wrapped(self):
+    """Decorator wrapper method.
+    """
+
+    scope_path = logic.getKeyNameFromFields(self.cleaned_data)
+
+    key_fields = {
+        'scope_path': scope_path,
+        'prefix': params['document_prefix'],
+        }
+
+    for field in fields:
+      link_id = self.cleaned_data.get(field)
+
+      if not link_id:
+        continue
+
+      key_fields['link_id'] = link_id
+      ref = document_logic.logic.getFromKeyFields(key_fields)
+
+      if not ref:
+        self._errors[field] = ErrorList([DEF_NO_SUCH_DOCUMENT_MSG])
+        del self.cleaned_data[field]
+      else:
+        self.cleaned_data['resolved_%s' % field] = ref
+
+    return self.cleaned_data
+
+  return wrapped
+
+
 def validate_user_edit(link_id_field, account_field):
   """Clean method for cleaning user edit form.