| OLD | NEW |
| (Empty) |
| 1 # coding: utf-8 | |
| 2 | |
| 3 # This file is part of the Adblock Plus web scripts, | |
| 4 # Copyright (C) 2006-2012 Eyeo GmbH | |
| 5 # | |
| 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 | |
| 8 # published by the Free Software Foundation. | |
| 9 # | |
| 10 # Adblock Plus is distributed in the hope that it will be useful, | |
| 11 # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
| 13 # GNU General Public License for more details. | |
| 14 # | |
| 15 # You should have received a copy of the GNU General Public License | |
| 16 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. | |
| 17 | |
| 18 import MySQLdb, os, re, subprocess | |
| 19 from sitescripts.utils import get_config | |
| 20 | |
| 21 def hg(args): | |
| 22 return subprocess.Popen(["hg"] + args, stdout = subprocess.PIPE) | |
| 23 | |
| 24 def extract_urls(filter_list_dir): | |
| 25 os.chdir(filter_list_dir) | |
| 26 process = hg(["log", "--template", "{desc}\n"]) | |
| 27 urls = set([]) | |
| 28 | |
| 29 for line in process.stdout: | |
| 30 match = re.search(r"\b(https?://\S*)", line) | |
| 31 if not match: | |
| 32 continue | |
| 33 | |
| 34 url = match.group(1).strip() | |
| 35 urls.add(url) | |
| 36 | |
| 37 return urls | |
| 38 | |
| 39 def print_statements(urls): | |
| 40 for url in urls: | |
| 41 escaped_url = MySQLdb.escape_string(url) | |
| 42 print "INSERT INTO crawler_sites (url) VALUES ('" + escaped_url + "');" | |
| 43 | |
| 44 if __name__ == "__main__": | |
| 45 filter_list_dir = get_config().get("crawler", "filter_list_repository") | |
| 46 urls = extract_urls(filter_list_dir) | |
| 47 print_statements(urls) | |
| OLD | NEW |