diff --git a/server/extension/sql/200_quotas.sql b/server/extension/sql/200_quotas.sql index 0b8ccd2..925dc3d 100644 --- a/server/extension/sql/200_quotas.sql +++ b/server/extension/sql/200_quotas.sql @@ -85,6 +85,18 @@ RETURNS SETOF cdb_dataservices_server.service_params AS $$ provider = user_obs_config.provider ret += [[service, monthly_quota, used_quota, soft_limit, provider]] + #-- Internal geocoder + service = 'internal_geocoder' + plpy.execute("SELECT cdb_dataservices_server._get_internal_geocoder_config({0}, {1})".format(plpy.quote_nullable(username), plpy.quote_nullable(orgname))) + user_geocoder_config = GD["user_internal_geocoder_config_{0}".format(username)] + user_service = UserMetricsService(user_geocoder_config, redis_conn) + + monthly_quota = user_geocoder_config.monthly_quota + used_quota = user_service.used_quota(user_geocoder_config.service_type, today) + soft_limit = user_geocoder_config.soft_limit + provider = user_geocoder_config.provider + ret += [[service, monthly_quota, used_quota, soft_limit, provider]] + return ret $$ LANGUAGE plpythonu;