| Left: | ||
| Right: | 
| OLD | NEW | 
|---|---|
| 1 # coding: utf-8 | 1 # coding: utf-8 | 
| 2 | 2 | 
| 3 # This file is part of the Adblock Plus web scripts, | 3 # This file is part of the Adblock Plus web scripts, | 
| 4 # Copyright (C) 2006-2015 Eyeo GmbH | 4 # Copyright (C) 2006-2015 Eyeo GmbH | 
| 5 # | 5 # | 
| 6 # Adblock Plus is free software: you can redistribute it and/or modify | 6 # Adblock Plus is free software: you can redistribute it and/or modify | 
| 7 # it under the terms of the GNU General Public License version 3 as | 7 # it under the terms of the GNU General Public License version 3 as | 
| 8 # published by the Free Software Foundation. | 8 # published by the Free Software Foundation. | 
| 9 # | 9 # | 
| 10 # Adblock Plus is distributed in the hope that it will be useful, | 10 # Adblock Plus is distributed in the hope that it will be useful, | 
| (...skipping 26 matching lines...) Expand all Loading... | |
| 37 html_escapes = { | 37 html_escapes = { | 
| 38 "<": "<", | 38 "<": "<", | 
| 39 ">": ">", | 39 ">": ">", | 
| 40 "&": "&", | 40 "&": "&", | 
| 41 "\"": """, | 41 "\"": """, | 
| 42 "'": "'", | 42 "'": "'", | 
| 43 } | 43 } | 
| 44 | 44 | 
| 45 class AttributeParser(HTMLParser.HTMLParser): | 45 class AttributeParser(HTMLParser.HTMLParser): | 
| 46 _string = None | 46 _string = None | 
| 47 _inside_fixed = False | |
| 48 _fixed_strings = None | |
| 47 _attrs = None | 49 _attrs = None | 
| 48 | 50 | 
| 49 def __init__(self, whitelist): | 51 def __init__(self, whitelist): | 
| 50 self._whitelist = whitelist | 52 self._whitelist = whitelist | 
| 51 | 53 | 
| 52 def parse(self, text, pagename): | 54 def parse(self, text, pagename): | 
| 53 self.reset() | 55 self.reset() | 
| 54 self._string = [] | 56 self._string = [] | 
| 57 self._fixed_strings = [] | |
| 58 self._inside_fixed = False | |
| 55 self._attrs = {} | 59 self._attrs = {} | 
| 56 self._pagename = pagename | 60 self._pagename = pagename | 
| 57 | 61 | 
| 58 try: | 62 try: | 
| 59 self.feed(text) | 63 self.feed(text) | 
| 60 return "".join(self._string), self._attrs | 64 return "".join(self._string), self._attrs, map(lambda s: "".join(s), self. _fixed_strings) | 
| 
 
Wladimir Palant
2015/03/26 20:48:56
Nit: ["".join(s) for s in self._fixed_strings] is
 
kzar
2015/03/26 22:11:32
Done.
 
 | |
| 61 finally: | 65 finally: | 
| 62 self._string = None | 66 self._string = None | 
| 63 self._attrs = None | 67 self._attrs = None | 
| 64 self._pagename = None | 68 self._pagename = None | 
| 69 self._inside_fixed = False | |
| 70 self._fixed_strings = None | |
| 65 | 71 | 
| 66 def handle_starttag(self, tag, attrs): | 72 def handle_starttag(self, tag, attrs): | 
| 67 if tag not in self._whitelist: | 73 if tag not in self._whitelist: | 
| 
 
Wladimir Palant
2015/03/26 20:48:56
No tags should be allowed inside <fix> - meaning |
 
kzar
2015/03/26 22:11:32
Done.
 
 | |
| 68 raise Exception("Unexpected HTML tag '%s' in localizable string on page %s " % (tag, self._pagename)) | 74 raise Exception("Unexpected HTML tag '%s' in localizable string on page %s " % (tag, self._pagename)) | 
| 69 self._attrs.setdefault(tag, []).append(attrs) | 75 if tag == "fix": | 
| 70 self._string.append("<%s>" % tag) | 76 self._inside_fixed = True | 
| 77 self._fixed_strings.append([]) | |
| 78 else: | |
| 79 self._attrs.setdefault(tag, []).append(attrs) | |
| 80 self._string.append("<%s>" % tag) | |
| 71 | 81 | 
| 72 def handle_endtag(self, tag): | 82 def handle_endtag(self, tag): | 
| 73 self._string.append("</%s>" % tag) | 83 if tag == "fix": | 
| 84 self._string.append("{%d}" % len(self._fixed_strings)) | |
| 85 self._inside_fixed = False | |
| 86 else: | |
| 87 self._string.append("</%s>" % tag) | |
| 88 | |
| 89 def _append_string(self, s): | |
| 
 
Wladimir Palant
2015/03/26 20:48:56
Nit: we are adding to self._string a lot, this hel
 
kzar
2015/03/26 22:11:32
Done.
 
 | |
| 90 if self._inside_fixed: | |
| 91 self._fixed_strings[-1].append(s) | |
| 92 else: | |
| 93 self._string.append(s) | |
| 74 | 94 | 
| 75 def handle_data(self, data): | 95 def handle_data(self, data): | 
| 76 # Note: lack of escaping here is intentional. The result is a locale string, | 96 # Note: lack of escaping here is intentional. The result is a locale string, | 
| 77 # HTML escaping is applied when this string is inserted into the document. | 97 # HTML escaping is applied when this string is inserted into the document. | 
| 78 self._string.append(data) | 98 self._append_string(data) | 
| 79 | 99 | 
| 80 def handle_entityref(self, name): | 100 def handle_entityref(self, name): | 
| 81 self._string.append(self.unescape("&%s;" % name)) | 101 self._append_string(self.unescape("&%s;" % name)) | 
| 82 | 102 | 
| 83 def handle_charref(self, name): | 103 def handle_charref(self, name): | 
| 84 self._string.append(self.unescape("&#%s;" % name)) | 104 self._append_string(self.unescape("&#%s;" % name)) | 
| 85 | 105 | 
| 86 class Converter: | 106 class Converter: | 
| 87 whitelist = set(["a", "em", "strong"]) | 107 whitelist = set(["a", "em", "fix", "strong"]) | 
| 
 
