Rietveld Code Review Tool
Help | Bug tracker | Discussion group | Source code

Unified Diff: sitescripts/submit_email/web/submit_email.py

Issue 5177883412660224: Issue 2234 - Add a WSGI controller to collect email addresses for the Adblock Browser iOS launch (Closed)
Patch Set: URL-encode language before inserting into URL Created April 28, 2015, 10:50 a.m.
Use n/p to move between diff chunks; N/P to move between comments.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « sitescripts/submit_email/web/__init__.py ('k') | sitescripts/utils.py » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: sitescripts/submit_email/web/submit_email.py
===================================================================
new file mode 100644
--- /dev/null
+++ b/sitescripts/submit_email/web/submit_email.py
@@ -0,0 +1,93 @@
+# 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 fcntl
+import hmac
+import hashlib
+import wsgiref.util
+from urlparse import parse_qsl, urljoin
+from urllib import urlencode, quote
+
+from sitescripts.utils import get_config, sendMail, encode_email_address
+from sitescripts.web import url_handler, form_handler, send_simple_response
+
+VERIFICATION_PATH = '/verifyEmail'
+
+def sign(config, data):
+ secret = config.get('submit_email', 'secret')
+ return hmac.new(secret, data, hashlib.sha1).hexdigest()
+
+@url_handler('/submitEmail')
+@form_handler
+def submit_email(environ, start_response, data):
+ email = data.get('email', '').strip()
+ try:
+ email = encode_email_address(email)
+ except ValueError:
+ return send_simple_response(
+ start_response, 400,
+ 'Please enter a valid email address.'
+ )
+
+ config = get_config()
+ params = [('email', email), ('signature', sign(config, email))]
+ lang = data.get('lang')
+ if lang:
+ params.append(('lang', lang))
+
+ sendMail(
+ config.get('submit_email', 'verification_email_template'),
+ {
+ 'recipient': email,
+ 'verification_url': '%s?%s' % (
+ urljoin(wsgiref.util.application_uri(environ), VERIFICATION_PATH),
+ urlencode(params)
+ )
+ }
+ )
+
+ return send_simple_response(
+ start_response, 200,
+ 'A confirmation email has been sent. Please check '
+ 'your email and click the confirmation link.'
+ )
+
+@url_handler(VERIFICATION_PATH)
+def verify_email(environ, start_response):
+ config = get_config()
+ params = dict(parse_qsl(environ.get('QUERY_STRING', '')))
+
+ email = params.get('email', '')
+ signature = params.get('signature', '')
+ if sign(config, email) != signature:
+ return send_simple_response(
+ start_response, 403,
+ 'Invalid signature in verification request.'
+ )
+
+ filename = config.get('submit_email', 'filename')
+ with open(filename, 'ab', 0) as file:
+ fcntl.lockf(file, fcntl.LOCK_EX)
+ try:
+ print >>file, email
+ finally:
+ fcntl.lockf(file, fcntl.LOCK_UN)
+
+ location = config.get('submit_email', 'successful_verification_redirect_location')
+ location = location.format(lang=quote(params.get('lang') or 'en', ''))
kzar 2015/04/28 11:13:42 Looks like a typo at the end there? ", ''"
Sebastian Noack 2015/04/28 11:31:25 The empty string at the end is the second argument
kzar 2015/04/28 11:33:41 Oh I see.
Sebastian Noack 2015/04/28 11:36:04 I meant (forward) slashes.
+ start_response('303 See Other', [('Location', location)])
+ return []
« no previous file with comments | « sitescripts/submit_email/web/__init__.py ('k') | sitescripts/utils.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld