Added cleanup_jobs management command for AC-323.

awx/main/management/commands/cleanup_jobs.py

@@ -0,0 +1,70 @@
+# Copyright (c) 2013 AnsibleWorks, Inc.
+# All Rights Reserved.
+
+# Python
+import datetime
+import logging
+from optparse import make_option
+
+# Django
+from django.core.management.base import NoArgsCommand, CommandError
+from django.db import transaction
+from django.contrib.auth.models import User
+from django.utils.dateparse import parse_datetime
+from django.utils.timezone import now, is_aware, make_aware
+
+# AWX
+from awx.main.models import Job
+
+class Command(NoArgsCommand):
+    '''
+    Management command to cleanup old jobs.
+    '''
+
+    help = 'Remove old jobs and events from the database.'
+
+    option_list = NoArgsCommand.option_list + (
+        make_option('--days', dest='days', type='int', default=90, metavar='N',
+                    help='Remove jobs executed more than N days ago'),
+        make_option('--dry-run', dest='dry_run', action='store_true',
+                    default=False, help='Dry run mode (show items that would '
+                    'be removed)'),
+    )
+
+    def cleanup_jobs(self):
+        #jobs_qs = Job.objects.exclude(status__in=('pending', 'running'))
+        #jobs_qs = jobs_qs.filter(created__lte=self.cutoff)
+        for job in Job.objects.all():
+            job_display = '"%s" (started %s, %d host summaries, %d events)' % \
+                (unicode(job), unicode(job.created),
+                job.job_host_summaries.count(), job.job_events.count())
+            if job.status in ('pending', 'running'):
+                action_text = 'would skip' if self.dry_run else 'skipping'
+                self.logger.debug('%s %s job %s', action_text, job.status, job_display)
+            elif job.created >= self.cutoff:
+                action_text = 'would skip' if self.dry_run else 'skipping'
+                self.logger.debug('%s %s', action_text, job_display)
+            else:
+                action_text = 'would delete' if self.dry_run else 'deleting'
+                self.logger.info('%s %s', action_text, job_display)
+                if not self.dry_run:
+                    job.delete()
+
+    def init_logging(self):
+        log_levels = dict(enumerate([logging.ERROR, logging.INFO,
+                                     logging.DEBUG, 0]))
+        self.logger = logging.getLogger('awx.main.commands.cleanup_jobs')
+        self.logger.setLevel(log_levels.get(self.verbosity, 0))
+        handler = logging.StreamHandler()
+        handler.setFormatter(logging.Formatter('%(message)s'))
+        self.logger.addHandler(handler)
+        self.logger.propagate = False
+
+    @transaction.commit_on_success
+    def handle_noargs(self, **options):
+        self.verbosity = int(options.get('verbosity', 1))
+        self.init_logging()
+        self.days = int(options.get('days', 90))
+        self.dry_run = bool(options.get('dry_run', False))
+        self.cutoff = now() - datetime.timedelta(days=self.days)
+        self.cleanup_jobs()