1# (c) 2018-2020 

2# MPIB <https://www.mpib-berlin.mpg.de/>, 

3# MPI-CBS <https://www.cbs.mpg.de/>, 

4# MPIP <http://www.psych.mpg.de/> 

5# 

6# This file is part of Castellum. 

7# 

8# Castellum is free software; you can redistribute it and/or modify it 

9# under the terms of the GNU Affero General Public License as published 

10# by the Free Software Foundation; either version 3 of the License, or 

11# (at your option) any later version. 

12# 

13# Castellum is distributed in the hope that it will be useful, but 

14# WITHOUT ANY WARRANTY; without even the implied warranty of 

15# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 

16# Affero General Public License for more details. 

17# 

18# You should have received a copy of the GNU Affero General Public 

19# License along with Castellum. If not, see 

20# <http://www.gnu.org/licenses/>. 

21 

22from django.contrib.gis.geos import Point 

23from django.core.management.base import BaseCommand 

24 

25from castellum.contacts.models import Address 

26from castellum.geofilters.geocode import geocode_address 

27from castellum.geofilters.models import Geolocation 

28 

29 

30def fetch_missing_geolocations(): 

31 geolocations = [] 

32 for address in Address.objects.filter(contact__geolocation=None): 

33 geocoded = geocode_address(address) 

34 if geocoded: 

35 geolocations.append(Geolocation( 

36 contact_id=address.contact_id, 

37 point=Point(geocoded.longitude, geocoded.latitude), 

38 )) 

39 Geolocation.objects.bulk_create(geolocations) 

40 

41 print('Created %i geolocations' % len(geolocations)) 

42 

43 

44class Command(BaseCommand): 

45 help = 'Fetch missing geolocations.' 

46 

47 def handle(self, *args, **options): 

48 fetch_missing_geolocations()