Issue 2144 - [adblockplus.org Anwiki to CMS migration] get_subscriptions filter expects a local sub…

filters/get_subscriptions.py

Index: filters/get_subscriptions.py
===================================================================
--- a/filters/get_subscriptions.py
+++ b/filters/get_subscriptions.py
@@ -12,31 +12,43 @@
 #
 # 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 codecs
 import os
 import tarfile
 import urllib
+from ConfigParser import SafeConfigParser
 
 def get_subscriptions(_):
-  from sitescripts.subscriptions.subscriptionParser import parse_file, calculate_supplemented
+  from sitescripts.subscriptions import subscriptionParser
 
   result = {}
   utf8_reader = codecs.getreader('utf8')
   source = urllib.urlopen("https://hg.adblockplus.org/subscriptionlist/archive/default.tar.gz")
+  orig_get_settings = subscriptionParser.get_settings
   try:
+    # Hack: monkey-patch subscriptionParser.get_settings()
+    settings = SafeConfigParser()
+    settings_handle = urllib.urlopen("https://hg.adblockplus.org/subscriptionlist/rawfile/default/settings")

[Sebastian Noack, 2015/03/13 20:06:34]

Consider moving that logic into subscriptionParser where we could keep using hg,
just falling back to a remote URL, if there is no local repo, or something like
that.

[Wladimir Palant, 2015/03/13 20:08:54]

Yes, that's the idea - the hacks here are merely the short-term solution.

+    try:
+      settings.readfp(utf8_reader(settings_handle))
+    finally:
+      settings_handle.close()
+    subscriptionParser.get_settings = lambda: settings
+
     with tarfile.open(fileobj=source, mode="r|gz") as archive:
       for fileinfo in archive:
         if os.path.splitext(fileinfo.name)[1] != ".subscription":
           continue
 
-        filedata = parse_file(fileinfo.name, utf8_reader(archive.extractfile(fileinfo)))
+        filedata = subscriptionParser.parse_file(fileinfo.name, utf8_reader(archive.extractfile(fileinfo)))
         if filedata.unavailable:
           continue
 
         result[filedata.name] = filedata
   finally:
     source.close()
+    subscriptionParser.get_settings = orig_get_settings
 
-  calculate_supplemented(result)
+  subscriptionParser.calculate_supplemented(result)
   return result.values()