Issue 2234 - Add a WSGI controller to collect email addresses for the Adblock Browser iOS launch

sitescripts/web.py

 # coding: utf-8
 
 # This file is part of the Adblock Plus web scripts,
 # Copyright (C) 2006-2015 Eyeo GmbH
 #
 # Adblock Plus is free software: you can redistribute it and/or modify
 # it under the terms of the GNU General Public License version 3 as
 # published by the Free Software Foundation.
 #
 # Adblock Plus is distributed in the hope that it will be useful,
 # but WITHOUT ANY WARRANTY; without even the implied warranty of
 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 # GNU General Public License for more details.
 #
 # You should have received a copy of the GNU General Public License
 # along with Adblock Plus.  If not, see <http://www.gnu.org/licenses/>.
 
 import base64
 import imp
 import importlib
 import re
+import httplib
 from urlparse import parse_qsl
 
 from sitescripts.utils import get_config
 
 handlers = {}
 authenticated_users = {}
 
 def url_handler(url):
   def decorator(func):
     registerUrlHandler(url, func)
@@ skipping 22 matching lines @@
         expected_username = config.get(config_section, "basic_auth_username")
         expected_password = config.get(config_section, "basic_auth_password")
         if username == expected_username and password == expected_password:
           return f(environ, start_response)
 
   realm = get_config().get("DEFAULT", "basic_auth_realm")
   start_response("401 UNAUTHORIZED",
                  [("WWW-Authenticate", 'Basic realm="%s"' % realm)])
   return ""
 
-def send_status_response(status, start_response):
+def send_simple_response(start_response, status_code, text=None):
+  status_text = httplib.responses[status_code]
+
+  status = '%d %s' % (status_code, status_text)
   start_response(status, [('Content-Type', 'text/plain')])
-  return [status]
+
+  if text is not None:
+    return [text]
+  return [status_text]
 
 def form_handler(func):
   def wrapper(environ, start_response):
     if environ['REQUEST_METHOD'] != 'POST':
-      return send_status_response('405 Method Not Allowed', start_response)
+      return send_simple_response(start_response, 405)
 
     if not environ.get('CONTENT_TYPE', '').startswith('application/x-www-form-urlencoded'):
-      return send_status_response('415 Unsupported Media Type', start_response)
+      return send_simple_response(start_response, 415)
 
     try:
       content_length = int(environ['CONTENT_LENGTH'])
     except (KeyError, ValueError):
       content_length = None
     if content_length is None or content_length < 0:
-      return send_status_response('411 Length Required', start_response)
+      return send_simple_response(start_response, 411)
 
     raw_data = parse_qsl(environ['wsgi.input'].read(content_length))
     try:
       data = {k.decode('utf-8'): v.decode('utf-8') for k, v in raw_data}
     except UnicodeDecodeError:
-      return send_status_response('400 Invalid character in form data', start_response)
+      return send_simple_response(start_response, 400, 'Invalid form data encoding')
 
     return func(environ, start_response, data)
 
   return wrapper
 
 def multiplex(environ, start_response):
   try:
     path = environ["PATH_INFO"]
     try:
       handler = handlers[path]
     except KeyError:
       handler = handlers[re.sub(r"[^/]+$", "", path)]
   except KeyError:
     start_response("404 Not Found", [("Content-Type", "text/plain")])
     return ["Not Found"]
 
   return handler(environ, start_response)
 
 for module in set(get_config().options("multiplexer")) - set(get_config().defaults()):
   module_path = get_config().get("multiplexer", module)
   if module_path:
     imp.load_source(module, module_path)
   else:
     importlib.import_module(module)