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

Add cdb_remaining_quota to server #302

Browse Source
This commit is contained in:
Rafa de la Torre 2016-11-08 18:47:22 +01:00
parent b00edf218a
commit 0e38af6ecd
1 changed files with 24 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
server/extension/sql/200_quotas.sql
View File

@ -13,3 +13,27 @@ RETURNS integer AS $$
else:
raise 'not implemented'
$$ LANGUAGE plpythonu;
CREATE OR REPLACE FUNCTION cdb_dataservices_server.cdb_remaining_quota(
username TEXT,
orgname TEXT,
service TEXT)
RETURNS integer AS $$
from datetime import date
from cartodb_services.metrics.user import UserMetricsService
plpy.execute("SELECT cdb_dataservices_server._connect_to_redis('{0}')".format(username))
redis_conn = GD["redis_connection_{0}".format(username)]['redis_metrics_connection']
if service == 'isolines':
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)]
monthly_quota = user_isolines_config.isolines_quota
user_service = UserMetricsService(user_isolines_config, redis_conn)
current_used = user_service.used_quota(user_isolines_config.service_type, date.today())
return monthly_quota - current_used
else:
raise 'not implemented'
$$ LANGUAGE plpythonu;