Bikarhêner:Balyozxane/skrîpt/py/removexebat.py

Ji Wîkîpediya, ensîklopediya azad.
#!/usr/bin/env python3
"""
Şablona Xebat ji rûpelan radike eger di nav 30 rojan de xebatek nehatibe kirin.

Bikaranîn

python pwb.py removexebat -ns:0 -cat:"Kategorî:Gotarên berdewam dike"  -always

python pwb.py removexebat -ns:0 -cat:"Kategorî:Gotarên berdewam dike" -showdiff

The following parameters are supported:

-always           The bot won't ask for confirmation when putting a page.

-showdiff         The bot will show the differences in the console.

-async            Edits will be performed asynchronously.

Use global -simulate option for test purposes. No changes to live wiki
will be done.

"""
#
# (C) Balyozxane
#
# Distributed under the terms of the MIT license.
#

import pywikibot
from pywikibot import pagegenerators
from pywikibot.bot import (
    AutomaticTWSummaryBot,
    ConfigParserBot,
    ExistingPageBot,
    SingleSiteBot,
)
import mwparserfromhell
from kucosmetics import CANCEL, CosmeticChangesToolkit
from datetime import datetime

VERBOSE = True

# This is required for the text that is shown when you run this script
# with the parameter -help.
docuReplacements = {'&params;': pagegenerators.parameterHelp}  # noqa: N816


class XebatBot(
    # Refer pywikobot.bot for generic bot classes
    SingleSiteBot,  # A bot only working on one site
    ConfigParserBot,  # A bot which reads options from scripts.ini setting file
    # CurrentPageBot,  # Sets 'current_page'. Process it in treat_page method.
    #                  # Not needed here because we have subclasses
    ExistingPageBot,  # CurrentPageBot which only treats existing pages
    AutomaticTWSummaryBot,  # Automatically defines summary; needs summary_key
):
    use_redirects = False  # treats non-redirects only
    summary_key = 'basic-changing'

    update_options = {
        'async': False,
        'showdiff': False,
        'ignore': CANCEL.MATCH,
    }

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        # Retrieve redirects for templates
        self.xebat_redirects = [redirect.lower() for redirect in self.get_template_redirects("Xebat")]
        self.bot_name = "User:Balyozxane/skrîpt/py/removexebat.py"

    def get_template_redirects(self, template_title):
        template_title = "Şablon:" + template_title
        template_page = pywikibot.Page(self.site, template_title)
        redirects = template_page.backlinks(filter_redirects=True, namespaces=[10])
        redirect_titles = [redirect.title(with_ns=False) for redirect in redirects]
        redirect_titles.append(template_title.split(":")[-1])

        if VERBOSE:
            print(f"{template_title} redirects:\n{redirect_titles}")
        return redirect_titles

    @staticmethod
    def is_older_than_30_days(last_edit_timestamp):
        # Convert the timestamp string to a datetime object
        last_edit_datetime = datetime.strptime(str(last_edit_timestamp), "%Y-%m-%dT%H:%M:%SZ")

        # Calculate the current datetime
        current_datetime = datetime.utcnow()

        # Calculate the difference between current datetime and last edit datetime
        time_difference = current_datetime - last_edit_datetime

        # Check if the time difference is greater than 30 days
        if time_difference.days > 30:
            if VERBOSE:
                print("Last edit was more than 30 days ago")
            return True
        else:
            if VERBOSE:
                print("Last edit is not more than 30 days ago")
            return False

    def do_kozmetik(self, old_text):
        kozmetik_cebu = ""
        cc_toolkit = CosmeticChangesToolkit(self.current_page,
                                            ignore=self.opt.ignore)
        new_text, summaries = cc_toolkit.change(old_text)
        applied_summaries = ', '.join(summaries.values())
        if new_text is not False and new_text != old_text:
            kozmetik_cebu = "; paqijiyên kozmetîk"
            if applied_summaries:
                kozmetik_cebu += f' ({applied_summaries}.)'

        return new_text, kozmetik_cebu

    def remove_template(self, parsed):
        for template in parsed.filter_templates():
            template_name = template.name.strip().lower()  # Convert template name to lowercase
            if template_name in self.xebat_redirects:
                parsed.remove(template)

        return str(parsed)

    def treat_page(self) -> None:

        if self.current_page.namespace() != 0:
            if VERBOSE:
                print("Skipping Namespace not 0.")
            return

        last_edit_timestamp = self.current_page.latest_revision.timestamp

        if not self.is_older_than_30_days(last_edit_timestamp):
            if VERBOSE:
                print("Skipping page...")
            return

        if VERBOSE:
            print("Removing template...")

        text = self.current_page.text
        wikicode = mwparserfromhell.parse(text)

        updated_text = self.remove_template(wikicode)

        cleaned_new_text, kozmetik_cebu = self.do_kozmetik(updated_text)

        summary = f'[[{self.bot_name}|Bot]]: Şablona {{{{[[Şablon:Xebat|xebat]]}}}} hat rakirin{kozmetik_cebu}'

        self.put_current(
            cleaned_new_text,
            summary=summary,
            asynchronous=self.opt['async'],
            show_diff=self.opt['showdiff']
        )


def main(*args: str) -> None:
    """
    Process command line arguments and invoke bot.

    If args is an empty list, sys.argv is used.

    :param args: command line arguments
    """
    options = {}
    # Process global arguments to determine desired site
    local_args = pywikibot.handle_args(args)

    # This factory is responsible for processing command line arguments
    # that are also used by other scripts and that determine on which pages
    # to work on.
    gen_factory = pagegenerators.GeneratorFactory()

    # Process pagegenerators arguments
    local_args = gen_factory.handle_args(local_args)

    # Parse your own command line arguments
    for arg in local_args:
        arg, _, value = arg.partition(':')
        option = arg[1:]
        if option in ('-always', '-async', '-showdiff'):
            options[option[1:]] = True
        elif option == '-ignore':
            value = value.upper()
            try:
                options['ignore'] = getattr(CANCEL, value)
            except AttributeError:
                raise ValueError(f'Unknown ignore mode {value!r}!')
        # take the remaining options as booleans.
        # You will get a hint if they aren't pre-defined in your bot class
        else:
            options[option] = True
    # The preloading option is responsible for downloading multiple
    # pages from the wiki simultaneously.
    gen = gen_factory.getCombinedGenerator(preload=True)

    # check if further help is needed
    if not pywikibot.bot.suggest_help(missing_generator=not gen):
        # pass generator and private options to the bot
        bot = XebatBot(generator=gen, **options)
        bot.run()  # guess what it does


if __name__ == '__main__':
    main()