mirror of
https://github.com/ansible/awx.git
synced 2026-03-09 05:29:26 -02:30
62 lines
2.3 KiB
Python
62 lines
2.3 KiB
Python
# Copyright (c) 2014 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 ActivityStream
|
|
|
|
class Command(NoArgsCommand):
|
|
'''
|
|
Management command to purge old activity stream events.
|
|
'''
|
|
|
|
help = 'Remove old activity stream events from the database'
|
|
|
|
option_list = NoArgsCommand.option_list + (
|
|
make_option('--days', dest='days', type='int', default=90, metavar='N',
|
|
help='Remove activity stream events more than N days old'),
|
|
make_option('--dry-run', dest='dry_run', action='store_true',
|
|
default=False, help='Dry run mode (show items that would '
|
|
'be removed)'),)
|
|
|
|
def init_logging(self):
|
|
log_levels = dict(enumerate([logging.ERROR, logging.INFO,
|
|
logging.DEBUG, 0]))
|
|
self.logger = logging.getLogger('awx.main.commands.cleanup_activitystream')
|
|
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
|
|
|
|
def cleanup_activitystream(self):
|
|
for asobj in ActivityStream.objects.all():
|
|
asobj_disp = '"%s" id: %s' % (unicode(asobj), asobj.id)
|
|
if asobj.timestamp >= self.cutoff:
|
|
if self.dry_run:
|
|
self.logger.info("would skip %s" % asobj_disp)
|
|
else:
|
|
if self.dry_run:
|
|
self.logger.info("would delete %s" % asobj_disp)
|
|
if not self.dry_run:
|
|
asobj.delete()
|
|
|
|
def handle_noargs(self, **options):
|
|
self.verbosity = int(options.get('verbosity', 1))
|
|
self.init_logging()
|
|
self.days = int(options.get('days', 30))
|
|
self.cutoff = now() - datetime.timedelta(days=self.days)
|
|
self.dry_run = bool(options.get('dry_run', False))
|
|
self.cleanup_activitystream()
|