Skip to main content

XMPP bots for humans

Project description

xbotlib

PyPI version Build Status

XMPP bots for humans

status: experimental

A friendly lightweight wrapper around slixmpp for writing XMPP bots in Python. The goal is to make writing and running XMPP bots easy and fun. xbotlib is a single file implementation which can easily be understood and extended. The xbotlib source code and ideas are largely borrowed/stolen/adapted/reimagined from the XMPP bot experiments that have gone on and are still going on in Varia.

We're lurking in xbotlibtest@muc.vvvvvvaria.org if you want to chat or just invite your bots for testing.

Install

$ pip install xbotlib

Example

Put the following in a echo.py file. This bot is pretty simple: it echoes back whatever message you send it. It is an easy way to get started.

from xbotlib import Bot

class EchoBot(Bot):

    def direct(self, message):
        self.reply(message.text, to=message.sender)

    def group(self, message):
        self.reply(message.content, room=message.room)

And then python echo.py. You will be asked a few questions in order to load the account details that your bot will be using. This will generate an echobot.conf file in the same working directory for further use. See the configuration section for more.

Read more in the API reference for how to write your own bots.

See more examples on git.vvvvvvaria.org.

API Reference

When writing your own bot, you always sub-class the Bot class provided from xbotlib. Then if you want to respond to a direct message, you write a direct function. If you want to respond to a group chat message, you write a group function. That's it for the basics.

Bot.direct(message)

Respond to direct messages.

Arguments:

  • message: received message (see SimpleMessage below for available attributes)

Bot.group(message)

Respond to a message in a group chat.

Arguments:

  • message: received message (see SimpleMessage below for available attributes)

Bot.serve(request)

Serve requests via the built-in web server.

Arguments:

  • request: the web request

SimpleMessage

A simple message interface.

Attributes:

  • text: the entire text of the message
  • content: the text of the message after the nick
  • sender: the user the message came from
  • room: the room the message came from
  • receiver: the receiver of the message
  • nick: the nickname of the sender
  • type: the type of message
  • url: The URL of a sent file

Bot

Bot.reply(message, to=None, room=None)

Send a reply back.

Arguments:

  • message: the message that is sent
  • to: the user to send the reply to
  • room: the room to send the reply to

Bot.respond(response, content_type="text/html")

Return a response via the web server.

Arguments:

  • response: the text of the response
  • content_type: the type of response

Other useful attributes on the Bot class are:

Working with your bot

Documentation

Add a help = "my help" to your Bot class like so.

class MyBot(Bot):
    help = "My help"

See more in the commands section on how to use this.

Commands

