arvinbehbahani562@gmail.com
41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
from django.core.management.base import BaseCommand #, CommandError
|
|
import json
|
|
from pathlib import Path
|
|
from ...models import Province, City
|
|
|
|
class Command(BaseCommand):
|
|
help = "push all provinces and cities to database"
|
|
|
|
def handle(self, *args, **options):
|
|
json_file_path = Path("/app/addresses/provinces_cities.json")
|
|
|
|
with open(json_file_path, "r", encoding="utf-8") as json_file:
|
|
all_data = json.load(json_file)
|
|
|
|
print("injecting all provinces to the database...")
|
|
|
|
for province_data in all_data:
|
|
province = Province.objects.get_or_create(
|
|
province_id=int(province_data.get("provinceId")),
|
|
province_name=str(province_data.get("provinceName")),
|
|
)
|
|
|
|
print("task complete :)")
|
|
|
|
print("injecting all the cities to the database ")
|
|
for city_data in all_data:
|
|
city = City.objects.get_or_create(
|
|
city_local_id=str(city_data.get("cityId")),
|
|
city_name=str(city_data.get("cityName")),
|
|
province_id=int(city_data.get("provinceId"))
|
|
)
|
|
|
|
print("task complete :)")
|
|
|
|
self.stdout.write(
|
|
self.style.SUCCESS(r"all the data successfully injected in the database :>")
|
|
)
|
|
|
|
|
|
|
|
|