from django.core.management.base import BaseCommand, CommandError

from games.utils import import_gamelist_file_to_db_for_system


class Command(BaseCommand):
    help = "Import all games found in a given gamelist XML file"

    def add_arguments(self, parser):
        parser.add_argument("system", type=str)
        parser.add_argument(
            "--file",
            action="store_true",
            help="Import from specific file",
        )

    def handle(self, *args, **options):
        games = import_gamelist_file_to_db_for_system(
            options["system"],
            options["file"],
        )
        self.stdout.write(self.style.SUCCESS(f"Successfully imported {len(games)}"))