bot.py

#! /usr/bin/env python3
# -*- coding: utf-8 -*-

"""
Module for the common base class for all Bots
"""

import logging
import re

LOG = logging.getLogger("bot")

class Bot():
    """Base class for things common between different protocols"""
    def __init__(self):
        self.CONFIG = {}
        self.ACTIONS = []
        self.GENERAL_ACTIONS = []
        self.MSG_LOG = logging.getLogger("message")

    def getConfig(self):
        """Return the current configuration"""
        return self.CONFIG

    def setConfig(self, config):
        """Set the current configuration"""
        self.CONFIG = config

    def registerActions(self, actions):
        """Register actions to use"""
        LOG.info("Adding actions")
        for action in actions:
            LOG.info("Adding action: %s", action.__name__)
        self.ACTIONS.extend(actions)

    def registerGeneralActions(self, actions):
        """Register general actions to use"""
        LOG.info("Adding actions")
        for action in actions:
            LOG.info("Adding action: %s", action.__name__)
        self.GENERAL_ACTIONS.extend(actions)

    @staticmethod
    def tokenize(message):
        """Split a message into normalized tokens"""
        return re.sub("[,.?:]", " ", message).strip().lower().split()