import json

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

from finance.exports import ExportError, write_export


class Command(BaseCommand):
    help = "Create a versioned portable financial export."

    def add_arguments(self, parser):
        parser.add_argument("--output", dest="name")

    def handle(self, *args, **options):
        try:
            result = write_export(options["name"])
        except ExportError as exc:
            raise CommandError(exc.code) from None
        self.stdout.write(json.dumps({
            "name": result.name,
            "version": result.version,
            "rows": result.row_counts,
        }, separators=(",", ":")))