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'),

        ),

    ]

from jsonfield import JSONField

    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'))

                          )

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

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

        result = None

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

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

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

            lon, lat = pyproj.transform(wgs84, target_projection, 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)

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

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

                continue  # skip unknown

            result = {}

            point_lon = str(feature['geometry']['coordinates'][0])

            point_lat = str(feature['geometry']['coordinates'][1])

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

                point_lon, point_lat = pyproj.transform(target_projection, wgs84,

                                                        point_lon, point_lat)

            result['lon'] = point_lon

            result['lat'] = point_lat

            result['address'] = {'country': 'France'}

            for attribute, properties in self.attributes_mapping:

                for field in properties:

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

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

                        break

            # return first found point

            break

        return result

FAKE_GEOLOCATED_FEATURE = '''

{

    "crs": {

        "properties": {

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

        },

        "type": "name"

    },

    "features": [

        {

            "geometry": {

                "coordinates": [

                    1913311.15,

                    4224001.54

                ],

                "type": "Point"

            },

            "geometry_name": "the_geom",

            "id": "ref_ban_metro.fid--204aa923_161d7e3f06a_28bb",

            "properties": {

                "code_insee": 38185,

                "code_post": 38000,

                "libelle_acheminement": "GRENOBLE",

                "nom_afnor": "RUE DE NEW YORK",

                "nom_commune": "Grenoble",

                "nom_voie": "rue de new-york",

                "numero": 10,

                "secteur_voirie": "Nord-Ouest"

            },

            "type": "Feature"

        },

        {

            "geometry": {

                "coordinates": [

                    1913308.65,

                    4223996.14

                ],

                "type": "Point"

            },

            "geometry_name": "the_geom",

            "id": "ref_ban_metro.fid--204aa923_161d7e3f06a_28bc",

            "properties": {

                "code_insee": 38185,

                "code_post": 38000,

                "libelle_acheminement": "GRENOBLE",

                "nom_afnor": "RUE DE NEW YORK",

                "nom_commune": "Grenoble",

                "nom_voie": "rue de new-york",

                "numero": 23

            },

            "type": "Feature"

        }

    ],

    "totalFeatures": 2,

    "type": "FeatureCollection"

}

'''

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

def test_reverse_geocoding(mocked_get, app, connector):

    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.183784', 'lon': '5.714885'})

    assert mocked_get.call_args[1]['params']['CQL_FILTER'] == 'DWITHIN(the_geom,Point(636178.088142 5650501.0682),5,meters)'

    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)'

    assert resp.json['address']['country'] == 'France'

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

    assert resp.json['address']['road'] == 'rue de new-york'

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

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