summaryrefslogtreecommitdiff
path: root/chimere/management/commands/v3_to_v2_routes.py
blob: d373e257f7c0c419845c9555c527c91dde9c06c4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
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)