Kaynağa Gözat

Handle Wikicode contexts correctly.

tags/v0.1
Ben Kurtovic 12 yıl önce
ebeveyn
işleme
62b997d2fd
1 değiştirilmiş dosya ile 10 ekleme ve 8 silme
  1. +10
    -8
      mwparserfromhell/wikicode.py

+ 10
- 8
mwparserfromhell/wikicode.py Dosyayı Görüntüle

@@ -84,16 +84,18 @@ class Wikicode(StringMixIn):
return True
return False

def _do_search(self, obj, value, recursive, callback):
def _do_search(self, obj, value, recursive, callback, context=None):
if recursive:
for i, node in enumerate(self.nodes):
nodes = context.nodes if context else self.nodes
for i, node in enumerate(nodes):
if self._is_equivalent(obj, node):
return callback(value, i)
return callback(self, value, i)
if self._contains(self._get_children(node), obj):
return self._do_search(obj, value, recursive, callback)
return self._do_search(obj, value, recursive, callback,
context=obj)
raise ValueError(obj)

callback(value, self.index(obj, recursive=False))
callback(self, value, self.index(obj, recursive=False))

def _get_tree(self, code, lines, marker=None, indent=0):
def write(*args):
@@ -157,15 +159,15 @@ class Wikicode(StringMixIn):
self.nodes.insert(index, node)

def insert_before(self, obj, value, recursive=True):
callback = lambda value, i: self.insert(i, value)
callback = lambda self, value, i: self.insert(i, value)
self._do_search(obj, value, recursive, callback)

def insert_after(self, obj, value, recursive=True):
callback = lambda value, i: self.insert(i + 1, value)
callback = lambda self, value, i: self.insert(i + 1, value)
self._do_search(obj, value, recursive, callback)

def replace(self, obj, value, recursive=True):
def callback(value, i):
def callback(self, value, i):
self.nodes.pop(i)
self.insert(i, value)



Yükleniyor…
İptal
Kaydet