Wladimir Palant
2015/03/26 20:48:56
It makes no sense to change this set - <fix> isn't
 
kzar
2015/03/26 22:11:32
Done.
 
 | |
| 88 | 108 | 
| 89 def __init__(self, params, key="pagedata"): | 109 def __init__(self, params, key="pagedata"): | 
| 90 self._params = params | 110 self._params = params | 
| 91 self._key = key | 111 self._key = key | 
| 92 self._attribute_parser = AttributeParser(self.whitelist) | 112 self._attribute_parser = AttributeParser(self.whitelist) | 
| 93 | 113 | 
| 94 # Read in any parameters specified at the beginning of the file | 114 # Read in any parameters specified at the beginning of the file | 
| 95 lines = params[key].splitlines(True) | 115 lines = params[key].splitlines(True) | 
| 96 while lines and re.search(r"^\s*[\w\-]+\s*=", lines[0]): | 116 while lines and re.search(r"^\s*[\w\-]+\s*=", lines[0]): | 
| 97 name, value = lines.pop(0).split("=", 1) | 117 name, value = lines.pop(0).split("=", 1) | 
| 98 params[name.strip()] = value.strip() | 118 params[name.strip()] = value.strip() | 
| 99 params[key] = "".join(lines) | 119 params[key] = "".join(lines) | 
| 100 | 120 | 
| 101 def localize_string(self, name, default, localedata, escapes): | 121 def localize_string(self, name, default, localedata, escapes): | 
| 102 def escape(s): | 122 def escape(s): | 
| 103 return re.sub(r".", | 123 return re.sub(r".", | 
| 104 lambda match: escapes.get(match.group(0), match.group(0)), | 124 lambda match: escapes.get(match.group(0), match.group(0)), | 
| 105 s, flags=re.S) | 125 s, flags=re.S) | 
| 106 def re_escape(s): | 126 def re_escape(s): | 
| 107 return re.escape(escape(s)) | 127 return re.escape(escape(s)) | 
| 108 | 128 | 
| 109 # Extract tag attributes from default string | 129 # Extract tag attributes from default string | 
| 110 default, saved_attributes = self._attribute_parser.parse(default, self._para ms["page"]) | 130 default, saved_attributes, fixed_strings = self._attribute_parser.parse(defa ult, self._params["page"]) | 
| 111 | 131 | 
| 112 # Get translation | 132 # Get translation | 
| 113 if self._params["locale"] != self._params["defaultlocale"] and name in local edata: | 133 if self._params["locale"] != self._params["defaultlocale"] and name in local edata: | 
| 114 result = localedata[name].strip() | 134 result = localedata[name].strip() | 
| 115 else: | 135 else: | 
| 116 result = default | 136 result = default | 
| 117 | 137 | 
| 138 # Insert fixed strings | |
| 139 for i in range(len(fixed_strings)): | |
| 140 result = re.sub(re_escape("{%d}" % (i + 1)), fixed_strings[i], result, 1) | |
| 
 
Wladimir Palant
2015/03/26 20:48:56
Escape the string being inserted? It looks like cu
 
kzar
2015/03/26 22:11:32
I originally did escape fixed_strings[i] here but
 
Wladimir Palant
2015/03/27 07:29:48
I see, you are inserting them before the escaping
 
 | |
| 141 | |
| 118 # Insert attributes | 142 # Insert attributes | 
| 119 result = escape(result) | 143 result = escape(result) | 
| 120 for tag in self.whitelist: | 144 for tag in self.whitelist: | 
| 121 saved = saved_attributes.get(tag, []) | 145 saved = saved_attributes.get(tag, []) | 
| 122 for attrs in saved: | 146 for attrs in saved: | 
| 123 attrs = map(lambda (name, value): '%s="%s"' % (escape(name), escape(valu e)), attrs) | 147 attrs = map(lambda (name, value): '%s="%s"' % (escape(name), escape(valu e)), attrs) | 
| 124 result = re.sub( | 148 result = re.sub( | 
| 125 r"%s([^<>]*?)%s" % (re_escape("<%s>" % tag), re_escape("</%s>" % tag)) , | 149 r"%s([^<>]*?)%s" % (re_escape("<%s>" % tag), re_escape("</%s>" % tag)) , | 
| 126 r'<%s %s>\1</%s>' % (tag, " ".join(attrs), tag), | 150 r'<%s %s>\1</%s>' % (tag, " ".join(attrs), tag), | 
| 127 result, 1, flags=re.S | 151 result, 1, flags=re.S | 
| (...skipping 217 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 345 stack.pop() | 369 stack.pop() | 
| 346 stack[-1]["subitems"].append(item) | 370 stack[-1]["subitems"].append(item) | 
| 347 stack.append(item) | 371 stack.append(item) | 
| 348 return structured | 372 return structured | 
| 349 | 373 | 
| 350 converters = { | 374 converters = { | 
| 351 "html": RawConverter, | 375 "html": RawConverter, | 
| 352 "md": MarkdownConverter, | 376 "md": MarkdownConverter, | 
| 353 "tmpl": TemplateConverter, | 377 "tmpl": TemplateConverter, | 
| 354 } | 378 } | 
| OLD | NEW |