Source code for kern.bots

# kern/bots/__init__.py
#
#

""" bots unit. """

__copyright__ = "Copyright 2015, B.H.J Thate"

## IMPORTS

from kern.utils.parse import txt_parse
from kern.utils.time import short_date
from kern.defines import BLA, ENDC
from kern.dispatcher import Dispatcher
from kern import kernel, fleet
from kern.thing import Thing

from kern.errors import RemoteDisconnect, NotImplemented

import logging
import socket
import time

## BOTS

class Bot(Dispatcher):

    """ Basic Bot. """

    default = ""
    cc = ""

    def __init__(zelf, *args, **kwargs):
        Dispatcher.__init__(zelf, *args, **kwargs)
        zelf._start = time.time()
        zelf.channels = ["#kern", ]

    def say(zelf, *args, **kwargs): print(args[1])
        
    ## CHANNELS

    def announce(zelf, *args, **kwargs):
        """ announce on channels. """
        zelf._last = time.time()
        logging.info("# announce %s" % " ".join(zelf.channels))
        for channel in zelf.channels: zelf.say(channel, args[0])

    def event(zelf, *args, **kwargs):
        event = Thing()
        event._target = zelf
        event.origin = "root@zelf"
        event.txt = args[0]
        return event

    def join(zelf, *args, **kwargs): pass
        
    def joining(zelf, *args, **kwargs):
        """ join channels.  """ 
        for channel in zelf.channels: zelf.join(channel)
            
## TESTBOT

class TestBot(Bot):

    """ USE WITH Bot.put() and Bot.read(). """

    testing = True
    cc = "!"
    default = ""

    def say(zelf, *args, **kwargs): logging.warn(args[1])

    def event(zelf, *args, **kwargs):
        o = Thing()
        o.origin = "test@kern"
        o._target = zelf
        o.loglevel = "warning"
        o.txt = args[0]
        o.parsed = txt_parse(zelf, o.txt)
        return o

## CMNDS

def join(event):
    if not event.parsed.args: return
    channel = event.parsed.args[0]
    for bot in fleet:
        if "join" not in bot: continue
        bot.join(channel)
        event.reply("join %s (%s)" % (channel, bot.server))

kernel.register("bot.join", join)

def part(event):
    if not event.parsed.args: return
    for bot in fleet:
        try: bot.part(event.parsed.args[0])
        except: continue
        event.reply("part %s (%s)" % (channel, bot.server))

kernel.register("bot.part", part)

def nick(event):
    if not event.parsed.args: return
    for bot in fleet:
        try: bot.nick(event.parsed.args[0])
        except: continue
        event.reply("nick %s" % bot.server)

kernel.register("bot.nick", nick)

def quit(event):
    if not event.parsed.args: return
    for bot in fleet:
        try: bot.quit(event.parsed.rest)
        except: continue
        event.reply("quit %s" % bot.server)
    
kernel.register("bot.quit", quit)