Produits Entr'ouvert » Passerelle

# -*- coding: utf-8 -*-

from __future__ import unicode_literals

from django.db import migrations, models

import jsonfield.fields

class Migration(migrations.Migration):

    dependencies = [

        ('opengis', '0004_auto_20180219_1613'),

    ]

    operations = [

        migrations.AddField(

            model_name='opengis',

            name='search_radius',

            field=models.IntegerField(default=5, verbose_name='Radius for point search'),

        ),

    ]

    search_radius = models.IntegerField(_('Radius for point search'), default=5)

    attributes_mapping = (('road', ('road', 'road_name', 'street', 'street_name', 'voie', 'nom_voie', 'rue')),

                          ('city', ('city', 'city_name', 'town', 'town_name', 'commune', 'nom_commune', 'ville', 'nom_ville')),

                          ('house_number', ('house_number', 'number', 'numero', 'numero_voie', 'numero_rue')),

                          ('postcode', ('postcode', 'postalCode', 'zipcode', 'codepostal', 'cp', 'code_postal', 'code_post')),

                          ('country', ('country', 'country_name', 'pays', 'nom_pays'))

                          )

    def convert_coordinates(self, lon, lat, reverse=False):

        lon, lat = float(lon), float(lat)

        if self.projection != 'EPSG:4326':

            wgs84 = pyproj.Proj(init='EPSG:4326')

            target_projection = pyproj.Proj(init=self.projection)

            if reverse:

                lon, lat = pyproj.transform(target_projection, wgs84, lon, lat)

            else:

                lon, lat = pyproj.transform(wgs84, target_projection, lon, lat)

        return lon, lat

    @endpoint(perm='can_access', description=_('Get feature info'))

    def reverse(self, request, lat, lon, **kwargs):

        result = None

        lon, lat = self.convert_coordinates(lon, lat)

        cql_filter = 'DWITHIN(the_geom,Point(%s %s),%s,meters)' % (lon, lat, self.search_radius)

        params = {

            'VERSION': self.get_wfs_service_version(),

            'SERVICE': 'WFS',

            'REQUEST': 'GetFeature',

            'TYPENAMES': self.query_layer,

            'OUTPUTFORMAT': 'json',

            'CQL_FILTER': cql_filter

        }

        response = self.requests.get(self.wfs_service_url, params=params)

        closest_feature = {}

        min_delta = None

        for feature in response.json().get('features'):

            if not feature['geometry']['type'] == 'Point':

                continue  # skip unknown

            lon_diff = abs(float(lon) - float(feature['geometry']['coordinates'][0]))

            lat_diff = abs(float(lat) - float(feature['geometry']['coordinates'][1]))

            delta = math.sqrt(lon_diff * lon_diff + lat_diff * lat_diff)

            if min_delta is None:

                min_delta = delta

            if delta <= min_delta:

                closest_feature = feature

        if closest_feature:

            result = {}

            point_lon = closest_feature['geometry']['coordinates'][0]

            point_lat = closest_feature['geometry']['coordinates'][1]

            point_lon, point_lat = self.convert_coordinates(point_lon, point_lat, reverse=True)

            result['lon'] = str(point_lon)

            result['lat'] = str(point_lat)

            result['address'] = {}

            for attribute, properties in self.attributes_mapping:

                for field in properties:

                    if closest_feature['properties'].get(field):

                        result['address'][attribute] = unicode(closest_feature['properties'][field])

                        break

        return result

FAKE_GEOLOCATED_FEATURE = '''{

    "crs": {

        "properties": {

            "name": "urn:ogc:def:crs:EPSG::3945"

        },

        "type": "name"

    },

    "features": [

        {

            "geometry": {

                "coordinates": [

                    1914059.51,

                    4224699.2

                ],

                "type": "Point"

            },

            "geometry_name": "the_geom",

            "properties": {

                "code_insee": 38185,

                "code_post": 38000,

                "nom_afnor": "BOULEVARD EDOUARD REY",

                "nom_commune": "Grenoble",

                "nom_voie": "boulevard \u00e9douard rey",

                "numero": 17

            },

            "type": "Feature"

        },

        {

            "geometry": {

                "coordinates": [

                    1914042.47,

                    4224665.2

                ],

                "type": "Point"

            },

            "geometry_name": "the_geom",

            "properties": {

                "code_insee": 38185,

                "code_post": 38000,

                "nom_commune": "Grenoble",

                "nom_voie": "place victor hugo",

                "numero": 2

            },

            "type": "Feature"

        },

        {

            "geometry": {

                "coordinates": [

                    1914035.7,

                    4224700.42

                ],

                "type": "Point"

            },

            "geometry_name": "the_geom",

            "properties": {

                "code_insee": 38185,

                "code_post": 38000,

                "nom_commune": "Grenoble",

                "nom_voie": "boulevard \u00e9douard rey",

                "numero": 28

            },

            "type": "Feature"

        },

        {

            "geometry": {

                "coordinates": [

                    1914018.64,

                    4224644.61

                ],

                "type": "Point"

            },

            "geometry_name": "the_geom",

            "properties": {

                "code_insee": 38185,

                "code_post": 38000,

                "nom_commune": "Grenoble",

                "nom_voie": "place victor hugo",

                "numero": 4

            },

            "type": "Feature"

        }

    ],

    "totalFeatures": 4,

    "type": "FeatureCollection"

}'''

@mock.patch('passerelle.utils.Request.get')

def test_reverse_geocoding(mocked_get, app, connector):

    connector.search_radius = 45

    connector.projection = 'EPSG:3945'

    connector.save()

    endpoint = utils.generic_endpoint_url('opengis', 'reverse', slug=connector.slug)

    assert endpoint == '/opengis/test/reverse'

    mocked_get.return_value = utils.FakedResponse(content=FAKE_GEOLOCATED_FEATURE, status_code=200)

    resp = app.get(endpoint, params={'lat':'45.1893469606986', 'lon': '5.72462060798'})

    assert mocked_get.call_args[1]['params']['CQL_FILTER'] == 'DWITHIN(the_geom,Point(1914061.48604 4224640.45779),45,meters)'

    assert resp.json['lon'] == '5.72407744145'

    assert resp.json['lat'] == '45.1893972656'

    assert resp.json['address']['house_number'] == '4'

    assert resp.json['address']['road'] == 'place victor hugo'

    assert resp.json['address']['postcode'] == '38000'

    assert resp.json['address']['city'] == 'Grenoble'

    connector.projection = 'EPSG:4326'

    connector.search_radius = 10

    connector.save()

    resp = app.get(endpoint, params={'lat':'45.183784', 'lon': '5.714885'})

    assert mocked_get.call_args[1]['params']['CQL_FILTER'] == 'DWITHIN(the_geom,Point(5.714885 45.183784),10,meters)'