github/reddit
1
1
Fork 0
mirror of https://github.com/reddit-archive/reddit.git synced 2026-04-05 03:00:15 -04:00
Code Issues Packages Projects Releases Wiki Activity

Limit requests from AppEngine apps by AppID not globally.

Browse Source
This commit is contained in:
Neil Williams
2012-02-01 13:28:27 -08:00
parent 8d9700cd33
commit 19dd459e94
1 changed files with 22 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
r2/r2/controllers/reddit_base.py
View File

@@ -413,15 +413,31 @@ def set_colors():
if color_rx.match(request.get.get('bordercolor') or ''):
c.bordercolor = request.get.get('bordercolor')
def ratelimit_agent(agent):
key = 'rate_agent_' + agent
if g.cache.get(key):
abort(429)
else:
g.cache.set(key, 't', time = 1)
appengine_re = re.compile(r'AppEngine-Google; \(\+http://code.google.com/appengine; appid: s~([a-z0-9-]{6,30})\)\Z')
def ratelimit_agents():
user_agent = request.user_agent
if not user_agent:
return
# parse out the appid for appengine apps
appengine_match = appengine_re.match(user_agent)
if appengine_match:
appid = appengine_match.group(1)
ratelimit_agent(appid)
return
user_agent = user_agent.lower()
for s in g.agents:
if s and user_agent and s in user_agent.lower():
key = 'rate_agent_' + s
if g.cache.get(s):
abort(429)
else:
g.cache.set(s, 't', time = 1)
if s and user_agent and s in user_agent:
ratelimit_agent(s)
def throttled(key):
return g.cache.get("throttle_" + key)