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: