Using pywikibot as library#

Pywikibot provides bot classes to develop your own script easily.

Minimal example#

Here is a minimal example script which shows their usage:

 1import pywikibot
 2from pywikibot import pagegenerators
 3from pywikibot.bot import ExistingPageBot
 4
 5class MyBot(ExistingPageBot):
 6
 7    update_options = {
 8        'text': 'This is a test text',
 9        'summary': 'Bot: a bot test edit with Pywikibot.',
10    }
11
12    def treat_page(self):
13        """Load the given page, do some changes, and save it."""
14        text = self.current_page.text
15        text += '\n' + self.opt.text
16        self.put_current(text, summary=self.opt.summary)
17
18def main():
19    """Parse command line arguments and invoke bot."""
20    options = {}
21    gen_factory = pagegenerators.GeneratorFactory()
22    # Option parsing
23    local_args = pywikibot.handle_args()  # global options
24    local_args = gen_factory.handle_args(local_args)  # generators options
25    for arg in local_args:
26        opt, sep, value = arg.partition(':')
27        if opt in ('-summary', '-text'):
28            options[opt[1:]] = value
29    MyBot(generator=gen_factory.getCombinedGenerator(), **options).run()
30
31if __name__ == '__main__':
32    main()

The script can be invoked from commandline like:

python mybot -site:wikipedia:test -page:Sandbox -text:"A text added to the sandbox"

Explanation#

1-3:

Import necessary framework code parts: pywikibot, pywikibot.pagegenerators, bot.ExistingPageBot.

5:

The bot is derived from ExistingPageBot. All pages from generator which does not exists are skipped.

7:

Every Bot has an always option which autoconfirms any changes if set to True. To expand all available options of a bot and set the default values of them, use update_options attribute or update available_options like it is shown in BasicBot below.

12:

All changes for each page are made in this treat_page method.

14:

current_page is the current pywikibot.Page object from generator.

15:

All bot options which are passed to the bot class when instantiating it are accessable via opt attribute. opt.always, opt.text and opt.summary are all available options for this bot class.

16:

Save the changes to the live wiki with put_current.

18:

Parse command line options inside this function.

20:

A dict which holds all options for the bot.

21:

pagegenerators.GeneratorFactory supports generators and filter options.

23:

Pywikibot provides global options like site specification or a simulate switch to prevent live wiki changes. These options are parsed by pywikibot.handle_args().

24:

Generators and filter options of pywikibot.pagegenerators are parsed here using GeneratorFactory.handle_args().

25:

Local options which are are available for the current bot are parsed in this loop.

29:

Create the bot passing keyword only parameters and run it.

Basic script#

scripts.basic is a more advanced sample script and shipped with the scripts folder. Here is the content:

#!/usr/bin/env python3
"""
An incomplete sample script.

This is not a complete bot; rather, it is a template from which simple
bots can be made. You can rename it to mybot.py, then edit it in
whatever way you want.

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


The following parameters are supported:

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

-text:            Use this text to be added; otherwise 'Test' is used

-replace:         Don't add text but replace it

-top              Place additional text on top of the page

-summary:         Set the action summary message for the edit.

This sample script is a
:py:obj:`ConfigParserBot <bot.ConfigParserBot>`. All settings can be
made either by giving option with the command line or with a settings file
which is scripts.ini by default. If you don't want the default values you can
add any option you want to change to that settings file below the [basic]
section like:

    [basic] ; inline comments starts with colon
    # This is a commend line. Assignments may be done with '=' or ':'
    text: A text with line break and
        continuing on next line to be put
    replace: yes ; yes/no, on/off, true/false and 1/0 is also valid
    summary = Bot: My first test edit with pywikibot

Every script has its own section with the script name as header.

In addition the following generators and filters are supported but
cannot be set by settings file:

&params;
"""
#
# (C) Pywikibot team, 2006-2022
#
# Distributed under the terms of the MIT license.
#
from __future__ import annotations

import pywikibot
from pywikibot import pagegenerators
from pywikibot.bot import (
    AutomaticTWSummaryBot,
    ConfigParserBot,
    ExistingPageBot,
    SingleSiteBot,
)


# 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 BasicBot(
    # 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
):

    """
    An incomplete sample bot.

    :ivar summary_key: Edit summary message key. The message that should be
        used is placed on /i18n subdirectory. The file containing these
        messages should have the same name as the caller script (i.e. basic.py
        in this case). Use summary_key to set a default edit summary message.

    :type summary_key: str
    """

    use_redirects = False  # treats non-redirects only
    summary_key = 'basic-changing'

    update_options = {
        'replace': False,  # delete old text and write the new text
        'summary': None,  # your own bot summary
        'text': 'Test',  # add this text from option. 'Test' is default
        'top': False,  # append text on top of the page
    }

    def treat_page(self) -> None:
        """Load the given page, do some changes, and save it."""
        text = self.current_page.text

        ################################################################
        # NOTE: Here you can modify the text in whatever way you want. #
        ################################################################

        # If you find out that you do not want to edit this page, just return.
        # Example: This puts Text on a page.

        # Retrieve your private option
        # Use your own text or use the default 'Test'
        text_to_add = self.opt.text

        if self.opt.replace:
            # replace the page text
            text = text_to_add

        elif self.opt.top:
            # put text on top
            text = text_to_add + text

        else:
            # put text on bottom
            text += text_to_add

        # if summary option is None, it takes the default i18n summary from
        # i18n subdirectory with summary_key as summary key.
        self.put_current(text, summary=self.opt.summary)


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 ('summary', 'text'):
            if not value:
                pywikibot.input('Please enter a value for ' + arg)
            options[option] = value
        # 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 = BasicBot(generator=gen, **options)
        bot.run()  # guess what it does


if __name__ == '__main__':
    main()

Note

Also see the documentation at Create your own script