Fix wiki page rewriting.

This commit is contained in:
Sergey Vartanov 2021-12-11 13:45:09 +03:00
parent 67c9b095d0
commit 05b9d39e21

View file

@ -150,11 +150,32 @@ def generate_new_text(
lines: list[str] = old_text.split("\n")
# If rendering section already exists.
start: Optional[int] = None
end: int = -1
for index, line in enumerate(lines):
if HEADER_PATTERN.match(line):
if start is not None:
end = index
break
if RENDERING_HEADER_PATTERN.match(line):
start = index
if start is not None:
return (
"\n".join(lines[:end])
+ "\n=== [[Röntgen]] icons in [[Map Machine]] ===\n"
+ f"\n{wiki_text}\n"
+ "\n".join(lines[end:])
), icons
# If Röntgen rendering section already exists.
start: Optional[int] = None
end: int = -1
for index, line in enumerate(lines):
if HEADER_PATTERN.match(line):
if start is not None:
@ -166,15 +187,13 @@ def generate_new_text(
if start is not None:
return (
"\n".join(lines[: start + 2])
+ "\n"
+ wiki_text
+ "\n"
+ f"\n{wiki_text}\n"
+ "\n".join(lines[end:])
), icons
# Otherwise.
headers = [None, None]
headers: list[Optional[int]] = [None, None]
for index, line in enumerate(lines):
for i, pattern in enumerate(HEADER_PATTERNS):