44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
from pyproj import Proj, transform, Geod
|
|
|
|
from qommon import misc
|
|
from qommon.storage import Contains
|
|
from wcs.wf.geolocate import GeolocateWorkflowStatusItem
|
|
|
|
geod = Geod(ellps='WGS84')
|
|
|
|
def get_coords(context):
|
|
coords = context.get('form_var_carte')
|
|
if coords:
|
|
lat, lon = coords.split(';')
|
|
return {'lat': float(lat), 'lon': float(lon)}
|
|
if 'form_var_numero' in context:
|
|
geolocate = GeolocateWorkflowStatusItem()
|
|
geolocate.method = 'address_string'
|
|
geolocate.address_string = '[form_var_numero] [form_var_voie], [form_var_commune], France'
|
|
return geolocate.geolocate_address_string(None)
|
|
return None
|
|
|
|
def get_close_demands(formdef, coords):
|
|
applied_filters = ['wf-%s' % x.id for x in formdef.workflow.get_not_endpoint_status()]
|
|
formdatas = formdef.data_class().select([Contains('status', applied_filters)])
|
|
counter = 0
|
|
for formdata in formdatas:
|
|
if not formdata.geolocations:
|
|
continue
|
|
formdata_coords = misc.normalize_geolocation(formdata.geolocations['base'])
|
|
|
|
distance = geod.inv(formdata_coords['lon'], formdata_coords['lat'],
|
|
coords['lon'], coords['lat'])[2]
|
|
formdata._distance = distance
|
|
formdata._coords = formdata_coords
|
|
if distance < 750:
|
|
counter += 1
|
|
formdata.counter = counter
|
|
yield formdata
|
|
|
|
if __name__ == '__builtin__':
|
|
coords = get_coords(vars())
|
|
result = ''
|
|
if 'form_objects' in vars():
|
|
result = get_close_demands(form_objects.formdef, coords)
|