Explorar el Código

added check for None node

pull/15/head
notconfusing hace 12 años
padre
commit
efc230e7e9
Se han modificado 1 ficheros con 3 adiciones y 2 borrados
  1. +3
    -2
      mwparserfromhell/wikicode.py

+ 3
- 2
mwparserfromhell/wikicode.py Ver fichero

@@ -56,8 +56,9 @@ class Wikicode(StringMixIn):
This is implemented by the ``__iternodes__()`` generator of ``Node``
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):
"""Return a ``Wikicode`` that contains *obj* in its descendants.


Cargando…
Cancelar
Guardar