| Left: | ||
| Right: |
| LEFT | RIGHT |
|---|---|
| 1 # This file is part of the Adblock Plus web scripts, | 1 # This file is part of the Adblock Plus web scripts, |
| 2 # Copyright (C) 2006-2016 Eyeo GmbH | 2 # Copyright (C) 2006-2016 Eyeo GmbH |
| 3 # | 3 # |
| 4 # Adblock Plus is free software: you can redistribute it and/or modify | 4 # Adblock Plus is free software: you can redistribute it and/or modify |
| 5 # it under the terms of the GNU General Public License version 3 as | 5 # it under the terms of the GNU General Public License version 3 as |
| 6 # published by the Free Software Foundation. | 6 # published by the Free Software Foundation. |
| 7 # | 7 # |
| 8 # Adblock Plus is distributed in the hope that it will be useful, | 8 # Adblock Plus is distributed in the hope that it will be useful, |
| 9 # but WITHOUT ANY WARRANTY; without even the implied warranty of | 9 # but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | 10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 11 # GNU General Public License for more details. | 11 # GNU General Public License for more details. |
| 12 # | 12 # |
| 13 # You should have received a copy of the GNU General Public License | 13 # You should have received a copy of the GNU General Public License |
| 14 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. | 14 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. |
| 15 | 15 |
| 16 import re | 16 import re |
| 17 import datetime | 17 import datetime |
| 18 import collections | 18 import collections |
| 19 from urlparse import parse_qsl | 19 from urlparse import parse_qsl |
| 20 from sitescripts.utils import get_config, sendMail, setupStderr | 20 from sitescripts.utils import get_config, sendMail, setupStderr |
|
Sebastian Noack
2016/09/12 22:11:49
There should be a blank line between corelib and o
Vasily Kuznetsov
2016/09/13 09:03:28
Done.
| |
| 21 from sitescripts.web import registerUrlHandler | 21 from sitescripts.web import registerUrlHandler |
| 22 | 22 |
| 23 | 23 |
| 24 class BadRequestError(Exception): | 24 class BadRequestError(Exception): |
| 25 pass | 25 pass |
| 26 | 26 |
| 27 | 27 |
| 28 class ConfDict(collections.OrderedDict): | 28 class ConfDict(collections.OrderedDict): |
| 29 __slots__ = ('value',) | 29 __slots__ = ('value',) |
|
Sebastian Noack
2016/09/12 22:11:50
This micro optimization doesn't do anything, since
Vasily Kuznetsov
2016/09/13 07:15:53
Yeah, with OrderedDict __slots__ is pretty much us
Vasily Kuznetsov
2016/09/13 09:03:27
Done.
| |
| 30 | 30 |
| 31 | 31 |
| 32 def get_config_items(): | 32 def get_config_items(): |
| 33 config = get_config() | 33 config = get_config() |
| 34 default_keys = set(config.defaults()) | 34 default_keys = set(config.defaults()) |
| 35 for name, value in config.items('formmail2'): | 35 for name, value in config.items('formmail2'): |
| 36 if name not in default_keys: | 36 if name not in default_keys: |
| 37 yield name, value | 37 yield name, value |
| 38 | 38 |
| 39 | 39 |
| (...skipping 10 matching lines...) Expand all Loading... | |
| 50 def conf_parse(conf_items): | 50 def conf_parse(conf_items): |
| 51 conf_dict = ConfDict() | 51 conf_dict = ConfDict() |
| 52 for key, value in conf_items: | 52 for key, value in conf_items: |
| 53 path = key.split('.') | 53 path = key.split('.') |
| 54 store_value(conf_dict, path, value) | 54 store_value(conf_dict, path, value) |
| 55 return conf_dict | 55 return conf_dict |
| 56 | 56 |
| 57 | 57 |
| 58 def post_handler(handler): | 58 def post_handler(handler): |
| 59 def wrapped_handler(environ, start_response): | 59 def wrapped_handler(environ, start_response): |
| 60 setupStderr(environ['wsgi.errors']) | 60 setupStderr(environ['wsgi.errors']) |
|
Sebastian Noack
2016/09/12 22:11:49
setupStderr() is depreacted. You should actually s
Vasily Kuznetsov
2016/09/13 07:15:53
Yeah, I saw the warning. Since we're not writing a
Vasily Kuznetsov
2016/09/13 09:03:27
Done.
| |
| 61 response_headers = [('Content-Type', 'text/plain; charset=utf-8')] | 61 response_headers = [('Content-Type', 'text/plain; charset=utf-8')] |
| 62 | 62 |
| 63 try: | 63 try: |
| 64 request_method = environ['REQUEST_METHOD'].upper() | 64 request_method = environ['REQUEST_METHOD'].upper() |
| 65 url_encoded = 'application/x-www-form-urlencoded' | 65 url_encoded = 'application/x-www-form-urlencoded' |
| 66 is_url_encoded = environ.get( | 66 is_url_encoded = environ.get( |
| 67 'CONTENT_TYPE', '').startswith(url_encoded) | 67 'CONTENT_TYPE', '').startswith(url_encoded) |
| 68 if request_method != 'POST' or not is_url_encoded: | 68 if request_method != 'POST' or not is_url_encoded: |
| 69 raise BadRequestError('Unsupported request method') | 69 raise BadRequestError('Unsupported request method') |
| 70 try: | 70 try: |
| 71 request_body_length = int(environ['CONTENT_LENGTH']) | 71 request_body_length = int(environ['CONTENT_LENGTH']) |
|
Sebastian Noack
2016/09/12 22:11:49
If CONTENT_LENGTH, is not given but required, 411
Vasily Kuznetsov
2016/09/13 07:15:52
Yes, that decorator does pretty much everything th
Vasily Kuznetsov
2016/09/13 09:03:27
In the end after removing all `form_handler` logic
| |
| 72 except: | 72 except: |
| 73 raise BadRequestError( | 73 raise BadRequestError( |
| 74 'Invalid or missing Content-Length header') | 74 'Invalid or missing Content-Length header') |
| 75 request_body = environ['wsgi.input'].read(request_body_length) | 75 request_body = environ['wsgi.input'].read(request_body_length) |
| 76 params = {} | 76 params = {} |
| 77 for key, value in parse_qsl(request_body): | 77 for key, value in parse_qsl(request_body): |
| 78 params[key] = value.decode('utf-8').strip() | 78 params[key] = value.decode('utf-8').strip() |
| 79 | 79 |
| 80 response = handler(params) | 80 response = handler(params) |
| 81 except BadRequestError as error: | 81 except BadRequestError as error: |
| 82 start_response('400 Bad Request', response_headers) | 82 start_response('400 Bad Request', response_headers) |
| 83 return str(error) | 83 return str(error) |
| 84 start_response('200 OK', response_headers) | 84 start_response('200 OK', response_headers) |
| 85 return response | 85 return response |
| 86 return wrapped_handler | 86 return wrapped_handler |
| 87 | 87 |
| 88 | 88 |
| 89 def make_error(field_name, spec, check_type, error_message): | 89 def make_error(spec, check_type, default_message): |
| 90 if check_type in spec: | 90 if check_type in spec: |
| 91 return spec[field_name].value | 91 return spec[check_type].value |
| 92 return error_message.format(field_name) | 92 return default_message |
| 93 | 93 |
| 94 | 94 |
| 95 def make_handler(name, config): | 95 def make_handler(name, config): |
| 96 try: | 96 try: |
| 97 url = config['url'].value | 97 url = config['url'].value |
| 98 except (KeyError, AttributeError): | 98 except (KeyError, AttributeError): |
| 99 raise Exception('No URL configured for form handler:' + name) | 99 raise Exception('No URL configured for form handler:' + name) |
| 100 try: | 100 try: |
| 101 template = config['template'].value | 101 template = config['template'].value |
| 102 except (KeyError, AttributeError): | 102 except (KeyError, AttributeError): |
| 103 raise Exception('No template configured for form handler:' + name) | 103 raise Exception('No template configured for form handler:' + name) |
| 104 try: | 104 try: |
| 105 fields = config['fields'] | 105 fields = config['fields'] |
| 106 for field, spec in fields.items(): | 106 for field, spec in fields.items(): |
| 107 spec.value = set(spec.value.replace(' ', '').split(',')) | 107 spec.value = set(spec.value.replace(' ', '').split(',')) |
|
Sebastian Noack
2016/09/12 22:11:49
It seems you want to ignore optional spaces around
Vasily Kuznetsov
2016/09/13 07:15:53
Perhaps regexes are a bit overkill here. How about
Vasily Kuznetsov
2016/09/13 09:03:27
Done.
| |
| 108 except KeyError: | 108 except KeyError: |
| 109 raise Exception('No fields configured for form handler:' + name) | 109 raise Exception('No fields configured for form handler:' + name) |
| 110 if len(fields) == 0: | 110 if len(fields) == 0: |
| 111 raise Exception('No fields configured for form handler:' + name) | 111 raise Exception('No fields configured for form handler:' + name) |
| 112 | 112 |
| 113 @post_handler | 113 @post_handler |
| 114 def handler(params): | 114 def handler(params): |
| 115 email_regex = r'^\w[\w.+!-]+@\w[\w.-]+\.[a-zA-Z]{2,6}$' | 115 email_regex = r'^\w[\w.+!-]+@\w[\w.-]+\.[a-zA-Z]{2,6}$' |
|
Sebastian Noack
2016/09/12 22:11:50
There is utils.encode_email_address(), which we us
Vasily Kuznetsov
2016/09/13 07:15:52
Yeah, this regex was just copied from formmail.py.
Vasily Kuznetsov
2016/09/13 09:03:27
Done.
| |
| 116 errors = [] | 116 errors = [] |
| 117 for field, spec in fields.items(): | 117 for field, spec in fields.items(): |
| 118 if 'mandatory' in spec.value: | 118 if 'mandatory' in spec.value: |
| 119 if field not in params.keys(): | 119 if field not in params.keys(): |
| 120 errors.append((make_error(field, spec, 'mandatory', | 120 errors.append(make_error(spec, 'mandatory', |
| 121 'No {} entered'.format(field)))) | 121 'No {} entered'.format(field))) |
| 122 if 'email' in spec.value and field in params.keys(): | 122 if 'email' in spec.value and field in params.keys(): |
| 123 if not re.search(email_regex, params[field]): | 123 if not re.search(email_regex, params[field]): |
| 124 errors.append(make_error(field, spec, 'email', | 124 errors.append(make_error(spec, 'email', |
| 125 'Invalid email')) | 125 'Invalid email')) |
| 126 if errors: | 126 if errors: |
| 127 raise BadRequestError('\n'.join(errors)) | 127 raise BadRequestError('\n'.join(errors)) |
| 128 template_args = { | 128 template_args = { |
| 129 'time': datetime.datetime.now(), | 129 'time': datetime.datetime.now(), |
| 130 'fields': {field: params.get(field, '') for field in fields} | 130 'fields': {field: params.get(field, '') for field in fields} |
| 131 } | 131 } |
| 132 sendMail(template, template_args) | 132 sendMail(template, template_args) |
| 133 return '' | 133 return '' |
| 134 return url, handler | 134 return url, handler |
| 135 | 135 |
| 136 | |
| 137 conf_dict = conf_parse(get_config_items()) | 136 conf_dict = conf_parse(get_config_items()) |
| 138 for name, config in conf_dict.items(): | 137 for name, config in conf_dict.items(): |
| 139 url, handler = make_handler(name, config) | 138 url, handler = make_handler(name, config) |
| 140 registerUrlHandler(url, handler) | 139 registerUrlHandler(url, handler) |
| LEFT | RIGHT |