def geocode(request):
geocode = Geocode()
if request.GET.get('search'):
result = geocode.search(request.GET.get('search'))
return HttpResponse(content=json.dumps(result), status=200, content_type="application/json")
if request.GET.get('result'):
result = geocode.add_bbox(json.loads(request.GET.get('result')))
return HttpResponse(content=json.dumps(result), status=200, content_type="application/json")
else:
return HttpResponse(status=204, content_type="application/json")
def geocode(request):
geocode = Geocode()
if request.GET.get('search'):
result = geocode.search(request.GET.get('search'))
return HttpResponse(content=json.dumps(result), status=200, content_type="application/json")
if request.GET.get('result'):
result = geocode.add_bbox(json.loads(request.GET.get('result')))
return HttpResponse(content=json.dumps(result), status=200, content_type="application/json")
else:
return HttpResponse(status=204, content_type="application/json")
def geocode_test(self, api_response):
self.mock_requests.get(settings.GEOCODING_API_URL, text=json.dumps(api_response), status_code=200)
geocode = Geocode()
result = geocode.search("test")
self.assertIsNotNone(result.get("features"))
self.assertEqual(result.get("type"), "FeatureCollection")
self.assertIsInstance(result.get("bbox"), list)
for feature in result.get("features"):
self.assertIsInstance(feature.get("bbox"), list)
properties = feature.get("properties")
self.assertIsInstance(properties, dict)
self.assertIsNotNone(feature.get('geometry'))
for property in GeocodeAdapter._properties:
self.assertTrue(property in properties)
def geocode_test(self, api_response, mock_get_session):
mock_get_session.return_value = self.session
self.adapter.register_uri("GET", settings.GEOCODING_API_URL, text=json.dumps(api_response), status_code=200)
geocode = Geocode()
result = geocode.search("test")
self.assertIsNotNone(result.get("features"))
self.assertEqual(result.get("type"), "FeatureCollection")
self.assertIsInstance(result.get("bbox"), list)
for feature in result.get("features"):
self.assertIsInstance(feature.get("bbox"), list)
properties = feature.get("properties")
self.assertIsInstance(properties, dict)
self.assertIsNotNone(feature.get("geometry"))
self.assertIsNotNone(feature.get("properties", {}).get("context_name"))
for property in GeocodeAdapter._properties:
self.assertTrue(property in properties)
def geocode_test(self, api_response):
self.mock_requests.get(settings.GEOCODING_API_URL,
text=json.dumps(api_response),
status_code=200)
geocode = Geocode()
result = geocode.search("test")
self.assertIsNotNone(result.get("features"))
self.assertEqual(result.get("type"), "FeatureCollection")
self.assertIsInstance(result.get("bbox"), list)
for feature in result.get("features"):
self.assertIsInstance(feature.get("bbox"), list)
properties = feature.get("properties")
self.assertIsInstance(properties, dict)
self.assertIsNotNone(feature.get('geometry'))
for property in GeocodeAdapter._properties:
self.assertTrue(property in properties)
def test_pelias_polygon_geometry(self, mock_get_session):
mock_get_session.return_value = self.session
polygonCoordinates = [[[0, 1], [1, 0], [0, 3]]]
bbox = [-71.1912490997, 42.227911131, -70.9227798807, 42.3969775021]
api_response = {
"features": [
{
"type": "Feature",
"geometry": {"type": "Polygon", "coordinates": polygonCoordinates},
"properties": {},
"bbox": bbox,
}
]
}
self.adapter.register_uri("GET", settings.GEOCODING_API_URL, text=json.dumps(api_response), status_code=200)
geocode = Geocode()
result = geocode.search("test")
self.assertEqual(result.get("features")[0].get("geometry").get("coordinates"), polygonCoordinates)
def test_pelias_point_geometry(self):
bbox = [-71.1912490997, 42.227911131, -70.9227798807, 42.3969775021]
api_response = {
"features": [
{
"type": "Feature",
"geometry": {
"type":"Point",
"coordinates": []
},
"bbox": bbox
}
]
}
self.mock_requests.get(settings.GEOCODING_API_URL, text=json.dumps(api_response), status_code=200)
geocode = Geocode()
result = geocode.search("test")
self.assertEqual(result.get('features')[0].get('geometry').get('coordinates'), [[[-71.1912490997, 42.227911131], [-70.9227798807, 42.227911131], [-70.9227798807, 42.3969775021], [-71.1912490997, 42.3969775021], [-71.1912490997, 42.227911131]]])
def test_pelias_polygon_geometry(self):
polygonCoordinates = [[[0, 1], [1, 0], [0, 3]]]
bbox = [-71.1912490997, 42.227911131, -70.9227798807, 42.3969775021]
api_response = {
"features": [{
"type": "Feature",
"geometry": {
"type": "Polygon",
"coordinates": polygonCoordinates
},
"bbox": bbox
}]
}
self.mock_requests.get(settings.GEOCODING_API_URL,
text=json.dumps(api_response),
status_code=200)
geocode = Geocode()
result = geocode.search("test")
self.assertEqual(
result.get('features')[0].get('geometry').get('coordinates'),
polygonCoordinates)
def search(request):
"""
Detects the query type and calls the relevant geocoder to get results
:param request: User request which should include a query parameter
:return: A geojson with features matching the search query
"""
q = request.GET.get('query', None)
if not q:
return HttpResponse(status=204, content_type="application/json")
error_string = "An unknown error occurred while querying for results, please contact an administrator."
degree_range = 0.05
if is_mgrs(q):
# check for necessary settings
if getattr(settings, 'CONVERT_API_URL') is None:
return HttpResponse('No Convert API specified', status=501)
if getattr(settings, 'REVERSE_GEOCODING_API_URL') is None:
return HttpResponse('No Reverse Geocode API specified', status=501)
# make call to convert which should return a geojson feature of the MGRS location
convert = CoordinateConverter()
try:
mgrs_data = convert.get(q)
except Exception:
return HttpResponse(content=error_string, status=500)
# if no feature geom return nothing
if not mgrs_data or not mgrs_data.get('geometry'):
return HttpResponse(status=204, content_type="application/json")
features = []
# save the mgrs feature to return later
if not mgrs_data.get('properties'):
mgrs_data['properties'] = {}
mgrs_data['properties']['bbox'] = [
mgrs_data.get('geometry').get('coordinates')[0] - degree_range,
mgrs_data.get('geometry').get('coordinates')[1] - degree_range,
mgrs_data.get('geometry').get('coordinates')[0] + degree_range,
mgrs_data.get('geometry').get('coordinates')[1] + degree_range
]
mgrs_data['source'] = 'MGRS'
features.append(mgrs_data)
# call reverse to get a list of results near the mgrs feature
reverse = ReverseGeocode()
try:
result = reverse.search({
"point.lat":
mgrs_data.get('geometry').get('coordinates')[1],
"point.lon":
mgrs_data.get('geometry').get('coordinates')[0]
})
except Exception:
return HttpResponse(content=error_string, status=500)
if result.get('features'):
# add the mgrs feature with the search results and return together
result['features'] = features + result['features']
return HttpResponse(content=json.dumps(result),
status=200,
content_type="application/json")
# if no results just return the MGRS feature in the response
return HttpResponse(content=json.dumps({'features': features}),
status=200,
content_type="application/json")
elif is_lat_lon(q):
coords = is_lat_lon(q)
# if no reverse url return 501
if getattr(settings, 'REVERSE_GEOCODING_API_URL') is None:
return HttpResponse('No Reverse Geocode API specified', status=501)
# make call to reverse geocode
reverse = ReverseGeocode()
try:
result = reverse.search({
"point.lat": coords[0],
"point.lon": coords[1]
})
except Exception:
return HttpResponse(content=error_string, status=500)
# create a feature representing the exact lat/lon being searched
point_feature = {
"geometry": {
"type": "Point",
"coordinates": [coords[1], coords[0]]
},
"source": "Coordinate",
"type": "Feature",
"properties": {
"name":
"{0} {1}, {2} {3}".format(
coords[0] if coords[0] >= 0 else coords[0] * -1,
"N" if coords[0] >= 0 else "S",
coords[1] if coords[1] >= 0 else coords[1] * -1,
"E" if coords[1] >= 0 else "W"),
"bbox": [
coords[1] - degree_range, coords[0] - degree_range,
coords[1] + degree_range, coords[0] + degree_range
]
}
}
# if there are results add the point feature and return them together
if result.get('features'):
result.get('features').insert(0, point_feature)
return HttpResponse(content=json.dumps(result),
status=200,
content_type="application/json")
# if there are no results return only the point feature
features = {'features': [point_feature]}
return HttpResponse(content=json.dumps(features),
status=200,
content_type="application/json")
else:
# make call to geocode with search
geocode = Geocode()
try:
result = geocode.search(q)
except Exception as e:
logger.error(e)
return HttpResponse(content=error_string, status=500)
return HttpResponse(content=json.dumps(result),
status=200,
content_type="application/json")
def search(request):
"""
Detects the query type and calls the relevant geocoder to get results
:param request: User request which should include a query parameter
:return: A geojson with features matching the search query
"""
q = request.GET.get('query', None)
if not q:
return HttpResponse(status=204, content_type="application/json")
error_string = "An unknown error occurred while querying for results, please contact an administrator."
degree_range = 0.05
if is_mgrs(q):
# check for necessary settings
if getattr(settings, 'CONVERT_API_URL') is None:
return HttpResponse('No Convert API specified', status=501)
if getattr(settings, 'REVERSE_GEOCODING_API_URL') is None:
return HttpResponse('No Reverse Geocode API specified', status=501)
# make call to convert which should return a geojson feature of the MGRS location
convert = CoordinateConverter()
try:
mgrs_data = convert.get(q)
except Exception:
return HttpResponse(
content=error_string,
status=500
)
# if no feature geom return nothing
if not mgrs_data or not mgrs_data.get('geometry'):
return HttpResponse(status=204, content_type="application/json")
features = []
# save the mgrs feature to return later
if not mgrs_data.get('properties'):
mgrs_data['properties'] = {}
mgrs_data['properties']['bbox'] = [
mgrs_data.get('geometry').get('coordinates')[0] - degree_range,
mgrs_data.get('geometry').get('coordinates')[1] - degree_range,
mgrs_data.get('geometry').get('coordinates')[0] + degree_range,
mgrs_data.get('geometry').get('coordinates')[1] + degree_range
]
mgrs_data['source'] = 'MGRS'
features.append(mgrs_data)
# call reverse to get a list of results near the mgrs feature
reverse = ReverseGeocode()
try:
result = reverse.search({
"point.lat": mgrs_data.get('geometry').get('coordinates')[1],
"point.lon": mgrs_data.get('geometry').get('coordinates')[0]
})
except Exception:
return HttpResponse(
content=error_string,
status=500
)
if result.get('features'):
# add the mgrs feature with the search results and return together
result['features'] = features + result['features']
return HttpResponse(content=json.dumps(result), status=200, content_type="application/json")
# if no results just return the MGRS feature in the response
return HttpResponse(content=json.dumps({'features': features}), status=200, content_type="application/json")
elif is_lat_lon(q):
coords = is_lat_lon(q)
# if no reverse url return 501
if getattr(settings, 'REVERSE_GEOCODING_API_URL') is None:
return HttpResponse('No Reverse Geocode API specified', status=501)
# make call to reverse geocode
reverse = ReverseGeocode()
try:
result = reverse.search({
"point.lat": coords[0],
"point.lon": coords[1]
})
except Exception:
return HttpResponse(
content=error_string,
status=500
)
# create a feature representing the exact lat/lon being searched
point_feature = {
"geometry": {
"type": "Point",
"coordinates": [coords[1], coords[0]]
},
"source": "Coordinate",
"type": "Feature",
"properties": {
"name": "{0} {1}, {2} {3}".format(
coords[0] if coords[0] >= 0 else coords[0] * -1,
"N" if coords[0] >= 0 else "S",
coords[1] if coords[1] >= 0 else coords[1] * -1,
"E" if coords[1] >= 0 else "W"
),
"bbox": [
coords[1] - degree_range,
coords[0] - degree_range,
coords[1] + degree_range,
coords[0] + degree_range
]
}
}
# if there are results add the point feature and return them together
if result.get('features'):
result.get('features').insert(0, point_feature)
return HttpResponse(content=json.dumps(result), status=200, content_type="application/json")
# if there are no results return only the point feature
features = {'features': [point_feature]}
return HttpResponse(content=json.dumps(features), status=200, content_type="application/json")
else:
# make call to geocode with search
geocode = Geocode()
try:
result = geocode.search(q)
except Exception as e:
logger.error(e)
return HttpResponse(
content=error_string,
status=500
)
return HttpResponse(content=json.dumps(result), status=200, content_type="application/json")