Issue 2846 - Service for processing Microsoft Edge announcement subscription

sitescripts/submit_email/web/submit_email.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 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'
+DEFAULT_PRODUCT = 'adblockbrowser'
 
 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))
 
+  product = data.get('product', DEFAULT_PRODUCT)
+  email_template = product + '_verification_email_template'
+  params.append(('product', product))
+
   sendMail(
-    config.get('submit_email', 'verification_email_template'),
+    config.get('submit_email', 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')
+  product = params.get('product', DEFAULT_PRODUCT)
+  filename = config.get('submit_email', product + '_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', ''))
   start_response('303 See Other', [('Location', location)])
   return []