| OLD | NEW | 
|---|
| 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 copy | 16 import copy | 
| 17 import json | 17 import json | 
| 18 import random | 18 import random | 
| 19 import time | 19 import time | 
| 20 import urlparse | 20 import urlparse | 
| 21 | 21 | 
| 22 from sitescripts.notifications.parser import load_notifications | 22 from sitescripts.notifications.parser import load_notifications | 
| 23 from sitescripts.web import url_handler | 23 from sitescripts.web import url_handler | 
| 24 | 24 | 
| 25 | 25 | 
| 26 def _determine_groups(version, notifications): | 26 def _determine_groups(version, notifications): | 
| 27     version_groups = dict(x.split("/") for x in version.split("-")[1:] | 27     version_groups = dict(x.split('/') for x in version.split('-')[1:] | 
| 28                           if x.count("/") == 1) | 28                           if x.count('/') == 1) | 
| 29     groups = [] | 29     groups = [] | 
| 30     for notification in notifications: | 30     for notification in notifications: | 
| 31         group_id = notification["id"] | 31         group_id = notification['id'] | 
| 32         if group_id in version_groups: | 32         if group_id in version_groups: | 
| 33             groups.append({"id": group_id, "variant": int(version_groups[group_i
     d])}) | 33             groups.append({'id': group_id, 'variant': int(version_groups[group_i
     d])}) | 
| 34     return groups | 34     return groups | 
| 35 | 35 | 
| 36 | 36 | 
| 37 def _assign_groups(notifications): | 37 def _assign_groups(notifications): | 
| 38     groups = [] | 38     groups = [] | 
| 39     selection = random.random() | 39     selection = random.random() | 
| 40     start = 0 | 40     start = 0 | 
| 41     for notification in notifications: | 41     for notification in notifications: | 
| 42         if "variants" not in notification: | 42         if 'variants' not in notification: | 
| 43             continue | 43             continue | 
| 44         group = {"id": notification["id"], "variant": 0} | 44         group = {'id': notification['id'], 'variant': 0} | 
| 45         groups.append(group) | 45         groups.append(group) | 
| 46         for i, variant in enumerate(notification["variants"]): | 46         for i, variant in enumerate(notification['variants']): | 
| 47             sample_size = variant["sample"] | 47             sample_size = variant['sample'] | 
| 48             end = start + sample_size | 48             end = start + sample_size | 
| 49             selected = sample_size > 0 and start <= selection <= end | 49             selected = sample_size > 0 and start <= selection <= end | 
| 50             start = end | 50             start = end | 
| 51             if selected: | 51             if selected: | 
| 52                 group["variant"] = i + 1 | 52                 group['variant'] = i + 1 | 
| 53                 break | 53                 break | 
| 54     return groups | 54     return groups | 
| 55 | 55 | 
| 56 | 56 | 
| 57 def _get_active_variant(notifications, groups): | 57 def _get_active_variant(notifications, groups): | 
| 58     for group in groups: | 58     for group in groups: | 
| 59         group_id = group["id"] | 59         group_id = group['id'] | 
| 60         variant = group["variant"] | 60         variant = group['variant'] | 
| 61         if variant == 0: | 61         if variant == 0: | 
| 62             continue | 62             continue | 
| 63         notification = next((x for x in notifications if x["id"] == group_id), N
     one) | 63         notification = next((x for x in notifications if x['id'] == group_id), N
     one) | 
