def _replace_bs_template_files(self, tpl: mwparserfromhell.nodes.Template,
template: pywikibot.Page) -> None:
assert self.site_config is not None
for icon_prefix, templates in self.site_config.bs_templates.items():
if template not in templates:
continue
for param in tpl.params:
if param.name.matches("1"):
prefix = icon_prefix.strip()
else:
prefix = ""
param_value = HTML_COMMENT.sub("", str(param.value)).strip()
try:
current_icon = BSiconPage(self.current_page.site,
name=prefix + param_value)
current_icon.title()
except (pywikibot.exceptions.Error, ValueError):
continue
new_icon = self.opt.bsicons_map.get(current_icon, None)
if not new_icon:
continue
# The replacement must have the same prefix.
if new_icon.name[:len(prefix)] == prefix:
param.value = str(param.value).replace(
param_value,
new_icon.name[len(prefix):] # noqa: E203
)
self.current_page.replacements.add(
Replacement(current_icon, new_icon))
def _replace_rt_template_files(
self, tpl: mwparserfromhell.nodes.Template) -> None:
# Written for [[:cs:Template:Železniční trať]].
for param in tpl.params:
param_value = HTML_COMMENT.sub("", str(param.value)).strip()
if param.name.matches("typ"):
if param_value[:2] == "ex":
current_name = "exl" + param_value[2:]
else:
current_name = "l" + param_value
else:
current_name = param_value
try:
current_icon = BSiconPage(self.current_page.site,
name=current_name)
current_icon.title()
except (pywikibot.exceptions.Error, ValueError):
continue
new_icon = self.opt.bsicons_map.get(current_icon, None)
if not new_icon:
continue
if param.name.matches("typ"):
if new_icon.name[:3] == "exl":
replacement = "ex" + new_icon.name[3:]
elif new_icon.name[:1] == "l":
replacement = new_icon.name[1:]
else:
pywikibot.log(f"{new_icon} cannot be used in |typ=.")
continue
else:
replacement = new_icon.name
param.value = str(param.value).replace(param_value, replacement)
self.current_page.replacements.add(
Replacement(current_icon, new_icon))
def _replace_routemap_files(self,
tpl: mwparserfromhell.nodes.Template) -> None:
for param in tpl.params:
if not re.search(r"^(?:map\d*|\d+)$", str(param.name).strip()):
continue
param_value = str(param.value)
for match in ROUTEMAP_BSICON.findall(param_value):
current_name = HTML_COMMENT.sub("", match[1]).strip()
try:
current_icon = BSiconPage(self.current_page.site,
name=current_name)
current_icon.title()
except (pywikibot.exceptions.Error, ValueError):
continue
new_icon = self.opt.bsicons_map.get(current_icon, None)
if not new_icon:
continue
param_value = param_value.replace(
"".join(match),
match[0] + match[1].replace(current_name, new_icon.name) +
match[2],
)
self.current_page.replacements.add(
Replacement(current_icon, new_icon))
param.value = param_value
def replace_gallery_files(
self, wikicode: mwparserfromhell.wikicode.Wikicode) -> None:
"""
Replace files in <gallery>.
:param wikicode: Parsed wikitext
"""
for tag in wikicode.ifilter_tags():
if tag.tag.lower() != "gallery":
continue
lines = str(tag.contents).splitlines()
for i, line in enumerate(lines):
title, sep, caption = removeDisabledParts(line).partition("|")
if not title:
continue
try:
current_icon = BSiconPage(self.current_page.site, title)
current_icon.title()
except (pywikibot.exceptions.Error, ValueError):
continue
new_icon = self.opt.bsicons_map.get(current_icon, None)
if new_icon:
lines[i] = f"{new_icon.title()}{sep}{caption}"
self.current_page.replacements.add(
Replacement(current_icon, new_icon))
if self.current_page.replacements:
tag.contents = "\n".join(lines) + "\n"
def replace_file_links(self, text: str) -> str:
"""
Return text with file links replaced.
:param text: Article text
"""
assert self.site_config is not None
for match in self.site_config.file_regex.finditer(
removeDisabledParts(text)):
try:
current_icon = BSiconPage(self.current_page.site,
match.group("filename"))
current_icon.title()
except (pywikibot.exceptions.Error, ValueError):
continue
new_icon = self.opt.bsicons_map.get(current_icon, None)
if new_icon:
text = text.replace(match.group("filename"),
new_icon.title(with_ns=False))
self.current_page.replacements.add(
Replacement(current_icon, new_icon))
return text
def output_move_log(
*,
site: pywikibot.site.APISite,
start: datetime.datetime,
end: datetime.datetime,
prefix: str,
) -> None:
"""Write move logevents to a page."""
logtype = "move"
log_page = Page(
site,
f"{prefix}/{logtype}/{start.strftime('%Y-%m')}",
)
if start.date().isoformat() in get_headers(log_page.text):
return
log_page = get_page_from_size(log_page)
if start.date().isoformat() in get_headers(log_page.text):
return
if not log_page.exists():
log_page.save(
text=f"{{{{nobots}}}}{{{{{prefix}}}}}",
summary="Create log page",
)
log_text = ""
for logevent in site.logevents(
logtype=logtype,
namespace=6,
start=start,
end=end,
reverse=True,
):
page = logevent.page()
row_text = "\n|-\n| "
try:
page = BSiconPage(page)
except ValueError:
row_text += page.title(as_link=True, textlink=True)
else:
row_text += f"{{{{bsn|{page.name}}}}}"
row_text += " || "
target = logevent.target_page
try:
target = BSiconPage(target)
except ValueError:
row_text += target.title(as_link=True, textlink=True)
else:
row_text += f"{{{{bsq2|{target.name}}}}}"
if not (isinstance(page, BSiconPage)
or isinstance(target, BSiconPage)):
continue
log_text += row_text
_handle_hidden(logevent.data)
log_text += (" || {r[logid]} || {r[timestamp]} || {r[user]} || "
"<nowiki>{r[comment]}</nowiki>".format(r=logevent.data))
if log_text:
log_text = (
f'{{| class="wikitable sortable mw-collapsible mw-collapsed"'
f"\n! Page !! Target !! logid !! Date/time !! User !! Summary"
f"{log_text}\n|}}")
else:
log_text = ": None"
try:
log_page.save(
text=log_text,
section="new",
summary=start.date().isoformat(),
minor=False,
force=True,
)
except pywikibot.exceptions.Error:
pywikibot.exception(tb=True)