Left: | ||
Right: |
LEFT | RIGHT |
---|---|
1 # This file is part of the Adblock Plus website, | 1 # This file is part of the Adblock Plus website, |
2 # Copyright (C) 2006-2015 Eyeo GmbH | 2 # Copyright (C) 2006-2015 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 codecs | 16 import codecs |
17 import os | 17 import os |
18 import tarfile | 18 import tarfile |
19 import urllib | 19 import urllib |
20 from ConfigParser import SafeConfigParser | 20 from ConfigParser import SafeConfigParser |
21 | 21 |
22 def get_subscriptions(_): | 22 def get_subscriptions(_): |
23 import sitescripts.subscriptions.subscriptionParser as subscriptionParser | 23 from sitescripts.subscriptions import subscriptionParser |
Sebastian Noack
2015/03/13 19:51:45
Nit: from ... import
Wladimir Palant
2015/03/13 19:58:51
Done.
| |
24 | 24 |
25 result = {} | 25 result = {} |
26 utf8_reader = codecs.getreader('utf8') | 26 utf8_reader = codecs.getreader('utf8') |
27 source = urllib.urlopen("https://hg.adblockplus.org/subscriptionlist/archive/d efault.tar.gz") | 27 source = urllib.urlopen("https://hg.adblockplus.org/subscriptionlist/archive/d efault.tar.gz") |
28 orig_get_settings = subscriptionParser.get_settings | 28 orig_get_settings = subscriptionParser.get_settings |
29 try: | 29 try: |
30 # Hack: monkey-patch subscriptionParser.get_settings() | 30 # Hack: monkey-patch subscriptionParser.get_settings() |
31 settings = SafeConfigParser() | 31 settings = SafeConfigParser() |
32 settings.readfp(utf8_reader(urllib.urlopen("https://hg.adblockplus.org/subsc riptionlist/rawfile/default/settings"))) | 32 settings_handle = urllib.urlopen("https://hg.adblockplus.org/subscriptionlis t/rawfile/default/settings") |
Sebastian Noack
2015/03/13 19:51:45
The file-like object returned by urllib.urlopen()
Wladimir Palant
2015/03/13 19:58:51
Done.
Sebastian Noack
2015/03/13 20:06:34
I verified that __enter__/__exit__ isnt't implemen
|
Sebastian Noack
2015/03/13 20:06:34
Consider moving that logic into subscriptionParser
Wladimir Palant
2015/03/13 20:08:54
Yes, that's the idea - the hacks here are merely t
|
33 try: | |
34 settings.readfp(utf8_reader(settings_handle)) | |
35 finally: | |
36 settings_handle.close() | |
33 subscriptionParser.get_settings = lambda: settings | 37 subscriptionParser.get_settings = lambda: settings |
34 | 38 |
35 with tarfile.open(fileobj=source, mode="r|gz") as archive: | 39 with tarfile.open(fileobj=source, mode="r|gz") as archive: |
36 for fileinfo in archive: | 40 for fileinfo in archive: |
37 if os.path.splitext(fileinfo.name)[1] != ".subscription": | 41 if os.path.splitext(fileinfo.name)[1] != ".subscription": |
38 continue | 42 continue |
39 | 43 |
40 filedata = subscriptionParser.parse_file(fileinfo.name, utf8_reader(arch ive.extractfile(fileinfo))) | 44 filedata = subscriptionParser.parse_file(fileinfo.name, utf8_reader(arch ive.extractfile(fileinfo))) |
41 if filedata.unavailable: | 45 if filedata.unavailable: |
42 continue | 46 continue |
43 | 47 |
44 result[filedata.name] = filedata | 48 result[filedata.name] = filedata |
45 finally: | 49 finally: |
46 source.close() | 50 source.close() |
47 subscriptionParser.get_settings = orig_get_settings | 51 subscriptionParser.get_settings = orig_get_settings |
48 | 52 |
49 subscriptionParser.calculate_supplemented(result) | 53 subscriptionParser.calculate_supplemented(result) |
50 return result.values() | 54 return result.values() |
LEFT | RIGHT |