import csv
import sys

from django.core.management.base import BaseCommand

from commcrawler.models import Crawl
from commcrawler.scrapy import launch_crawl


class Command(BaseCommand):
    help = 'Launch a crawl'

    def add_arguments(self, parser):
        parser.add_argument(
            '--quiet', dest='quiet', action='store_true',
            help='Quiet output')

    def handle(self, *args, **options):
        quiet = options['quiet']
        q = Crawl.objects.filter(status="C")
        if not q.count():
            sys.stdout.write('No crawl waiting. Exit.\n')
            return

        crawls = dict([(c.pk, c) for c in q.all()])
        available_ids = crawls.keys()
        c_id = None
        while c_id not in available_ids:
            sys.stdout.write('Which crawl to launch (type the number):\n')
            for crawl_id, crawl in crawls.items():
                sys.stdout.write('* {} - {}\n'.format(crawl_id, crawl))
                sys.stdout.flush()
            try:
                c_id = int(input("# "))
            except ValueError:
                c_id = None
        current_crawl = crawls[c_id]
        launch_crawl(current_crawl)