import datetime from django.conf import settings from django.core.management.base import BaseCommand from apps.rss_feeds.tasks import TaskBrokenFeeds, TaskFeeds class Command(BaseCommand): def add_arguments(self, parser): parser.add_argument("-f", "--feed", default=None) parser.add_argument("-a", "--all", default=False, action="store_true") parser.add_argument( "-b", "--broken", help="Task broken feeds that havent been fetched in a day.", default=False, action="store_true", ) parser.add_argument( "-V", "--verbose", action="store_true", dest="verbose", default=False, help="Verbose output." ) def handle(self, *args, **options): if options["broken"]: TaskBrokenFeeds.apply() else: TaskFeeds.apply()