You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

45 lines
1.5 KiB

from oyoyo.client import IRCClient
from oyoyo.cmdhandler import DefaultCommandHandler, protected
from oyoyo.parse import parse_nick
import logging
import botconfig
import wolfgame
class WolfBotHandler(DefaultCommandHandler):
def __init__(self, client):
super().__init__(client)
def privmsg(self, rawnick, chan, msg):
if chan != botconfig.NICK: #not a PM
for x in wolfgame.COMMANDS.keys():
if msg.startswith(x):
msg = msg.replace(x, "", 1)
wolfgame.COMMANDS[x](self.client, rawnick, chan, msg.lstrip())
else:
for x in wolfgame.PM_COMMANDS.keys():
if msg.startswith(x):
msg = msg.replace(x, "", 1)
wolfgame.PM_COMMANDS[x](self.client, rawnick, msg.lstrip())
@protected
def __unhandled__(self, cmd, *args):
if cmd in wolfgame.HOOKS.keys():
largs = list(args)
for i,arg in enumerate(largs):
if arg: largs[i] = arg.decode('ascii')
wolfgame.HOOKS[cmd](*largs)
else:
logging.debug('unhandled command %s(%s)' % (cmd, args))
def main():
logging.basicConfig(level=logging.DEBUG)
cli = IRCClient(WolfBotHandler, host=botconfig.HOST, port=botconfig.PORT, nickname=botconfig.NICK,
connect_cb=wolfgame.connect_callback)
conn = cli.connect()
while True:
next(conn)
if __name__ == "__main__":
main()