55 lines
3.4 KiB
PL/PgSQL
55 lines
3.4 KiB
PL/PgSQL
--DO NOT MODIFY THIS FILE, IT IS GENERATED AUTOMATICALLY FROM SOURCES
|
|
-- Complain if script is sourced in psql, rather than via CREATE EXTENSION
|
|
\echo Use "ALTER EXTENSION cdb_dataservices_server UPDATE TO '0.15.1'" to load this file. \quit
|
|
|
|
-- Here goes your code to upgrade/downgrade
|
|
|
|
DROP FUNCTION IF EXISTS cdb_dataservices_server.cdb_conf_getconf(text);
|
|
|
|
-- Geocodes a street address given a searchtext and a state and/or country
|
|
CREATE OR REPLACE FUNCTION cdb_dataservices_server.cdb_geocode_street_point(username TEXT, orgname TEXT, searchtext TEXT, city TEXT DEFAULT NULL, state_province TEXT DEFAULT NULL, country TEXT DEFAULT NULL)
|
|
RETURNS Geometry AS $$
|
|
import cartodb_services
|
|
cartodb_services.init(plpy, GD)
|
|
|
|
from cartodb_services.config.user import User
|
|
from cartodb_services.config.configs import ConfigsFactory
|
|
from cartodb_services.config.hires_geocoder_config import HiResGeocoderConfigFactory
|
|
from cartodb_services.request.request import RequestFactory
|
|
|
|
user = User(username, orgname)
|
|
configs = ConfigsFactory.get(user)
|
|
request = RequestFactory().create(user, configs, 'cdb_geocode_street_point')
|
|
|
|
# TODO change to hires_geocoder_config = HiResGeocoderConfigFactory.get(request)
|
|
hires_geocoder_config = HiResGeocoderConfigFactory(configs).get(user)
|
|
|
|
if hires_geocoder_config.provider == 'here':
|
|
here_plan = plpy.prepare("SELECT cdb_dataservices_server._cdb_here_geocode_street_point($1, $2, $3, $4, $5, $6) as point; ", ["text", "text", "text", "text", "text", "text"])
|
|
return plpy.execute(here_plan, [username, orgname, searchtext, city, state_province, country], 1)[0]['point']
|
|
elif hires_geocoder_config.provider == 'google':
|
|
google_plan = plpy.prepare("SELECT cdb_dataservices_server._cdb_google_geocode_street_point($1, $2, $3, $4, $5, $6) as point; ", ["text", "text", "text", "text", "text", "text"])
|
|
return plpy.execute(google_plan, [username, orgname, searchtext, city, state_province, country], 1)[0]['point']
|
|
elif hires_geocoder_config.provider == 'mapzen':
|
|
mapzen_plan = plpy.prepare("SELECT cdb_dataservices_server._cdb_mapzen_geocode_street_point($1, $2, $3, $4, $5, $6) as point; ", ["text", "text", "text", "text", "text", "text"])
|
|
return plpy.execute(mapzen_plan, [username, orgname, searchtext, city, state_province, country], 1)[0]['point']
|
|
else:
|
|
raise Exception('Requested geocoder is not available')
|
|
$$ LANGUAGE plpythonu;
|
|
|
|
|
|
CREATE OR REPLACE FUNCTION cdb_dataservices_server.cdb_here_geocode_street_point(username TEXT, orgname TEXT, searchtext TEXT, city TEXT DEFAULT NULL, state_province TEXT DEFAULT NULL, country TEXT DEFAULT NULL)
|
|
RETURNS Geometry AS $$
|
|
plpy.execute("SELECT cdb_dataservices_server._connect_to_redis('{0}')".format(username))
|
|
redis_conn = GD["redis_connection_{0}".format(username)]['redis_metrics_connection']
|
|
plpy.execute("SELECT cdb_dataservices_server._get_geocoder_config({0}, {1})".format(plpy.quote_nullable(username), plpy.quote_nullable(orgname)))
|
|
user_geocoder_config = GD["user_geocoder_config_{0}".format(username)]
|
|
|
|
if user_geocoder_config.heremaps_geocoder:
|
|
here_plan = plpy.prepare("SELECT cdb_dataservices_server._cdb_here_geocode_street_point($1, $2, $3, $4, $5, $6) as point; ", ["text", "text", "text", "text", "text", "text"])
|
|
return plpy.execute(here_plan, [username, orgname, searchtext, city, state_province, country], 1)[0]['point']
|
|
else:
|
|
raise Exception('Here geocoder is not available for your account.')
|
|
|
|
$$ LANGUAGE plpythonu;
|