blob: 5d6afe9e28d2216f59e3058d5d08094ce26edbc6 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
#!/usr/bin/python3
import sys, re
import bz2
import sqlite3
usageInfo = f"usage: {sys.argv[0]} title1\n"
usageInfo += "Looks up a page with title title1 in a wikipedia dump,\n"
usageInfo += "using a dump index db, and prints the corresponding <page>.\n"
if len(sys.argv) != 2:
print(usageInfo, file=sys.stderr)
sys.exit(1)
dumpFile = "enwiki-20220501-pages-articles-multistream.xml.bz2"
indexDb = "dumpIndex.db"
pageTitle = sys.argv[1]
# Searching index file
print("Lookup offset in index db")
dbCon = sqlite3.connect(indexDb)
dbCur = dbCon.cursor()
row = dbCur.execute("SELECT title, offset, next_offset FROM offsets WHERE title = ?",
(pageTitle.replace("_", " "),)).fetchone()
if row == None:
print("Title not found")
sys.exit(0)
(_, pageOffset, endOffset) = row
dbCon.close()
print("Found chunk at offset {}".format(pageOffset))
# Read dump file
print("Reading dump file")
content = []
with open(dumpFile, mode='rb') as file:
# Get uncompressed chunk
file.seek(pageOffset)
compressedData = file.read(None if endOffset == -1 else endOffset - pageOffset)
data = bz2.BZ2Decompressor().decompress(compressedData).decode()
# Look in chunk for page
lines = data.splitlines()
lineIdx = 0
found = False
pageNum = 0
while not found:
line = lines[lineIdx]
if line.lstrip() == "<page>":
pageNum += 1
if pageNum > 100:
print("ERROR: Did not find title after 100 pages")
break
lineIdx += 1
titleLine = lines[lineIdx]
if titleLine.lstrip() == '<title>' + pageTitle + '</title>':
found = True
print("Found title in chunk as page {}".format(pageNum))
content.append(line)
content.append(titleLine)
while True:
lineIdx += 1
line = lines[lineIdx]
content.append(line)
if line.lstrip() == "</page>":
break
lineIdx += 1
# Print content
print("Content: ")
print("\n".join(content))
|
