seperated the wolfbot file
This commit is contained in:
Jimmy Cao
parent
27a252e558
commit
d65e91aac0
7
decorators.py
Normal file
7
decorators.py
Normal file
@ -0,0 +1,7 @@
|
||||
def generate(fdict):
|
||||
def cmd(s):
|
||||
def dec(f):
|
||||
fdict[s] = f
|
||||
return f
|
||||
return dec
|
||||
return cmd
|
||||
142
wolfbot.py
142
wolfbot.py
@ -3,26 +3,14 @@ from oyoyo.cmdhandler import DefaultCommandHandler, protected
|
||||
from oyoyo.parse import parse_nick
|
||||
import logging
|
||||
import botconfig
|
||||
|
||||
def connect_callback(cli):
|
||||
cli.identify(botconfig.PASS)
|
||||
cli.join(botconfig.CHANNEL)
|
||||
cli.msg("ChanServ", "op "+botconfig.CHANNEL)
|
||||
cli.msg(botconfig.CHANNEL, "\u0002Wolfbot2 is here.\u0002")
|
||||
|
||||
G_PM_COMMAND = []
|
||||
G_COMMAND = []
|
||||
COMMANDS = {}
|
||||
PM_COMMANDS = {}
|
||||
|
||||
HOOKS = {}
|
||||
import wolfgame
|
||||
|
||||
def cmd(s, pm = False):
|
||||
def dec(f):
|
||||
if s is None and pm:
|
||||
G_PM_COMMAND = f
|
||||
G_PM_COMMAND.append(f)
|
||||
elif s is None and not pm:
|
||||
G_COMMAND = f
|
||||
G_COMMAND.append(f)
|
||||
elif pm:
|
||||
PM_COMMANDS[s] = f
|
||||
else:
|
||||
@ -42,139 +30,35 @@ class WolfBotHandler(DefaultCommandHandler):
|
||||
|
||||
def privmsg(self, rawnick, chan, msg):
|
||||
if chan != botconfig.NICK: #not a PM
|
||||
for x in COMMANDS.keys():
|
||||
for x in wolfgame.COMMANDS.keys():
|
||||
if msg.startswith(x):
|
||||
msg = msg.replace(x, "", 1)
|
||||
COMMANDS[x](self.client, rawnick, chan, msg.lstrip())
|
||||
wolfgame.COMMANDS[x](self.client, rawnick, chan, msg.lstrip())
|
||||
else:
|
||||
for x in PM_COMMANDS.keys():
|
||||
for x in wolfgame.PM_COMMANDS.keys():
|
||||
if msg.startswith(x):
|
||||
msg = msg.replace(x, "", 1)
|
||||
PM_COMMANDS[x](self.client, rawnick, msg.lstrip())
|
||||
wolfgame.PM_COMMANDS[x](self.client, rawnick, msg.lstrip())
|
||||
print(wolfgame.COMMANDS)
|
||||
|
||||
@protected
|
||||
def __unhandled__(self, cmd, *args):
|
||||
if cmd in HOOKS.keys():
|
||||
if cmd in wolfgame.HOOKS.keys():
|
||||
largs = list(args)
|
||||
for i,arg in enumerate(largs):
|
||||
if arg: largs[i] = arg.decode('ascii')
|
||||
HOOKS[cmd](*largs)
|
||||
wolfgame.HOOKS[cmd](*largs)
|
||||
else:
|
||||
logging.debug('unhandled command %s(%s)' % (cmd, args))
|
||||
|
||||
def main():
|
||||
logging.basicConfig(level=logging.DEBUG)
|
||||
cli = IRCClient(WolfBotHandler, host="irc.freenode.net", port=6667, nickname=botconfig.NICK,
|
||||
connect_cb=connect_callback)
|
||||
cli = IRCClient(WolfBotHandler, host=botconfig.HOST, port=botconfig.PORT, nickname=botconfig.NICK,
|
||||
connect_cb=wolfgame.connect_callback)
|
||||
|
||||
conn = cli.connect()
|
||||
while True:
|
||||
next(conn)
|
||||
|
||||
|
||||
# Game Logic Begins:
|
||||
|
||||
import vars
|
||||
|
||||
def reset_game():
|
||||
vars.GAME_STARTED = False
|
||||
vars.ROLES = {"person" : []}
|
||||
vars.PHASE = "none"
|
||||
|
||||
# Command Handlers:
|
||||
@cmd("!say", pm=True)
|
||||
def say(cli, rawnick, rest): # To be removed later
|
||||
cli.msg(botconfig.CHANNEL, "{0} says: {1}".format(parse_nick(rawnick)[0], rest))
|
||||
|
||||
|
||||
@cmd("!bye", pm=True)
|
||||
@cmd("!bye", pm=False)
|
||||
def forced_exit(cli, rawnick, *rest): # Admin Only
|
||||
if parse_nick(rawnick)[0] in botconfig.ADMINS:
|
||||
cli.quit("Forced quit from admin")
|
||||
raise SystemExit
|
||||
|
||||
@cmd("!exec", pm=False)
|
||||
def py(cli, rawnick, chan, rest):
|
||||
if parse_nick(rawnick)[0] in botconfig.ADMINS:
|
||||
exec(rest)
|
||||
|
||||
@cmd("!ping", pm=False)
|
||||
def pinger(cli, rawnick, chan, rest):
|
||||
vars.PINGING = True
|
||||
TO_PING = []
|
||||
|
||||
@hook("whoreply")
|
||||
def on_whoreply(server, dunno, chan, dunno1, dunno2, dunno3, user, status, dunno4):
|
||||
if not vars.PINGING: return
|
||||
if user in (botconfig.NICK, parse_nick(rawnick)[0]): return # Don't ping self.
|
||||
|
||||
if vars.PINGING and 'G' not in status and '+' not in status:
|
||||
# TODO: check if the user has AWAY'D himself
|
||||
TO_PING.append(user)
|
||||
|
||||
@hook("endofwho")
|
||||
def do_ping(*args):
|
||||
if not vars.PINGING: return
|
||||
|
||||
chan = args[2]
|
||||
cli.msg(chan, "PING! "+" ".join(TO_PING))
|
||||
vars.PINGING = False
|
||||
|
||||
HOOKS.pop("whoreply")
|
||||
HOOKS.pop("endofwho")
|
||||
|
||||
cli.send("WHO "+chan)
|
||||
|
||||
|
||||
@cmd("!join", pm=False)
|
||||
def join(cli, rawnick, chan, rest):
|
||||
if vars.PHASE != "none":
|
||||
return
|
||||
|
||||
vars.GAME_STARTED = True
|
||||
|
||||
nick = parse_nick(rawnick)[0]
|
||||
cli.msg(chan, '{0} has started a game of Werewolf. \
|
||||
Type "!join" to join. Type "!start" to start the game. \
|
||||
Type "!wait" to increase join wait time.'.format(nick))
|
||||
|
||||
vars.ROLES["person"].append(nick)
|
||||
vars.PHASE = "join"
|
||||
|
||||
|
||||
|
||||
@cmd("!stats", pm=False)
|
||||
def stats(cli, rawnick, chan, rest):
|
||||
if vars.PHASE == "none":
|
||||
return
|
||||
nick = parse_nick(rawnick)[0]
|
||||
pl = []
|
||||
for x in vars.ROLES.values(): pl.extend(x)
|
||||
if len(pl) > 1:
|
||||
cli.msg(chan, '{0}: {1} players: {2}'.format(nick,
|
||||
len(pl), ", ".join(pl)))
|
||||
else:
|
||||
cli.msg(chan, '{0}: 1 player: {1}'.format(nick, pl[0]))
|
||||
|
||||
msg = []
|
||||
for role in vars.ROLES.keys():
|
||||
num = len(vars.ROLES[role])
|
||||
if num > 1:
|
||||
msg.append("{0} {1}".format(num, plural(role)))
|
||||
else:
|
||||
msg.append("{0} {1}".format(num, role))
|
||||
if len(msg) > 2: # More than 2 roles to say
|
||||
msg[-1] = "and "+msg[-1]+"."
|
||||
msg[0] = "{0}: There are ".format(nick) + msg[0]
|
||||
cli.msg(chan, ", ".join(msg))
|
||||
elif len(msg) == 2: # 2 roles to say
|
||||
cli.msg(chan, "{0}: There are ".format(nick) + msg[0],
|
||||
"and", msg[1] + ".")
|
||||
elif len(msg) == 1:
|
||||
cli.msg(chan, "{0}: There is ".format(nick) + msg[0] + ".")
|
||||
|
||||
# Game Logic Ends
|
||||
next(conn)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
120
wolfgame.py
Normal file
120
wolfgame.py
Normal file
@ -0,0 +1,120 @@
|
||||
from oyoyo.parse import parse_nick
|
||||
import vars
|
||||
import botconfig
|
||||
import decorators
|
||||
|
||||
COMMANDS = {}
|
||||
PM_COMMANDS = {}
|
||||
HOOKS = {}
|
||||
|
||||
cmd = decorators.generate(COMMANDS)
|
||||
pmcmd = decorators.generate(PM_COMMANDS)
|
||||
hook = decorators.generate(HOOKS)
|
||||
|
||||
# Game Logic Begins:
|
||||
|
||||
def connect_callback(cli):
|
||||
cli.identify(botconfig.PASS)
|
||||
cli.join(botconfig.CHANNEL)
|
||||
cli.msg("ChanServ", "op "+botconfig.CHANNEL)
|
||||
cli.msg(botconfig.CHANNEL, "\u0002Wolfbot2 is here.\u0002")
|
||||
|
||||
def reset_game():
|
||||
vars.GAME_STARTED = False
|
||||
vars.ROLES = {"person" : []}
|
||||
vars.PHASE = "none"
|
||||
|
||||
# Command Handlers:
|
||||
@cmd("!say")
|
||||
def say(cli, rawnick, rest): # To be removed later
|
||||
cli.msg(botconfig.CHANNEL, "{0} says: {1}".format(parse_nick(rawnick)[0], rest))
|
||||
|
||||
|
||||
@pmcmd("!bye")
|
||||
@cmd("!bye")
|
||||
def forced_exit(cli, rawnick, *rest): # Admin Only
|
||||
if parse_nick(rawnick)[0] in botconfig.ADMINS:
|
||||
cli.quit("Forced quit from admin")
|
||||
raise SystemExit
|
||||
|
||||
@cmd("!exec")
|
||||
def py(cli, rawnick, chan, rest):
|
||||
if parse_nick(rawnick)[0] in botconfig.ADMINS:
|
||||
exec(rest)
|
||||
|
||||
@cmd("!ping")
|
||||
def pinger(cli, rawnick, chan, rest):
|
||||
vars.PINGING = True
|
||||
TO_PING = []
|
||||
|
||||
@hook("whoreply")
|
||||
def on_whoreply(server, dunno, chan, dunno1, dunno2, dunno3, user, status, dunno4):
|
||||
if not vars.PINGING: return
|
||||
if user in (botconfig.NICK, parse_nick(rawnick)[0]): return # Don't ping self.
|
||||
|
||||
if vars.PINGING and 'G' not in status and '+' not in status:
|
||||
# TODO: check if the user has AWAY'D himself
|
||||
TO_PING.append(user)
|
||||
|
||||
@hook("endofwho")
|
||||
def do_ping(*args):
|
||||
if not vars.PINGING: return
|
||||
|
||||
chan = args[2]
|
||||
cli.msg(chan, "PING! "+" ".join(TO_PING))
|
||||
vars.PINGING = False
|
||||
|
||||
HOOKS.pop("whoreply")
|
||||
HOOKS.pop("endofwho")
|
||||
|
||||
cli.send("WHO "+chan)
|
||||
|
||||
|
||||
@cmd("!join")
|
||||
def join(cli, rawnick, chan, rest):
|
||||
if vars.PHASE != "none":
|
||||
return
|
||||
|
||||
vars.GAME_STARTED = True
|
||||
|
||||
nick = parse_nick(rawnick)[0]
|
||||
cli.msg(chan, '{0} has started a game of Werewolf. \
|
||||
Type "!join" to join. Type "!start" to start the game. \
|
||||
Type "!wait" to increase join wait time.'.format(nick))
|
||||
|
||||
vars.ROLES["person"].append(nick)
|
||||
vars.PHASE = "join"
|
||||
|
||||
|
||||
|
||||
@cmd("!stats")
|
||||
def stats(cli, rawnick, chan, rest):
|
||||
if vars.PHASE == "none":
|
||||
return
|
||||
nick = parse_nick(rawnick)[0]
|
||||
pl = []
|
||||
for x in vars.ROLES.values(): pl.extend(x)
|
||||
if len(pl) > 1:
|
||||
cli.msg(chan, '{0}: {1} players: {2}'.format(nick,
|
||||
len(pl), ", ".join(pl)))
|
||||
else:
|
||||
cli.msg(chan, '{0}: 1 player: {1}'.format(nick, pl[0]))
|
||||
|
||||
msg = []
|
||||
for role in vars.ROLES.keys():
|
||||
num = len(vars.ROLES[role])
|
||||
if num > 1:
|
||||
msg.append("{0} {1}".format(num, plural(role)))
|
||||
else:
|
||||
msg.append("{0} {1}".format(num, role))
|
||||
if len(msg) > 2: # More than 2 roles to say
|
||||
msg[-1] = "and "+msg[-1]+"."
|
||||
msg[0] = "{0}: There are ".format(nick) + msg[0]
|
||||
cli.msg(chan, ", ".join(msg))
|
||||
elif len(msg) == 2: # 2 roles to say
|
||||
cli.msg(chan, "{0}: There are ".format(nick) + msg[0],
|
||||
"and", msg[1] + ".")
|
||||
elif len(msg) == 1:
|
||||
cli.msg(chan, "{0}: There is ".format(nick) + msg[0] + ".")
|
||||
|
||||
# Game Logic Ends
|
||||
Loading…
x
Reference in New Issue
Block a user