Using @<command> in direct messages and <nick>, @<command> (the , is optional, anything will be accepted here and there doesn't seem to be a consensus on what is most common way to "at" another user in XMPP) in group chats, here are the supported commands.

  • @uptime: how long the bot has been running
  • @help: the help text for what the bot does

There are also more general status commands which all bots respond to.

  • @bots: status check on who is a bot in the group chat

These commands will be detected in any part of the message sent to the bot. So you can write echobot, can we see your @uptime, or I'd love to know which @bots are here.

Avatars

By default, xbotlib will look for an avatar.png (so far tested with .png but other file types may work) file alongside your Python script which contains your bot implementation. You can also specify another path using the --avatar option on the command-line interface. The images should ideally have a height of 64 and a width of 64 pixels each.

Configuration

All the ways you can pass configuration details to your bot. There are three ways to configure your bot, the configuration file, command-line interface and the environment. Use whichever one suits you best. The values are loaded in the following order: command-line > configuration file > environment. This means you can override everything from the command-line easily.

Using the .conf configuration file

If you run simply run your Python script which contains the bot then xbotlib will generate a configuration for you by asking a few questions. This is the simplest way to run your bot locally.

  • account: the account of the bot
  • password: the password of the bot account
  • nick: the nickname of the bot
  • avatar: the avatar of the bot (default: avatar.png)
  • redis_url: the Redis connection URL
  • rooms: a list of rooms to automatically join
  • no_auto_join: disable auto-join when invited (default: False)
  • template: the port to serve from (default: index.html.j2)
  • serve: turn on the web server (default: False)
  • port: the port to serve from (default: 8080)
  • storage: storage back-end (default: file)
  • storage_file: path to file based storage back-end (default: <nick>.json>)

Using the command-line interface

Every bot accepts a number of comand-line arguments to load configuration. You can use the --help option to see what is available (e.g. python bot.py --help).

  • -h, --help: show this help message and exit
  • -d, --debug: enable verbose debug logs
  • -a ACCOUNT, --account ACCOUNT: account for the bot account
  • -p PASSWORD, --password PASSWORD: password for the bot account
  • -n NICK, --nick NICK: nickname for the bot account
  • -av AVATAR, --avatar AVATAR: avatar for the bot account (default: avatar.png)
  • -ru REDIS_URL, --redis-url REDIS_URL: redis storage connection URL
  • -r ROOMS [ROOMS ...], --rooms ROOMS [ROOMS ...]: Rooms to automatically join
  • -naj, --no-auto-join: disable automatically joining rooms when invited (default: False)
  • -pt PORT, --port PORT: the port to serve from (default: 8080)
  • -t TEMPLATE, --template TEMPLATE: the template to render (default: index.html.j2)
  • -s, --serve: turn on the web server (default: False)
  • -st {file,redis}, --storage {file,redis}: choice of storage back-end (default: file)
  • -stf STORAGE_FILE, --storage-file STORAGE_FILE: path to file based storage back-end (default: <nick>.json)

Using the environment

xbotlib will try to read the following configuration values from the environment if it cannot read them from a configuration file or the command-line interface. This can be useful when doing remote server deployments.

  • XBOT_ACCOUNT: The bot account
  • XBOT_PASSWORD: The bot password
  • XBOT_NICK: The bot nickname
  • XBOT_AVATAR: The bot avatar icon (default: avatar.png)
  • XBOT_REDIS_URL: Redis key store connection URL
  • XBOT_ROOMS: The rooms to automatically join
  • XBOT_NO_AUTO_JOIN: Disable auto-joining on invite (default: False)
  • XBOT_TEMPLATE: the template to render (default: index.html.j2)
  • XBOT_SERVE: Turn on the web server (default: False)
  • XBOT_PORT: The port to serve from (default: 8080)
  • XBOT_STORAGE: choice of storage back-end (default: file)
  • XBOT_STORAGE_FILE: path to file based storage back-end (default: <nick>.json)

Storage back-end

In order to store data you can make use of the self.db attribute of the Bot class. It is a Python dictionary which will be saved to disk automatically for you as a <nick>.json in your current working directory. The name and path to this file can be configured.

def group(self, message):
    if not message.room in self.db.keys():
        self.db[message.room] = "visited"

If you want to inspect the database when the bot is not running, you can look in the file directly.

$ cat mybot.json

For more advanced use cases, xbotlib also supports Redis as a storage back-end. You'll need to configure this (e.g. --storage redis) as the default uses the filesystem approach mentioned above. The same self.db will then be passed as a Redis connection object. You will also need to install additional dependencies using pip install xbotlib[redis].

Loading Plugins

You can specify a plugins = [...] on your bot definition and they will be automatically loaded when you start your bot.

class MyBot(Bot):
    plugins = ["xep_0066"]

See here for the list of supported plugins.

Serving HTTP

Firstly, you'll need to install additional dependencies.

$ pip install xbotlib[web]

Your bot will run a web server if you configure it to do so. Use the --serve option on the command-line, the serve = True configuration option or the XBOT_SERVE=True environment variable.

If you're running your bot locally, just visit 0.0.0.0:8080 to see. The default response is just some placeholder text. You can write your own responses using the Bot.serve function.

xbotlib provides a small wrapper API for Jinja2 which allows you to easily template and generate HTML. The web server is provided by aiohttp.

The default template search path is index.html.j2 in the current working directory. This can be configured through the usual configuration entrypoints.

Here's a small example that renders a random ASCII letter.

index.html.j2

<h1>{{ letter }}</h1>

bot.py

from string import ascii_letters

def serve(self, request):
    letter = choice(ascii_letters)
    rendered = self.template.render(letter=letter)
    return self.respond(rendered)

Please note the use of the return keyword here. The serve function must return a response that will be passed to the web server. This function can return any content type that you might find on the web (e.g. HTML, XML, JSON) but you must specify the content_type=... keyword argument for respond.

If you want to pass data from your direct/group functions to the serve function, you'll need to make use of some type of persistent storage. Your serve function can read from the storage back-end and then respond.

Deploy your bots

See bots.varia.zone.

Roadmap

See the issue tracker.

Changes

See the CHANGELOG.md.

License

See the LICENSE.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

xbotlib-0.15.0.tar.gz (28.0 kB view hashes)

Uploaded Source

Built Distribution

xbotlib-0.15.0-py3-none-any.whl (23.1 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page