From 83cb3d192444172158b1a242a41b5edbebdf0dd9 Mon Sep 17 00:00:00 2001
From: Ned Batchelder <ned@nedbatchelder.com>
Date: Thu, 18 Jul 2013 09:56:07 -0400
Subject: [PATCH] Add batch and sleep arguments so we can control the speed
 from the command line.

---
 .../management/commands/clean_history.py      | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

diff --git a/lms/djangoapps/courseware/management/commands/clean_history.py b/lms/djangoapps/courseware/management/commands/clean_history.py
index 8b4f3d4d6d..30b0df8627 100644
--- a/lms/djangoapps/courseware/management/commands/clean_history.py
+++ b/lms/djangoapps/courseware/management/commands/clean_history.py
@@ -12,6 +12,7 @@ import datetime
 import json
 import logging
 import optparse
+import time
 import traceback
 
 from django.core.management.base import NoArgsCommand
@@ -24,12 +25,24 @@ class Command(NoArgsCommand):
     help = "Deletes unneeded rows from the StudentModuleHistory table."
 
     option_list = NoArgsCommand.option_list + (
+        optparse.make_option(
+            '--batch',
+            type='int',
+            default=100,
+            help="Batch size, number of module_ids to examine in a transaction.",
+        ),
         optparse.make_option(
             '--dry-run',
             action='store_true',
             default=False,
             help="Don't change the database, just show what would be done.",
         ),
+        optparse.make_option(
+            '--sleep',
+            type='float',
+            default=0,
+            help="Seconds to sleep between batches.",
+        ),
     )
 
     def handle_noargs(self, **options):
@@ -39,7 +52,7 @@ class Command(NoArgsCommand):
         smhc = StudentModuleHistoryCleaner(
             dry_run=options["dry_run"],
         )
-        smhc.main()
+        smhc.main(batch_size=options["batch"], sleep=options["sleep"])
 
 
 class StudentModuleHistoryCleaner(object):
@@ -54,7 +67,7 @@ class StudentModuleHistoryCleaner(object):
         self.next_student_module_id = 0
         self.last_student_module_id = 0
 
-    def main(self, batch_size=None):
+    def main(self, batch_size=None, sleep=0):
         """Invoked from the management command to do all the work."""
 
         batch_size = batch_size or self.BATCH_SIZE
@@ -74,6 +87,8 @@ class StudentModuleHistoryCleaner(object):
             if not self.dry_run:
                 self.commit()
             self.save_state()
+            if sleep:
+                time.sleep(sleep)
 
     def say(self, message):
         """