2
0
mirror of https://github.com/WikiTeam/wikiteam synced 2024-11-18 21:27:45 +00:00
wikiteam/wikipediadownloader.py
emijrp ab64c8f1c8 skiping option
git-svn-id: https://wikiteam.googlecode.com/svn/trunk@171 31edc4fc-5e31-b4c4-d58b-c8bc928bcb95
2011-06-28 21:27:29 +00:00

95 lines
3.4 KiB
Python

# -*- coding: utf-8 -*-
# Copyright (C) 2011 WikiTeam
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import re
import sys
import os
import time
import urllib
f = urllib.urlopen('http://dumps.wikimedia.org/backup-index.html')
raw = f.read()
f.close()
m = re.compile(r'<a href="(?P<project>[^>]+)/(?P<date>\d+)">[^<]+</a>: <span class=\'done\'>Dump complete</span>').finditer(raw)
projects = []
for i in m:
projects.append([i.group('project'), i.group('date')])
projects.reverse() #oldest project dump, download first
#projects = [['enwiki', '20110405']]
start = ''
if len(sys.argv) == 2:
start = sys.argv[1].lower()
for project, date in projects:
if start:
if start != project:
print 'Skipping %s, %s' % (project, date)
continue
else:
start = '' #reset
time.sleep(1) #ctrl-c
f = urllib.urlopen('http://dumps.wikimedia.org/%s/%s/' % (project, date))
htmlproj = f.read()
#print htmlproj
f.close()
for dumpclass in ['pages-meta-history\d*\.xml\.7z']:
corrupted = True
maxretries = 3
while corrupted and maxretries > 0:
maxretries -= 1
m = re.compile(r'<a href="(?P<urldump>http://[^/>]+/%s/%s/%s-%s-%s)">' % (project, date, project, date, dumpclass)).finditer(htmlproj)
urldumps = []
for i in m:
urldumps.append(i.group('urldump')) #enwiki is splitted in several files
#print urldumps
for urldump in urldumps:
dumpfilename = urldump.split('/')[-1]
path = '%s/%s' % (dumpfilename[0], project)
if not os.path.exists(path):
os.makedirs(path)
os.system('wget -c %s -O %s/%s' % (urldump, path, dumpfilename))
#md5check
os.system('md5sum %s/%s > md5' % (path, dumpfilename))
f = open('md5', 'r')
raw = f.read()
f.close()
md51 = re.findall(r'(?P<md5>[a-f0-9]{32})\s+%s/%s' % (path, dumpfilename), raw)[0]
print md51
f = urllib.urlopen('http://dumps.wikimedia.org/%s/%s/%s-%s-md5sums.txt' % (project, date, project, date))
raw = f.read()
f.close()
f = open('%s/%s-%s-md5sums.txt' % (path, project, date), 'w')
f.write(raw)
f.close()
md52 = re.findall(r'(?P<md5>[a-f0-9]{32})\s+%s' % (dumpfilename), raw)[0]
print md52
if md51 == md52:
print 'md5sum is correct for this file, horay! \o/'
print '\n'*3
corrupted = False
else:
os.remove('%s/%s' % (path, dumpfilename))