if "DJANGO_SETTINGS_MODULE" not in os.environ:
# Set up django
print("Setting up Django")
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "project_conf.settings")
import django
django.setup()
from django.db.utils import IntegrityError
from main.models import State, District, SubDistrict
fail_message = "bulk_create on {} failed. Falling back to individual saves."
pass_message = "bulk_create on {} succeeded."
if args.states is not None:
state_dicts = get_state_dicts(args.states)
states = [State(**d) for d in state_dicts]
try:
State.objects.bulk_create(states)
print(pass_message.format('State'))
except IntegrityError:
print(fail_message.format('State'))
for state in states:
state.save()
if args.districts is not None:
district_dicts = get_district_dicts(args.districts)
districts = [District(**d) for d in district_dicts]
try:
District.objects.bulk_create(districts)
print(pass_message.format('District'))
except IntegrityError:
