willson-v-

from pywikibot.exceptions import PageSaveRelatedError
from pywikibot import textlib
import pywikibot
import sys
from pywikibot import pagegenerators
from pywikibot.bot import ExistingPageBot, SingleSiteBot

class LinkBot(
    SingleSiteBot,
    ExistingPageBot
):

    update_options = {
        'text': 'This is a test text',
        'summary': 'Bot: a bot test edit with Pywikbot.'
    }

    def treat_page(self):
        """Load the given page, do some changes, and save it."""
        # text = self.current_page.text
        # text += '\n' + self.opt.text
        # self.put_current(text, summary=self.opt.summary)
        text = self.current_page.text
        old_text = text
        catpl = pywikibot.Category(self.current_page.site, "包含旧站链接的页面")
        bad_link = []
        for link in self.current_page.extlinks():
            if "sstmlt" in link:
                bad_link.append(link)
        if bad_link:
            pywikibot.output(f"找到旧站链接：{bad_link}")
            current_cats = textlib.getCategoryLinks(text, self.current_page.site)
            pywikibot.output('该页面当前分类包括：')
            for cat in current_cats:
                pywikibot.output(f"\t* {cat.title()}")
            if catpl in current_cats:
                pywikibot.output(f'{self.current_page.title()}已在分类{catpl.title()}中，跳过。')
            elif self.current_page.namespace() == self.current_page.site.namespaces.TEMPLATE:
                pywikibot.output(f"这是一个模板页面, 跳过。")
            elif self.current_page.namespace() == self.current_page.site.namespaces.MODULE:
                pywikibot.output(f"这是一个模块页面, 跳过。")
            else:
                pywikibot.output(f"正在为{catpl.title(as_link=True)}增加分类...")
                current_cats.append(catpl)
                text = textlib.replaceCategoryLinks(text, current_cats, site=self.current_page.site)
                bad_link_str = f"{bad_link}"
                bad_link_str = bad_link_str if len(bad_link_str) < 100 else bad_link_str[:100]+"...（下略）]"
                summary=f"检测到该页面包含旧站链接{bad_link_str}，将其增加到分类:包含旧站链接的页面。"
                try:
                    self.userPut(
                        self.current_page, 
                        old_text,
                        text,
                        summary=summary
                    )
                except PageSaveRelatedError as error:
                    pywikibot.output(f'无法保存{self.current_page.title(as_link=True)}，因为{error}')
        else:
            current_cats = textlib.getCategoryLinks(text, self.current_page.site)
            if catpl in current_cats:
                pywikibot.output('该页面当前已不存在旧站链接。正在将其从分类中移除...')
                current_cats.remove(catpl)
                pywikibot.output('该页面当前分类包括：')
                for cat in current_cats:
                    pywikibot.output(f"\t* {cat.title()}")
                text = textlib.replaceCategoryLinks(text, current_cats, site=self.current_page.site)
                summary=f"检测到该页面不再包含旧站链接，将其从 分类:包含旧站链接的页面 中移除。"
                try:
                    self.userPut(
                        self.current_page, 
                        old_text,
                        text,
                        summary=summary
                    )
                except PageSaveRelatedError as error:
                    pywikibot.output(f'无法保存{self.current_page.title(as_link=True)}，因为{error}')
            else:
                pywikibot.output("一切正常。")


def main(*args: str) -> None:
    """Parse command line arguments and invoke bot."""
    options = {}
    gen_factory = pagegenerators.GeneratorFactory()
    # Option parsing
    local_args = pywikibot.handle_args(args)  # global options
    local_args = gen_factory.handle_args(local_args)  # generators options
    for arg in local_args:
        opt, _, value = arg.partition(':')
        if opt in ('-summary', '-text'):
            options[opt[1:]] = value
    combined_generator = generator=gen_factory.getCombinedGenerator()
    if not combined_generator:
        pywikibot.output("No page was specified. use -start to search all pages.")
        exit(0)
    else:
        LinkBot(generator=gen_factory.getCombinedGenerator(), **options).run()

if __name__ == '__main__':
    main()