# HG changeset patch
# User Lennard de Rijk <ljvderijk@gmail.com>
# Date 1234696426 0
# Node ID c0ff6fc3192ea7500610d52c2a61d911cef218fc
# Parent  6655b1e89adbdbec6ee48c365ce04e87ce01b59b
Added clean_user_is_current to cleaning and fixed some indentation.

Patch by: Lennard de Rijk
Reviewed by: to-be-reviewed

diff -r 6655b1e89adb -r c0ff6fc3192e app/soc/logic/cleaning.py
--- a/app/soc/logic/cleaning.py	Sat Feb 14 23:56:49 2009 +0000
+++ b/app/soc/logic/cleaning.py	Sun Feb 15 11:13:46 2009 +0000
@@ -129,14 +129,33 @@
   @check_field_is_empty(field_name)
   def wrapped(self):
     link_id = clean_link_id(field_name)(self)
-  
+
     user_entity = user_logic.logic.getForFields({'link_id': link_id}, 
         unique=True)
-  
+
     if not user_entity:
       # user does not exist
       raise forms.ValidationError("This user does not exist.")
-  
+
+    return user_entity
+  return wrapped
+
+
+def clean_user_is_current(field_name):
+  """Check if the field_name value is a valid link_id and resembles the 
+     current user.
+  """
+
+  @check_field_is_empty(field_name)
+  def wrapped(self):
+    link_id = clean_link_id(field_name)(self)
+
+    user_entity = user_logic.logic.getForCurrentAccount()
+
+    if not user_entity or user_entity.link_id != link_id:
+      # this user is not the current user
+      raise forms.ValidationError("This user is not you.")
+
     return user_entity
   return wrapped
 
@@ -149,14 +168,14 @@
   @check_field_is_empty(field_name)
   def wrapped(self):
     link_id = clean_link_id(field_name)(self)
-  
+
     user_entity = user_logic.logic.getForFields({'link_id': link_id}, 
         unique=True)
-  
+
     if user_entity:
       # user exists already
       raise forms.ValidationError("There is already a user with this link id.")
-  
+
     return link_id
   return wrapped
 
@@ -168,16 +187,16 @@
 
   @check_field_is_empty(field_name)
   def wrapped(self):
-    
+
     clean_user_field = clean_existing_user(field_name)
     user_entity = clean_user_field(self)
-    
+
     current_user_entity = user_logic.logic.getForCurrentAccount()
-    
+
     if user_entity.key() == current_user_entity.key():
       # users are equal
       raise forms.ValidationError("You cannot enter yourself here.")
-    
+
     return user_entity
   return wrapped