import sys

from django.core.management.base import BaseCommand

from podcasts.models import PodcastFeed
from podcasts.views import _refresh_feeds


class Command(BaseCommand):
    help = 'Refresh podcast feeds (fetch new episodes from RSS)'

    def add_arguments(self, parser):
        parser.add_argument('--feed-id', type=int, help='Refresh only this feed ID')
        parser.add_argument('--limit', type=int, default=0, help='Max number of feeds to refresh')

    def handle(self, *args, **options):
        qs = PodcastFeed.objects.all().order_by('last_refreshed_at')

        if options['feed_id']:
            qs = qs.filter(pk=options['feed_id'])

        if options['limit']:
            qs = qs[: options['limit']]

        if not qs.exists():
            self.stdout.write('No feeds to refresh.')
            return

        # Parallel downloads, one per distinct feed URL (see _refresh_feeds).
        refreshed, failed, new = _refresh_feeds(qs)
        self.stdout.write(f'{refreshed} feeds refreshed, {failed} failed, +{new} episodes')