3 files changed, 53 insertions, 0 deletions
diff --git a/Post/management/__init__.py b/Post/management/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/Post/management/__init__.py
diff --git a/Post/management/commands/__init__.py b/Post/management/commands/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/Post/management/commands/__init__.py
diff --git a/Post/management/commands/culldb.py b/Post/management/commands/culldb.py
new file mode 100644
index 0000000..698537c
--- /dev/null
+++ b/Post/management/commands/culldb.py
@@ -0,0 +1,53 @@
+# error-reporting-tool -  culldb
+#
+# Copyright (C) 2015 Intel Corporation
+#
+# Licensed under the MIT license, see COPYING.MIT for details
+
+# Create your views here.
+# vi: tabstop=8 expandtab shiftwidth=4 softtabstop=4
+
+from django.core.management.base import BaseCommand, CommandError
+from Post.models import Build, BuildFailure
+from optparse import make_option
+import time
+import sys
+
+class Command(BaseCommand):
+    help = 'Culls the database to size in rows'
+    args = '<size>'
+    option_list = BaseCommand.option_list + (
+        make_option('-i',
+                    '--info',
+                    dest='info',
+                    action="store_true",
+                    help='Show the current database size'),
+    )
+
+    def handle(self, *args, **options):
+        count = Build.objects.count()
+
+        if options['info']:
+            print "Current builds table size: %d" % Build.objects.count()
+            return
+
+        if len(args) > 0 and args[0]:
+            try:
+                new_size = int(args[0])
+            except ValueError:
+                print "Not a valid size"
+                return
+
+
+            num_to_delete = count - new_size
+            print "\nReducing the database size to %d which will DELETE %d rows" % (new_size, num_to_delete)
+            i = 1
+            while i != 0:
+                i = i-1
+                print 'Ctrl+c TO CANCEL. Executing in... %d \r' % i,
+                sys.stdout.flush()
+                time.sleep(1)
+
+            q = Build.objects.all()[:num_to_delete].values_list('pk',
+                                                                flat=True)
+            Build.objects.filter(pk__in=list(q)).delete()