From 3ada3aa4be127bd51c3e200eb818232e5d9e02d3 Mon Sep 17 00:00:00 2001
From: Neil Williams <neil@reddit.com>
Date: Wed, 2 Jan 2013 18:16:36 -0800
Subject: [PATCH] ErrorController: Fix Retry-After header not being sent.

---
 r2/r2/controllers/error.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/r2/r2/controllers/error.py b/r2/r2/controllers/error.py
index ca42a3ab9..6b3c22f4c 100644
--- a/r2/r2/controllers/error.py
+++ b/r2/r2/controllers/error.py
@@ -28,7 +28,7 @@ import pylons
 
 from paste.httpexceptions import HTTPFound, HTTPMovedPermanently
 from pylons.i18n import _
-from pylons import c, g, request
+from pylons import c, g, request, response
 
 try:
     # place all r2 specific imports in here.  If there is a code error, it'll
@@ -133,8 +133,9 @@ class ErrorController(RedditController):
     def send429(self):
         c.response.status_code = 429
 
-        if 'retry_after' in request.environ:
-            c.response.headers['Retry-After'] = str(request.environ['retry_after'])
+        retry_after = request.environ.get("retry_after")
+        if retry_after:
+            response.headers["Retry-After"] = str(retry_after)
             template_name = '/ratelimit_toofast.html'
         else:
             template_name = '/ratelimit_throttled.html'