45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
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):
|
|
h = msg.replace(x, "", 1).lstrip()
|
|
wolfgame.COMMANDS[x](self.client, rawnick, chan, h)
|
|
else:
|
|
for x in wolfgame.PM_COMMANDS.keys():
|
|
if msg.startswith(x):
|
|
h = msg.replace(x, "", 1).lstrip()
|
|
wolfgame.PM_COMMANDS[x](self.client, rawnick, h)
|
|
|
|
@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](self.client, *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() |