56 lines
2.8 KiB
Python
56 lines
2.8 KiB
Python
from datetime import datetime, date
|
|
from mock import Mock
|
|
|
|
|
|
def build_redis_user_config(redis_conn, username, quota=100, soft_limit=False,
|
|
service="heremaps", isolines_quota=0,
|
|
end_date=datetime.today()):
|
|
user_redis_name = "rails:users:{0}".format(username)
|
|
redis_conn.hset(user_redis_name, 'soft_geocoding_limit', soft_limit)
|
|
redis_conn.hset(user_redis_name, 'geocoding_quota', quota)
|
|
redis_conn.hset(user_redis_name, 'here_isolines_quota', isolines_quota)
|
|
redis_conn.hset(user_redis_name, 'geocoder_type', service)
|
|
redis_conn.hset(user_redis_name, 'period_end_date', end_date)
|
|
redis_conn.hset(user_redis_name, 'google_maps_client_id', '')
|
|
redis_conn.hset(user_redis_name, 'google_maps_api_key', '')
|
|
|
|
|
|
def build_redis_org_config(redis_conn, orgname, quota=100, isolines_quota=0,
|
|
end_date=datetime.today()):
|
|
org_redis_name = "rails:orgs:{0}".format(orgname)
|
|
redis_conn.hset(org_redis_name, 'geocoding_quota', quota)
|
|
redis_conn.hset(org_redis_name, 'here_isolines_quota', isolines_quota)
|
|
redis_conn.hset(org_redis_name, 'period_end_date', end_date)
|
|
redis_conn.hset(org_redis_name, 'google_maps_client_id', '')
|
|
redis_conn.hset(org_redis_name, 'google_maps_api_key', '')
|
|
|
|
|
|
def increment_geocoder_uses(redis_conn, username, orgname=None,
|
|
date=date.today(), service='geocoder_here',
|
|
metric='success_responses', amount=20):
|
|
prefix = 'org' if orgname else 'user'
|
|
entity_name = orgname if orgname else username
|
|
yearmonth = date.strftime('%Y%m')
|
|
redis_name = "{0}:{1}:{2}:{3}:{4}".format(prefix, entity_name,
|
|
service, metric, yearmonth)
|
|
redis_conn.zincrby(redis_name, date.day, amount)
|
|
|
|
|
|
def build_plpy_mock(empty=False):
|
|
plpy_mock = Mock()
|
|
if not empty:
|
|
plpy_mock.execute.side_effect = _plpy_execute_side_effect
|
|
|
|
return plpy_mock
|
|
|
|
|
|
def _plpy_execute_side_effect(*args, **kwargs):
|
|
if args[0] == "SELECT cartodb.CDB_Conf_GetConf('heremaps_conf') as conf":
|
|
return [{'conf': '{"geocoder": {"app_id": "app_id", "app_code": "code", "geocoder_cost_per_hit": 1}, "isolines": {"app_id": "app_id", "app_code": "code"}}'}]
|
|
elif args[0] == "SELECT cartodb.CDB_Conf_GetConf('mapzen_conf') as conf":
|
|
return [{'conf': '{"routing": {"api_key": "valhalla-Z61FWEs", "monthly_quota": 1500000}, "geocoder": {"api_key": "search-d744tp0", "monthly_quota": 1500000}}'}]
|
|
elif args[0] == "SELECT cartodb.CDB_Conf_GetConf('logger_conf') as conf":
|
|
return [{'conf': '{"geocoder_log_path": "/dev/null"}'}]
|
|
elif args[0] == "SELECT cartodb.CDB_Conf_GetConf('data_observatory_conf') as conf":
|
|
return [{'conf': '{"monthly_quota": 100000}'}]
|