wrote the day transition,
added !leave, !quit, and QUIT/PART handlers
This commit is contained in:
Jimmy Cao
parent
d75ddd641c
commit
2affdfa555
22
vars.py
22
vars.py
@ -1,8 +1,13 @@
|
||||
PING_WAIT = 300 # Seconds
|
||||
MINIMUM_WAIT = 60
|
||||
MINIMUM_WAIT = 0 # debug, change to 60 for normal
|
||||
EXTRA_WAIT = 20
|
||||
MAXIMUM_WAITED = 2 # limit for amount of !wait's
|
||||
MAX_SHOTS = 2
|
||||
NIGHT_TIME_LIMIT = 90
|
||||
|
||||
NO_VICTIMS_MESSAGES = ("The body of a young penguin pet is found",
|
||||
"A pool of blood and wolf paw prints are found.",
|
||||
"Traces of wolf fur are found")
|
||||
|
||||
# These change ingame
|
||||
ROLES = {"person" : []}
|
||||
@ -10,10 +15,13 @@ ORIGINAL_ROLES = None
|
||||
PHASE = "none" # "join", "day", or "night"
|
||||
LAST_PING = 0
|
||||
CURSED = "" # nickname of cursed villager
|
||||
GAME_START_TIME = 0
|
||||
CAN_START_TIME = 0
|
||||
WAITED = 0
|
||||
GUNNERS = {}
|
||||
VICTIM = "" # nickname of to-be-killed villager
|
||||
SEEN = [] # list of seers that have had visions
|
||||
DEAD = [] # list of people who are dead
|
||||
TRAITOR = ""
|
||||
TIMERS = [None, None] # nightlimit, daylimit
|
||||
|
||||
is_role = lambda plyr, rol: rol in ROLES and plyr in ROLES[rol]
|
||||
|
||||
@ -33,4 +41,10 @@ def list_players_and_roles():
|
||||
for x in ROLES.keys():
|
||||
for p in ROLES[x]:
|
||||
plr[p] = x
|
||||
return plr
|
||||
return plr
|
||||
|
||||
get_role = lambda plyr: list_players_and_roles()[plyr]
|
||||
|
||||
def del_player(pname):
|
||||
prole = get_role(pname)
|
||||
ROLES[prole].remove(pname)
|
||||
272
wolfgame.py
272
wolfgame.py
@ -2,7 +2,8 @@ from oyoyo.parse import parse_nick
|
||||
import vars
|
||||
import botconfig
|
||||
import decorators
|
||||
import time
|
||||
from datetime import datetime, timedelta
|
||||
import threading
|
||||
import random
|
||||
|
||||
COMMANDS = {}
|
||||
@ -33,28 +34,29 @@ def say(cli, nick, rest): # To be removed later
|
||||
@pmcmd("!bye", admin_only=True)
|
||||
@cmd("!bye", admin_only=True)
|
||||
def forced_exit(cli, nick, *rest): # Admin Only
|
||||
reset_game(cli, nick, rest[0], None)
|
||||
reset_game(cli, nick, botconfig.CHANNEL, None)
|
||||
cli.quit("Forced quit from admin")
|
||||
raise SystemExit
|
||||
|
||||
@cmd("!exec")
|
||||
@cmd("!exec", admin_only=True)
|
||||
def py(cli, nick, chan, rest):
|
||||
if nick in botconfig.ADMINS:
|
||||
exec(rest)
|
||||
exec(rest)
|
||||
|
||||
@cmd("!ping")
|
||||
def pinger(cli, nick, chan, rest):
|
||||
if vars.LAST_PING + 300 > time.time():
|
||||
cli.notice(nick, "This command is ratelimited. \
|
||||
Please wait a while before using it again.")
|
||||
if (vars.LAST_PING and
|
||||
vars.LAST_PING + timedelta(seconds=300) > datetime.now()):
|
||||
cli.notice(nick, ("This command is ratelimited. " +
|
||||
"Please wait a while before using it again."))
|
||||
return
|
||||
|
||||
vars.LAST_PING = time.time()
|
||||
vars.LAST_PING = datetime.now()
|
||||
vars.PINGING = True
|
||||
TO_PING = []
|
||||
|
||||
@hook("whoreply")
|
||||
def on_whoreply(server, dunno, chan, dunno1, dunno2, dunno3, user, status, dunno4):
|
||||
def on_whoreply(server, dunno, chan, dunno1,
|
||||
dunno2, dunno3, user, status, dunno4):
|
||||
if not vars.PINGING: return
|
||||
if user in (botconfig.NICK, nick): return # Don't ping self.
|
||||
|
||||
@ -81,7 +83,7 @@ def join(cli, nick, chan, rest):
|
||||
cli.mode(chan, "+v", nick, nick+"!*@*")
|
||||
vars.ROLES["person"].append(nick)
|
||||
vars.PHASE = "join"
|
||||
vars.CAN_START_TIME = time.time() + vars.MINIMUM_WAIT
|
||||
vars.CAN_START_TIME = datetime.now() + timedelta(seconds=vars.MINIMUM_WAIT)
|
||||
cli.msg(chan, '\u0002{0}\u0002 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))
|
||||
@ -117,14 +119,206 @@ def stats(cli, nick, chan, rest):
|
||||
message.append("\u0002(0}\u0002 {1}".format(count, vars.plural(role)))
|
||||
else:
|
||||
message.append("\u0002{0}\u0002 {1}".format(count, role))
|
||||
if len(vars.ROLES["wolf"]) > 1:
|
||||
vb = "are"
|
||||
else:
|
||||
vb = "is"
|
||||
cli.msg(chan,
|
||||
"{0}: There are {1}, and {2}.".format(nick,
|
||||
"{0}: There {verb} {1}, and {2}.".format(nick,
|
||||
", ".join(message[0:-1]),
|
||||
message[-1]))
|
||||
message[-1]), verb=vb)
|
||||
|
||||
def transition_night(cli, chan):
|
||||
vars.PHASE = "night"
|
||||
def del_player(cli, nick, died_in_game = True):
|
||||
cli.mode(botconfig.CHANNEL, "-v", "{0} {0}!*@*".format(nick))
|
||||
if vars.PHASE != "join" and died_in_game:
|
||||
cli.mode(botconfig.CHANNEL, "+q", "{0} {0}!*@*".format(nick))
|
||||
vars.DEAD.append(nick)
|
||||
vars.del_player(nick)
|
||||
|
||||
def leave(cli, what, nick):
|
||||
if nick not in vars.list_players(): # not playing
|
||||
return
|
||||
msg = ""
|
||||
if what in ("!quit", "!leave"):
|
||||
msg = ("\u0002{0}\u0002 died of an unknown disease. "+
|
||||
"S/He was a \u0002{1}\u0002.")
|
||||
died_in_game = True
|
||||
elif what == "part":
|
||||
msg = ("\u0002{0}\u0002 died due to eating poisonous berries. "+
|
||||
"Appears (s)he was a \u0002{1}\u0002.")
|
||||
elif what == "quit":
|
||||
msg = ("\u0002{0}\u0002 died due to a fatal attack by wild animals. "+
|
||||
"Appears (s)he was a \u0002{1}\u0002.")
|
||||
elif what == "kick":
|
||||
msg = ("\u0002{0}\u0002 died due to falling off a cliff. "+
|
||||
"Appears (s)he was a \u0002{1}\u0002.")
|
||||
msg = msg.format(nick, vars.get_role(nick))
|
||||
cli.msg(botconfig.CHANNEL, msg)
|
||||
del_player(cli, nick, died_in_game)
|
||||
|
||||
cmd("!leave")(lambda cli, nick, chan, *rest: leave(cli, "!leave", nick))
|
||||
cmd("!quit")(lambda cli, nick, chan, *rest: leave(cli, "!quit", nick))
|
||||
hook("part")(lambda cli, nick, chan, *rest: leave(cli, "part", nick))
|
||||
hook("quit")(lambda cli, nick, chan, *rest: leave(cli, "quit", nick))
|
||||
|
||||
def transition_day(cli):
|
||||
chan = botconfig.CHANNEL
|
||||
|
||||
vars.PHASE = "day"
|
||||
vars.DAY_START_TIME = datetime.now()
|
||||
td = vars.DAY_START_TIME - vars.NIGHT_START_TIME
|
||||
vars.NIGHT_TIMEDELTA += td
|
||||
min, sec = td.seconds // 60, td.seconds % 60
|
||||
|
||||
message = "Night lasted \u0002{0:0>2}:{1:0>2}\u0002. It is now daytime. \
|
||||
The villagers awake, thankful for surviving the night, \
|
||||
and search the village... ".format(min, sec)
|
||||
if not vars.VICTIM:
|
||||
message += random.choice(vars.NO_VICTIMS_MESSAGES)
|
||||
cli.msg(chan, message);
|
||||
return
|
||||
# TODO: check if visited is harlot
|
||||
|
||||
dead = []
|
||||
|
||||
message += "The dead body of \u0002{0}\u0002, a \
|
||||
\u0002{1}\u0002, is found. Those remaining mourn his/her \
|
||||
death.".format(vars.VICTIM, vars.get_role(vars.VICTIM))
|
||||
dead.append(vars.VICTIM)
|
||||
# TODO: check if harlot also died
|
||||
cli.msg(chan, message)
|
||||
|
||||
for deadperson in dead:
|
||||
del_player(cli, deadperson, True)
|
||||
|
||||
|
||||
|
||||
|
||||
def chk_nightdone(cli):
|
||||
if (len(vars.SEEN) == len(vars.ROLES["seer"]) and
|
||||
vars.VICTIM and vars.PHASE == "night"):
|
||||
if vars.TIMERS[0]:
|
||||
vars.TIMERS[0].cancel() # cancel timer
|
||||
vars.TIMERS[0] = None
|
||||
transition_day(cli)
|
||||
|
||||
|
||||
@pmcmd("!kill")
|
||||
@pmcmd("kill")
|
||||
def kill(cli, nick, rest):
|
||||
if vars.PHASE == "none":
|
||||
cli.msg(nick, "No game is currently running.")
|
||||
return
|
||||
if not nick in vars.list_players():
|
||||
cli.msg(nick, "You're currently playing")
|
||||
return
|
||||
if not (vars.is_role(nick, "wolf") or vars.is_role(nick, "traitor")):
|
||||
cli.msg(nick, "Only a wolf may use this command")
|
||||
return
|
||||
if vars.PHASE != "night":
|
||||
cli.msg(nick, "You may only kill people at night.")
|
||||
return
|
||||
victim = rest.split(" ")[0].strip()
|
||||
if not victim:
|
||||
cli.msg(nick, "Not enough parameters")
|
||||
return
|
||||
if victim not in vars.list_players():
|
||||
cli.msg(nick,"\u0002{0}\u0002 is currently not playing.".format(victim))
|
||||
return
|
||||
if victim == nick:
|
||||
cli.msg(nick, "Suicide is bad. Don't do it.")
|
||||
return
|
||||
if victim in vars.ROLES["wolf"]:
|
||||
cli.msg(nick, "You may only kill villagers, not other wolves")
|
||||
return
|
||||
vars.VICTIM = victim
|
||||
cli.msg(nick, "You have selected \u0002{0}\u0002 to be killed".format(victim))
|
||||
chk_nightdone(cli)
|
||||
|
||||
@pmcmd("see")
|
||||
@pmcmd("!see")
|
||||
def see(cli, nick, rest):
|
||||
if vars.PHASE == "none":
|
||||
cli.msg(nick, "No game is currently running.")
|
||||
return
|
||||
if not nick in vars.list_players():
|
||||
cli.msg(nick, "You're currently playing")
|
||||
return
|
||||
if not vars.is_role(nick, "seer"):
|
||||
cli.msg(nick, "Only a seer may use this command")
|
||||
return
|
||||
if vars.PHASE != "night":
|
||||
cli.msg(nick, "You may have visions at night.")
|
||||
return
|
||||
if nick in vars.SEEN:
|
||||
cli.msg(nick, "You may only have one vision per round.")
|
||||
victim = rest.split(" ")[0].strip()
|
||||
if not victim:
|
||||
cli.msg(nick, "Not enough parameters")
|
||||
return
|
||||
if victim not in vars.list_players():
|
||||
cli.msg(nick,"\u0002{0}\u0002 is \
|
||||
currently not playing.".format(victim))
|
||||
return
|
||||
if vars.CURSED == nick:
|
||||
role = "wolf"
|
||||
elif vars.TRAITOR == nick:
|
||||
role = "villager"
|
||||
else:
|
||||
role = vars.get_role(victim)
|
||||
cli.msg(nick, "You have a vision; in this vision, \
|
||||
you see that \u0002{0}\u0002 is a \u0002{1}\u0002!".format(victim,
|
||||
role))
|
||||
vars.SEEN.append(nick)
|
||||
chk_nightdone(cli)
|
||||
|
||||
@pmcmd("")
|
||||
def relay(cli, nick, rest):
|
||||
badguys = vars.ROLES.get("wolf", []) + vars.ROLES.get("traitor", [])
|
||||
if len(badguys) > 1:
|
||||
if vars.is_role(nick, "wolf") or vars.is_role(nick, "traitor"):
|
||||
badguys.remove(nick) # remove self from list
|
||||
for badguy in badguys:
|
||||
cli.msg(badguy, "{0} says: {1}".format(nick, rest))
|
||||
|
||||
def transition_night(cli):
|
||||
vars.PHASE = "night"
|
||||
vars.VICTIM = "" # nickname of cursed villager
|
||||
vars.SEEN = [] # list of seers that have had visions
|
||||
vars.NIGHT_START_TIME = datetime.now()
|
||||
|
||||
chan = botconfig.CHANNEL
|
||||
cli.msg(chan, "It is now nighttime. All players \
|
||||
check for PMs from me for instructions. If you did not receive \
|
||||
one, simply sit back, relax, and wait patiently for morning.")
|
||||
|
||||
t = threading.Timer(vars.NIGHT_TIME_LIMIT, transition_day, [cli])
|
||||
vars.TIMERS[0] = t
|
||||
t.start()
|
||||
|
||||
# send PMs
|
||||
pl = vars.list_players()
|
||||
for wolf in vars.ROLES["wolf"]:
|
||||
cli.msg(wolf, 'You are a \u0002wolf\u0002. It is your job to kill all the \
|
||||
villagers. Use "kill <nick>" to kill a villager. Also, if \
|
||||
you send a PM to me, it will be relayed to all other wolves.')
|
||||
_pl = pl[:]
|
||||
_pl.remove(wolf) # remove self from list
|
||||
for i, player in enumerate(_pl):
|
||||
if vars.is_role(player, "wolf"):
|
||||
_pl[i] = player + " (wolf)"
|
||||
elif vars.is_role(player, "traitor"):
|
||||
_pl[i] = player + " (traitor)"
|
||||
cli.msg(wolf, "Players: "+", ".join(_pl))
|
||||
|
||||
for seer in vars.ROLES["seer"]:
|
||||
_pl = pl[:]
|
||||
_pl.remove(seer) # remove self from list
|
||||
cli.msg(seer, 'You are a \u0002seer\u0002. \
|
||||
It is your job to detect the wolves, you may have a vision once per night. \
|
||||
Use "see <nick>" to see the role of a player.')
|
||||
cli.msg(seer, "Players: "+", ".join(_pl))
|
||||
|
||||
@cmd("!start")
|
||||
def start(cli, nick, chan, rest):
|
||||
pl = vars.list_players()
|
||||
@ -137,10 +331,10 @@ def start(cli, nick, chan, rest):
|
||||
if nick not in pl:
|
||||
cli.notice(nick, "You're currently not playing.")
|
||||
return
|
||||
now = time.time()
|
||||
if vars.CAN_START_TIME > now:
|
||||
cli.msg(chan, "Please wait at least {0} more seconds.".format(
|
||||
int(vars.CAN_START_TIME - now)))
|
||||
now = datetime.now()
|
||||
dur = int((vars.CAN_START_TIME - now).total_seconds())
|
||||
if dur > 0:
|
||||
cli.msg(chan, "Please wait at least {0} more seconds.".format(dur))
|
||||
return
|
||||
|
||||
if len(pl) < 4:
|
||||
@ -153,6 +347,7 @@ def start(cli, nick, chan, rest):
|
||||
nwolves = 0
|
||||
ndrunk = 0
|
||||
ncursed = 0
|
||||
ntraitor = 0
|
||||
|
||||
if len(pl) >= 8:
|
||||
nharlots = 1
|
||||
@ -189,17 +384,24 @@ def start(cli, nick, chan, rest):
|
||||
vars.ROLES["villager"] = pl
|
||||
|
||||
if ncursed:
|
||||
CURSED = random.choice(var.ROLES["villager"] + \
|
||||
var.ROLES.get("harlot", []) +\
|
||||
var.ROLES.get("village drunk", []))
|
||||
vars.CURSED = random.choice(vars.ROLES["villager"] + \
|
||||
vars.ROLES.get("harlot", []) +\
|
||||
vars.ROLES.get("village drunk", []))
|
||||
if ntraitor:
|
||||
possible = vars.ROLES["villager"]
|
||||
if ncursed:
|
||||
possible.remove(vars.CURSED) # Cursed traitors are not allowed
|
||||
vars.TRAITOR = random.choice(possible)
|
||||
|
||||
cli.msg(chan, "{0}: Welcome to Werewolf, the popular detective/social \
|
||||
party game (a theme of Mafia).".format(", ".join(vars.list_players())))
|
||||
cli.mode(chan, "+m")
|
||||
|
||||
vars.GAME_START_TIME = time.time()
|
||||
vars.ORIGINAL_ROLES = dict(vars.ROLES) # Make a copy
|
||||
transition_night(cli, chan)
|
||||
vars.DAY_TIMEDELTA = timedelta(0)
|
||||
vars.NIGHT_TIMEDELTA = timedelta(0)
|
||||
vars.DEAD = []
|
||||
transition_night(cli)
|
||||
|
||||
@cmd("!wait")
|
||||
def wait(cli, nick, chan, rest):
|
||||
@ -217,11 +419,11 @@ def wait(cli, nick, chan, rest):
|
||||
cli.msg(chan, "Limit has already been reached for extending the wait time.")
|
||||
return
|
||||
|
||||
now = time.time()
|
||||
now = datetime.now()
|
||||
if now > vars.CAN_START_TIME:
|
||||
vars.CAN_START_TIME = now + vars.EXTRA_WAIT
|
||||
vars.CAN_START_TIME = now + timedelta(seconds=vars.EXTRA_WAIT)
|
||||
else:
|
||||
vars.CAN_START_TIME += vars.EXTRA_WAIT
|
||||
vars.CAN_START_TIME += timedelta(seconds=vars.EXTRA_WAIT)
|
||||
vars.WAITED += 1
|
||||
cli.msg(chan, "{0} increased the wait \
|
||||
time by {1} seconds.".format(nick, vars.EXTRA_WAIT))
|
||||
@ -229,15 +431,21 @@ time by {1} seconds.".format(nick, vars.EXTRA_WAIT))
|
||||
@cmd("!reset", admin_only = True)
|
||||
def reset_game(cli, nick, chan, rest):
|
||||
vars.PHASE = "none"
|
||||
|
||||
if vars.TIMERS[0]:
|
||||
vars.TIMERS[0].cancel()
|
||||
vars.TIMERS[0] = None
|
||||
cli.mode(chan, "-m")
|
||||
for pl in vars.list_players():
|
||||
cli.mode(chan, "-v", pl)
|
||||
|
||||
for plr in vars.list_players():
|
||||
cli.mode(chan, "-v", "{0} {0}!*@*".format(plr))
|
||||
for deadguy in vars.DEAD:
|
||||
cli.mode(chan, "-q", "{0} {0}!*@*".format(deadguy))
|
||||
|
||||
vars.ROLES = {"person" : []}
|
||||
vars.ORIGINAL_ROLES = None
|
||||
vars.CURSED = ""
|
||||
vars.GAME_START_TIME = 0
|
||||
vars.CAN_START_TIME = 0
|
||||
vars.CAN_START_TIME = timedelta(0)
|
||||
vars.GUNNERS = {}
|
||||
vars.WAITED = 0
|
||||
vars.VICTIM = ""
|
||||
|
||||
Loading…
Reference in New Issue
Block a user