cartodb4/dataservices-api
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add the hires_geocoder to cdb_monthly_quota #302

Browse Source
This commit is contained in:
Rafa de la Torre
2016-11-10 17:00:25 +01:00
parent 7a56dc698f
commit 7a32ace1ac
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
server/extension/sql/200_quotas.sql
+9 -2
View File
@@ -3,15 +3,22 @@ CREATE OR REPLACE FUNCTION cdb_dataservices_server.cdb_monthly_quota(
orgname TEXT, orgname TEXT,
service TEXT) service TEXT)
RETURNS integer AS $$ RETURNS integer AS $$
import cartodb_services.metrics.quota as quota
plpy.execute("SELECT cdb_dataservices_server._connect_to_redis('{0}')".format(username)) plpy.execute("SELECT cdb_dataservices_server._connect_to_redis('{0}')".format(username))
redis_conn = GD["redis_connection_{0}".format(username)]['redis_metrics_connection'] redis_conn = GD["redis_connection_{0}".format(username)]['redis_metrics_connection']
if service == 'isolines': if service == quota.Service.ISOLINES:
plpy.execute("SELECT cdb_dataservices_server._get_isolines_routing_config({0}, {1})".format(plpy.quote_nullable(username), plpy.quote_nullable(orgname))) plpy.execute("SELECT cdb_dataservices_server._get_isolines_routing_config({0}, {1})".format(plpy.quote_nullable(username), plpy.quote_nullable(orgname)))
user_isolines_config = GD["user_isolines_routing_config_{0}".format(username)] user_isolines_config = GD["user_isolines_routing_config_{0}".format(username)]
return user_isolines_config.isolines_quota monthly_quota = user_isolines_config.isolines_quota
elif service == quota.Service.HIRES_GEOCODER:
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)]
monthly_quota = user_geocoder_config.geocoding_quota
else: else:
raise 'not implemented' raise 'not implemented'
return monthly_quota
$$ LANGUAGE plpythonu; $$ LANGUAGE plpythonu;