Rename constructor arguments for consistency
This commit is contained in:
@@ -11,7 +11,7 @@ RETURNS JSON AS $$
|
|||||||
cartodb_services.init(plpy, GD)
|
cartodb_services.init(plpy, GD)
|
||||||
|
|
||||||
service_config = ServiceConfiguration(service, username, orgname)
|
service_config = ServiceConfiguration(service, username, orgname)
|
||||||
rate_limit_config = RateLimitsConfigBuilder(service_config.server, service_config.user, service_config.org, service=service, user=username, org=orgname).get()
|
rate_limit_config = RateLimitsConfigBuilder(service_config.server, service_config.user, service_config.org, service=service, username=username, orgname=orgname).get()
|
||||||
if rate_limit_config.is_limited():
|
if rate_limit_config.is_limited():
|
||||||
return json.dumps({'limit': rate_limit_config.limit, 'period': rate_limit_config.period})
|
return json.dumps({'limit': rate_limit_config.limit, 'period': rate_limit_config.period})
|
||||||
else:
|
else:
|
||||||
|
|||||||
@@ -6,10 +6,10 @@ class RateLimitsConfigLegacyBuilder(object):
|
|||||||
Build a RateLimitsConfig object using the *legacy* configuration classes
|
Build a RateLimitsConfig object using the *legacy* configuration classes
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def __init__(self, redis_connection, db_conn, service, user, org):
|
def __init__(self, redis_connection, db_conn, service, username, orgname):
|
||||||
self._service = service
|
self._service = service
|
||||||
self._username = user
|
self._username = username
|
||||||
self._orgname = org
|
self._orgname = orgname
|
||||||
self._redis_connection = redis_connection
|
self._redis_connection = redis_connection
|
||||||
self._db_conn = db_conn
|
self._db_conn = db_conn
|
||||||
|
|
||||||
|
|||||||
@@ -49,14 +49,13 @@ class RateLimitsConfigBuilder(object):
|
|||||||
from the user/org/server configuration.
|
from the user/org/server configuration.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
# TODO: user->username, org->orgname
|
def __init__(self, server_conf, user_conf, org_conf, service, username, orgname):
|
||||||
def __init__(self, server_conf, user_conf, org_conf, service, user, org):
|
|
||||||
self._server_conf = server_conf
|
self._server_conf = server_conf
|
||||||
self._user_conf = user_conf
|
self._user_conf = user_conf
|
||||||
self._org_conf = org_conf
|
self._org_conf = org_conf
|
||||||
self._service = service
|
self._service = service
|
||||||
self._username = user
|
self._username = username
|
||||||
self._orgname = org
|
self._orgname = orgname
|
||||||
|
|
||||||
def get(self):
|
def get(self):
|
||||||
# Order of precedence is user_conf, org_conf, server_conf
|
# Order of precedence is user_conf, org_conf, server_conf
|
||||||
|
|||||||
@@ -19,5 +19,5 @@ class LegacyServiceManager(ServiceManagerBase):
|
|||||||
|
|
||||||
self.quota_service = QuotaService(self.config, redis_conn)
|
self.quota_service = QuotaService(self.config, redis_conn)
|
||||||
|
|
||||||
rate_limit_config = RateLimitsConfigLegacyBuilder(redis_conn, plpy, service=service, user=username, org=orgname).get()
|
rate_limit_config = RateLimitsConfigLegacyBuilder(redis_conn, plpy, service=service, username=username, orgname=orgname).get()
|
||||||
self.rate_limiter = RateLimiter(rate_limit_config, redis_conn)
|
self.rate_limiter = RateLimiter(rate_limit_config, redis_conn)
|
||||||
|
|||||||
@@ -62,7 +62,7 @@ class ServiceManager(ServiceManagerBase):
|
|||||||
self.logger = Logger(logger_config)
|
self.logger = Logger(logger_config)
|
||||||
|
|
||||||
self.config = config_builder(service_config.server, service_config.user, service_config.org, username, orgname).get()
|
self.config = config_builder(service_config.server, service_config.user, service_config.org, username, orgname).get()
|
||||||
rate_limit_config = RateLimitsConfigBuilder(service_config.server, service_config.user, service_config.org, service=service, user=username, org=orgname).get()
|
rate_limit_config = RateLimitsConfigBuilder(service_config.server, service_config.user, service_config.org, service=service, username=username, orgname=orgname).get()
|
||||||
|
|
||||||
redis_metrics_connection = RedisMetricsConnectionFactory(service_config.environment, service_config.server).get()
|
redis_metrics_connection = RedisMetricsConnectionFactory(service_config.environment, service_config.server).get()
|
||||||
|
|
||||||
|
|||||||
@@ -62,8 +62,8 @@ class TestRateLimitsConfig(TestCase):
|
|||||||
user_conf=self.empty_redis_config,
|
user_conf=self.empty_redis_config,
|
||||||
org_conf=self.empty_redis_config,
|
org_conf=self.empty_redis_config,
|
||||||
service='geocoder',
|
service='geocoder',
|
||||||
user=self.username,
|
username=self.username,
|
||||||
org=self.orgname
|
orgname=self.orgname
|
||||||
).get()
|
).get()
|
||||||
plpy_mock._define_result("CDB_Conf_GetConf\('rate_limits'\)", [])
|
plpy_mock._define_result("CDB_Conf_GetConf\('rate_limits'\)", [])
|
||||||
assert_equal(read_config, config)
|
assert_equal(read_config, config)
|
||||||
@@ -88,8 +88,8 @@ class TestRateLimitsConfig(TestCase):
|
|||||||
user_conf=self.empty_redis_config,
|
user_conf=self.empty_redis_config,
|
||||||
org_conf=self.org_config,
|
org_conf=self.org_config,
|
||||||
service='geocoder',
|
service='geocoder',
|
||||||
user=self.username,
|
username=self.username,
|
||||||
org=self.orgname
|
orgname=self.orgname
|
||||||
).get()
|
).get()
|
||||||
plpy_mock._define_result("CDB_Conf_GetConf\('rate_limits'\)", [])
|
plpy_mock._define_result("CDB_Conf_GetConf\('rate_limits'\)", [])
|
||||||
assert_equal(read_config, org_config)
|
assert_equal(read_config, org_config)
|
||||||
@@ -117,8 +117,8 @@ class TestRateLimitsConfig(TestCase):
|
|||||||
user_conf=self.user_config,
|
user_conf=self.user_config,
|
||||||
org_conf=self.org_config,
|
org_conf=self.org_config,
|
||||||
service='geocoder',
|
service='geocoder',
|
||||||
user=self.username,
|
username=self.username,
|
||||||
org=self.orgname
|
orgname=self.orgname
|
||||||
).get()
|
).get()
|
||||||
plpy_mock._define_result("CDB_Conf_GetConf\('rate_limits'\)", [])
|
plpy_mock._define_result("CDB_Conf_GetConf\('rate_limits'\)", [])
|
||||||
assert_equal(read_config, user_config)
|
assert_equal(read_config, user_config)
|
||||||
|
|||||||
Reference in New Issue
Block a user