# HG changeset patch
# User Sverre Rabbelier <srabbelier@gmail.com>
# Date 1254692419 -7200
# Node ID a0dd39fa090b81edf8f21b393e2bfe3295e7ef48
# Parent  99f2d9b1ff7d528c49e035c3a136dca59c50cbd5
Add update for Documents

diff -r 99f2d9b1ff7d -r a0dd39fa090b app/soc/tasks/updates/module_conversion.py
--- a/app/soc/tasks/updates/module_conversion.py	Sun Oct 04 23:37:26 2009 +0200
+++ b/app/soc/tasks/updates/module_conversion.py	Sun Oct 04 23:40:19 2009 +0200
@@ -429,3 +429,34 @@
 
   # task completed, return
   return
+
+
+@decorators.iterative_task(document_logic)
+def runDocumentUpdate(request, entities, context, *args, **kwargs):
+  """AppEngine Task that updates Document entities.
+
+  Args:
+    request: Django Request object
+    entities: list of Document entities to update
+    context: the context of this task
+  """
+
+  from soc.modules.gsoc.logic.models.organization import logic as org_logic
+  from soc.modules.gsoc.logic.models.program import logic as program_logic
+
+  for entity in entities:
+    if entity.prefix == 'org':
+      org_entity = org_logic.getFromKeyName(entity.key().id_or_name())
+      entity.scope = org_entity
+      entity.home_for = org_entity if entity.home_for else None
+
+    if entity.prefix == 'program':
+      program_entity = program_logic.getFromKeyName(entity.key().id_or_name())
+      entity.scope = program_entity
+      entity.home_for = program_entity if entity.home_for else None
+
+  # store all Documents
+  db.put(entities)
+
+  # task completed, return
+  return