Переглянути джерело

Add support for Wikilink.args for `File:` links

pull/301/head
Fred Brennan 1 рік тому
джерело
коміт
849746be4b
1 змінених файлів з 41 додано та 6 видалено
  1. +41
    -6
      src/mwparserfromhell/nodes/wikilink.py

+ 41
- 6
src/mwparserfromhell/nodes/wikilink.py Переглянути файл

@@ -28,10 +28,17 @@ __all__ = ["Wikilink"]
class Wikilink(Node):
"""Represents an internal wikilink, like ``[[Foo|Bar]]``."""

def __init__(self, title, text=None):
def __init__(self, title, args=None):
super().__init__()
self.title = title
self.text = text
self.args = args
if args is not None:
if not '|' in args:
self.text = args
else:
self.text = parse_anything(args).nodes[len(self.args)-1:]
else:
self.text = None

def __str__(self):
if self.text is not None:
@@ -51,6 +58,10 @@ class Wikilink(Node):
def __showtree__(self, write, get, mark):
write("[[")
get(self.title)
if self.args is not None:
write(" | ")
mark()
get(self.args)
if self.text is not None:
write(" | ")
mark()
@@ -63,17 +74,41 @@ class Wikilink(Node):
return self._title

@property
def args(self):
"""The args (if any), as a :class:`.list` object."""
return self._args

@property
def text(self):
"""The text to display (if any), as a :class:`.Wikicode` object."""
return self._text

@args.setter
def args(self, value):
if arg := parse_anything(value):
self._args = [node for node in str(arg.nodes[0]).split('|')]
if len(self._args) > 0:
self._text = str(arg)[len('|'.join(str(a) for a in self._args)):]
self._args.pop()
if len(self._text) == 0:
self._text = None
if len(self._args) == 0:
self._args = None
elif not hasattr(arg, 'nodes'):
self._args = None
else:
self._args = None

@title.setter
def title(self, value):
self._title = parse_anything(value)
if value is not None:
self._title = parse_anything(value)
else:
self._title = None

@text.setter
def text(self, value):
if value is None:
self._text = None
else:
if value is not None:
self._text = parse_anything(value)
else:
self._text = None

Завантаження…
Відмінити
Зберегти