从MediaWiki XML中提取页面标题和贡献者

时间:2012-12-31 17:44:59

标签: python xml-parsing python-2.7 mediawiki elementtree

我有一个非常大的(7GB)MediaWiki XML转储,它包含对Wiki每个页面进行的每次更改的记录。我正在尝试记录哪些用户为每个页面做出了贡献,因此我想从XML中提取它。

XML看起来像:

<mediawiki xmlns="http://www.mediawiki.org/xml/export-0.3/">
 <page>
  <title>Unique Page title</title>
  <id>11</id>
  <restrictions>sysop</restrictions>
  <revision>
    <id>11</id>
    <timestamp>2005-10-26T02:23:03Z</timestamp>
    <contributor>
      <ip>MediaWiki default</ip>
    </contributor>
    <text xml:space="preserve">i</text>
  </revision>
 </page>
 <page> ... </page>
 <page> ... </page>
 ...
</mediawiki>

对于这个大小的文件,我相信我需要使用iterparse。现在,我只是想打印标题,但是当我运行以下代码时,它会打印“无”。

NS = '{http://www.mediawiki.org/xml/export-0.3/}'
from xml.etree.ElementTree import iterparse
with open('XMLFile.xml') as f:
    for event, elem in iterparse(f):
        if elem.tag == NS + 'page':
            for node in elem:
                if node.tag == NS + 'title':
                    print node.text()
        elem.clear()

3 个答案:

答案 0 :(得分:3)

尝试在迭代解析过程中直接拉出'title'元素,而不是进行二次循环:

NS = '{http://www.mediawiki.org/xml/export-0.3/}'
from xml.etree.ElementTree import iterparse
with open('XMLFile.xml') as f:
    for event, elem in iterparse(f):
            if elem.tag == NS + 'title':
                print elem.text
            elem.clear()

似乎对我有用。

答案 1 :(得分:1)

我没有使用Python和iterparse的经验,但一般来说,使用迭代XML解析器执行此操作的方式如下:

  • 在解析循环之外,设置变量以存储当前页面标题和贡献者列表。
  • 在循环内部,只要打开page标记,就重置变量。
  • 遇到title标记时,请将页面标题变量设置为其内容。
  • 遇到contributor标记时,请将其内容添加到参与者列表中。
  • 关闭page标记后,输出收集的标题和贡献者列表。

答案 2 :(得分:1)

在打印None元素的文本内容时,您得到title,因为您过早地使用elem.clear()“。默认情况下,iterparse()仅生成“结束”事件。当page的“结束”事件被发出时,其所有子元素(包括title)都已被清除(清空)。

如果问题代码中的elem.clear()只向右移动一个缩进级别(四个空格),它将按预期工作。使代码有效的另一种方法是将iterparse(f)更改为iterparse(f, events=["start"])

node.text()应为node.text

有关iterparse()的详细信息,请参阅http://effbot.org/zone/element-iterparse.htm


假设XML转储(mw.xml)如下所示:

 
<mediawiki xmlns="http://www.mediawiki.org/xml/export-0.3/">
  <page>
    <title>Unique Page title 1</title>
    <id>11</id>
    <restrictions>sysop</restrictions>
    <revision>
      <id>11</id>
      <timestamp>2005-10-26T02:23:03Z</timestamp>
      <contributor>
       <username>Alice</username>
      </contributor>
      <text xml:space="preserve">i</text>
    </revision>
  </page>

  <page>
    <title>Unique Page title 2</title>
    <id>11</id>
    <restrictions>sysop</restrictions>
    <revision>
      <id>11</id>
      <timestamp>2005-10-26T02:23:03Z</timestamp>
      <contributor>
       <username>Bob</username>
      </contributor>
      <text xml:space="preserve">j</text>
    </revision>
  </page>
</mediawiki>

以下是关于如何获得头衔和贡献者的建议:

from xml.etree.ElementTree import iterparse

NS = '{http://www.mediawiki.org/xml/export-0.3/}'

with open('mw.xml') as f:
    for event, elem in iterparse(f):
        if elem.tag == '{0}page'.format(NS):
            title = elem.find("{0}title".format(NS))
            contr = elem.find(".//{0}username".format(NS))

            if title is not None:
                print title.text
            if contr is not None:
                print contr.text

            elem.clear()

输出:

Unique Page title 1 
Alice
Unique Page title 2 
Bob

我假设你想要贡献者的用户名。根据最新的XML Schemacontributor可以包含usernameip和/或id子元素(对于0.3版本的子元素也是如此)模式)。