# HG changeset patch
# User Sverre Rabbelier <srabbelier@gmail.com>
# Date 1240162947 0
# Node ID 46e490606f3d6bdba04d85b79605a90b8a7a9459
# Parent  96d8083cf974a9ce7b4c99858f795dc53f56c443
Added deleteEntities methods to stats.py

Patch by: Sverre Rabbelier

diff -r 96d8083cf974 -r 46e490606f3d scripts/stats.py
--- a/scripts/stats.py	Sun Apr 19 16:23:29 2009 +0000
+++ b/scripts/stats.py	Sun Apr 19 17:42:27 2009 +0000
@@ -272,6 +272,28 @@
   job_logic.updateOrCreateFromFields(job_fields)
 
 
+def deleteEntities(model, step_size=25):
+  """Deletes all entities of the specified type
+  """
+
+  print "Deleting..."
+  count = 0
+
+  while True:
+    entities = model.all().fetch(step_size)
+
+    if not entities:
+      break
+
+    for entity in entities:
+      entity.delete()
+
+    count += step_size
+
+    print "deleted %d entities" % count
+
+  print "Done"
+
 def loadPickle(name):
   """Loads a pickle.
   """
@@ -299,6 +321,9 @@
   from soc.models.student import Student
   from soc.models.mentor import Mentor
   from soc.models.org_admin import OrgAdmin
+  from soc.models.job import Job
+  from soc.models.student_proposal import StudentProposal
+  from soc.models.student_project import StudentProject
 
   def slotSaver(org, value):
     org.slots = value
@@ -313,6 +338,8 @@
       'orgStats': orgStats,
       'printPopularity': printPopularity,
       'saveValues': saveValues,
+      'getEntities': getEntities,
+      'deleteEntities': deleteEntities,
       'getOrgs': getEntities(Organization),
       'getUsers': getEntities(User),
       'getStudents': getEntities(Student),
@@ -323,10 +350,13 @@
       'convertProposals': convertProposals,
       'addFollower': addFollower,
       'Organization': Organization,
+      'Job': Job,
       'User': User,
       'Student': Student,
       'Mentor': Mentor,
       'OrgAdmin': OrgAdmin,
+      'StudentProject': StudentProject,
+      'StudentProposal': StudentProposal,
       'slotSaver': slotSaver,
       'popSaver': popSaver,
       'rawSaver': rawSaver,