54 lines
2.0 KiB
Python
54 lines
2.0 KiB
Python
import logging
|
|
|
|
from django.core.management.base import BaseCommand
|
|
from geo.models import City, Region, Province, Country
|
|
import json
|
|
|
|
class Command(BaseCommand):
|
|
|
|
def handle(self, *args, **kwargs):
|
|
|
|
logging.info('Deleting all instances of Country, Region, Province, City')
|
|
City.objecs.all().delete()
|
|
Province.objecs.all().delete()
|
|
Region.objecs.all().delete()
|
|
Country.objecs.all().delete()
|
|
|
|
# create country for spain
|
|
country = Country.objects.create(name='España')
|
|
|
|
path = 'locations.json'
|
|
locations = json.loads(open(path).read())
|
|
|
|
region_counter = 0
|
|
# Regions loop
|
|
for location in locations:
|
|
if location['model'] == 'locations.region':
|
|
logging.info(f"Creating Region Object {location['fields']['name']}...")
|
|
name = location['fields']['name']
|
|
Region.objects.create(name=name, country=country, id=location['pk'])
|
|
region_counter += 1
|
|
|
|
province_counter = 0
|
|
# Provinces loop
|
|
for location in locations:
|
|
if location['model'] == 'locations.province':
|
|
logging.info(f"Creating Province Object {location['fields']['name']}...")
|
|
name = location['fields']['name']
|
|
Province.objects.create(name=name, region=Region.objects.get(id=location['fields']['region']), id=location['pk'])
|
|
province_counter += 1
|
|
|
|
city_counter = 0
|
|
# Cities loop
|
|
print('Creating cities...')
|
|
for location in locations:
|
|
if location['model'] == 'locations.city':
|
|
name = location['fields']['name']
|
|
City.objects.create(name=name, province=Province.objects.get(id=location['fields']['province']), id=location['pk'])
|
|
city_counter += 1
|
|
|
|
logging.info(f"Region instances created: {region_counter}")
|
|
logging.info(f"Province instances created: {province_counter}")
|
|
logging.info(f"City instances created: {city_counter}")
|
|
|