from io import StringIO
import json

from django.core.management import call_command
from django.core.management.base import BaseCommand


class Command(BaseCommand):
    help = "Route dump for v2"

    def handle(self, *args, **options):
        out = StringIO()
        call_command('dumpdata', 'chimere.Route', indent=4, stdout=out)
        vals = out.getvalue()
        routes = json.loads(vals)
        new_routes = []
        for route in routes:
            data = route.copy()
            data['fields'].pop('weight')
            data['fields'].pop('normalised_weight')
            data['fields'].pop('color')
            new_routes.append(data)
        res = json.dumps(new_routes)
        self.stdout.write(res)