kit-eco.social/consumocuidado-server
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

improvements to loadgisdata

Browse Source
This commit is contained in:
Sam
2021-02-11 12:51:59 +00:00
parent 2de6a66af1
commit 20a4040aa7
3 changed files with 81 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files

141
core/management/commands/loadgisdata.py
View File

@@ -17,9 +17,12 @@ class Command(BaseCommand):
help = 'Load GIS data into database'
def handle(self, *args, **kwargs):
print(self.help)
logging.info('Deleting all instances of Country, Region, Province, City')
City.objects.all().delete()
Province.objects.all().delete()
Region.objects.all().delete()
Country.objects.all().delete()
print('Deleted all instances of Country, Region, Province, City')
ds = DataSource('datasets/gadm36_ESP.gpkg')
@@ -31,86 +34,94 @@ class Command(BaseCommand):
# country instances
logging.info("loading country instances")
print('Creating Country instances')
country_counter = 0
for feature in ds[0]:
geom = GEOSGeometry(str(feature['geometry']))
if feature['geometry']['type'] == "MultiPolygon":
poly_list = []
for poly in geom:
poly_list.append(poly)
print(poly_list)
else:
poly_list = geom
geom_geos = MultiPolygon(poly_list)
name = feature['properties']['NAME_1']
if name == 'España':
SPAIN = Country.objects.create(name=name,poly=geom_geos)
Country.objects.create(name=name,poly=geom_geos)
country_counter += 1
logging.info(f"Country instance created for: {name}")
try:
# calculate geometry data
geom = GEOSGeometry(str(feature.geom))
polygon_list = []
for polygon in geom:
polygon_list.append(polygon)
geom_geos = MultiPolygon(polygon_list)
# create instance
name = feature.get('NAME_0')
if name == 'Spain':
SPAIN = Country.objects.create(name='España',geo=geom_geos)
Country.objects.create(name=name,geo=geom_geos)
country_counter += 1
logging.info(f"Country instance created for: {name}")
except Exception as e:
logging.error(f"[{type(e)}] {str(e)}")
import ipdb; ipdb.set_trace()
# region instances
logging.info("loading region instances")
print('Creating Region instances')
region_counter = 0
for feature in ds[1]:
geom = GEOSGeometry(str(feature['geometry']))
if feature['geometry']['type'] == "MultiPolygon":
poly_list = []
for poly in geom:
poly_list.append(poly)
print(poly_list)
else:
poly_list = geom
geom_geos = MultiPolygon(poly_list)
name = feature['properties']['NAME_1']
Region.objects.create(name=name,poly=geom_geos, country=SPAIN)
region_counter += 1
logging.info(f"Region instance created for: {name}")
try:
# calculate geometry data
geom = GEOSGeometry(str(feature.geom))
polygon_list = []
for polygon in geom:
polygon_list.append(polygon)
geom_geos = MultiPolygon(polygon_list)
# get region name
name = feature.get('NAME_1')
# create instance
Region.objects.create(name=name, geo=geom_geos, country=SPAIN)
region_counter += 1
logging.info(f"Region instance created for: {name}")
except Exception as e:
logging.error(f"[{type(e)}] {str(e)}")
import ipdb; ipdb.set_trace()
# province instances
logging.info("loading province instances")
print('Creating Province instances')
province_counter = 0
for feature in ds[2]:
geom = GEOSGeometry(str(feature['geometry']))
if feature['geometry']['type'] == "MultiPolygon":
poly_list = []
for poly in geom:
poly_list.append(poly)
print(poly_list)
else:
poly_list = geom
geom_geos = MultiPolygon(poly_list)
name = feature['properties']['NAME_1']
parent_region = Region.objects.get(id=location['fields']['region'])
Province.objects.create(name=name,poly=geom_geos, country=SPAIN)
province_counter += 1
logging.info(f"Province instance created for: {name}")
try:
# calculate geometry data
geom = GEOSGeometry(str(feature.geom))
polygon_list = []
for polygon in geom:
polygon_list.append(polygon)
geom_geos = MultiPolygon(polygon_list)
# get name and parent
name = feature['properties']['NAME_2']
parent_region = Region.objects.get(name=feature.get('NAME_1'))
# create instance
Province.objects.create(name=name, geo=geom_geos, region=parent_region)
province_counter += 1
logging.info(f"Province instance created for: {name}")
except Exception as e:
logging.error(f"[{type(e)}] {str(e)}")
import ipdb; ipdb.set_trace()
# city instances
logging.info("loading city instances")
print('Creating City instances')
city_counter = 0
for feature in ds[4]:
geom = GEOSGeometry(str(feature['geometry']))
if feature['geometry']['type'] == "MultiPolygon":
poly_list = []
for poly in geom:
poly_list.append(poly)
print(poly_list)
else:
poly_list = geom
try:
# calculate geometry data
geom = GEOSGeometry(str(feature.geom))
polygon_list = []
for polygon in geom:
polygon_list.append(polygon)
geom_geos = MultiPolygon(polygon_list)
geom_geos = MultiPolygon(poly_list)
name = feature['properties']['NAME_1']
City.objects.create(name=name,poly=geom_geos, country=SPAIN)
city_counter += 1
logging.debug(f"City instance created for: {name}")
# get name and parent
name = feature['properties']['NAME_3']
parent_province = Region.objects.get(name=feature.get('NAME_2'))
City.objects.create(name=name, geo=geom_geos, province=parent_province)
city_counter += 1
logging.debug(f"City instance created for: {name}")
except Exception as e:
logging.error(f"[{type(e)}] {str(e)}")
import ipdb; ipdb.set_trace()
logging.info(f"Country instances created: {country_counter}")
logging.info(f"Region instances created: {region_counter}")