| 64         if not notification: | 64         if not notification: | 
| 65             continue | 65             continue | 
| 66         notification = copy.deepcopy(notification) | 66         notification = copy.deepcopy(notification) | 
| 67         notification.update(notification["variants"][variant - 1]) | 67         notification.update(notification['variants'][variant - 1]) | 
| 68         for key_to_remove in ("sample", "variants"): | 68         for key_to_remove in ('sample', 'variants'): | 
| 69             notification.pop(key_to_remove, None) | 69             notification.pop(key_to_remove, None) | 
| 70         return notification | 70         return notification | 
| 71 | 71 | 
| 72 | 72 | 
| 73 def _can_be_shown(notification): | 73 def _can_be_shown(notification): | 
| 74     return notification.get("title", None) and notification.get("message", None) | 74     return notification.get('title', None) and notification.get('message', None) | 
| 75 | 75 | 
| 76 | 76 | 
| 77 def _generate_version(groups): | 77 def _generate_version(groups): | 
| 78     version = time.strftime("%Y%m%d%H%M", time.gmtime()) | 78     version = time.strftime('%Y%m%d%H%M', time.gmtime()) | 
| 79     for group in groups: | 79     for group in groups: | 
| 80         version += "-%s/%s" % (group["id"], group["variant"]) | 80         version += '-%s/%s' % (group['id'], group['variant']) | 
| 81     return version | 81     return version | 
| 82 | 82 | 
| 83 | 83 | 
| 84 def _get_notifications_to_send(notifications, groups): | 84 def _get_notifications_to_send(notifications, groups): | 
| 85     active_variant = _get_active_variant(notifications, groups) | 85     active_variant = _get_active_variant(notifications, groups) | 
| 86     if active_variant: | 86     if active_variant: | 
| 87         return [active_variant] if _can_be_shown(active_variant) else [] | 87         return [active_variant] if _can_be_shown(active_variant) else [] | 
| 88 | 88 | 
| 89     notifications_to_send = [] | 89     notifications_to_send = [] | 
| 90     for notification in notifications: | 90     for notification in notifications: | 
| 91         if not _can_be_shown(notification): | 91         if not _can_be_shown(notification): | 
| 92             continue | 92             continue | 
| 93         if "variants" in notification: | 93         if 'variants' in notification: | 
| 94             notification = copy.deepcopy(notification) | 94             notification = copy.deepcopy(notification) | 
| 95             del notification["variants"] | 95             del notification['variants'] | 
| 96         notifications_to_send.append(notification) | 96         notifications_to_send.append(notification) | 
| 97     return notifications_to_send | 97     return notifications_to_send | 
| 98 | 98 | 
| 99 | 99 | 
| 100 def _create_response(notifications, groups): | 100 def _create_response(notifications, groups): | 
| 101     return { | 101     return { | 
| 102         "version": _generate_version(groups), | 102         'version': _generate_version(groups), | 
| 103         "notifications": _get_notifications_to_send(notifications, groups) | 103         'notifications': _get_notifications_to_send(notifications, groups) | 
| 104     } | 104     } | 
| 105 | 105 | 
| 106 | 106 | 
| 107 @url_handler("/notification.json") | 107 @url_handler('/notification.json') | 
| 108 def notification(environ, start_response): | 108 def notification(environ, start_response): | 
| 109     params = urlparse.parse_qs(environ.get("QUERY_STRING", "")) | 109     params = urlparse.parse_qs(environ.get('QUERY_STRING', '')) | 
| 110     version = params.get("lastVersion", [""])[0] | 110     version = params.get('lastVersion', [''])[0] | 
| 111     notifications = load_notifications() | 111     notifications = load_notifications() | 
| 112     groups = _determine_groups(version, notifications) | 112     groups = _determine_groups(version, notifications) | 
| 113     notifications = [x for x in notifications if not x.get("inactive", False)] | 113     notifications = [x for x in notifications if not x.get('inactive', False)] | 
| 114     if not groups: | 114     if not groups: | 
| 115         groups = _assign_groups(notifications) | 115         groups = _assign_groups(notifications) | 
| 116     response = _create_response(notifications, groups) | 116     response = _create_response(notifications, groups) | 
| 117     response_headers = [("Content-Type", "application/json; charset=utf-8"), | 117     response_headers = [('Content-Type', 'application/json; charset=utf-8'), | 
| 118                         ("ABP-Notification-Version", response["version"])] | 118                         ('ABP-Notification-Version', response['version'])] | 
| 119     response_body = json.dumps(response, ensure_ascii=False, indent=2, | 119     response_body = json.dumps(response, ensure_ascii=False, indent=2, | 
| 120                                separators=(",", ": "), | 120                                separators=(',', ': '), | 
| 121                                sort_keys=True).encode("utf-8") | 121                                sort_keys=True).encode('utf-8') | 
| 122     start_response("200 OK", response_headers) | 122     start_response('200 OK', response_headers) | 
| 123     return response_body | 123     return response_body | 
| OLD | NEW | 
|---|