Browse Source

added check for None node

pull/15/head
notconfusing 12 years ago
parent
commit
efc230e7e9
1 changed files with 3 additions and 2 deletions
  1. +3
    -2
      mwparserfromhell/wikicode.py

+ 3
- 2
mwparserfromhell/wikicode.py View File

@@ -56,8 +56,9 @@ class Wikicode(StringMixIn):
This is implemented by the ``__iternodes__()`` generator of ``Node`` This is implemented by the ``__iternodes__()`` generator of ``Node``
classes, which by default yields itself and nothing more. classes, which by default yields itself and nothing more.
""" """
for context, child in node.__iternodes__(self._get_all_nodes):
yield child
if node: #fixes bug on malformed wikitext from wikipedia dump
for context, child in node.__iternodes__(self._get_all_nodes):
yield child


def _get_context(self, node, obj): def _get_context(self, node, obj):
"""Return a ``Wikicode`` that contains *obj* in its descendants. """Return a ``Wikicode`` that contains *obj* in its descendants.


Loading…
Cancel
Save