from django.conf import settings
from django.core.management.base import BaseCommand

from ...services import TweetDownloadService


class Command(BaseCommand):
    def add_arguments(self, parser):
        parser.add_argument(
            "--days_back",
            default=1,
            required=False,
            type=int,
            help="Stáří tweetů ve dnech",
        )

    def handle(self, *args, **options):
        tds = TweetDownloadService(
            bearer_token=settings.TWITTER_BEARER_TOKEN, days_back=options["days_back"]
        )
        tds.perform_update()

        self.stdout.write("\nUpdate of tweets finished!")