!roles/!gamestats accept partial names, new complete_match/get_victim functions to clean up code for role commands (reverted from commit c1a937f76373abe93497234fc81c8e0873da6bee)

This commit is contained in:
jacob1 2014-12-14 16:30:25 -05:00
parent c1a937f763
commit bc60357ae5

View File

@ -205,33 +205,6 @@ def check_for_modes(cli, rnick, chan, modeaction, *target):
elif change in var.USERS[trgt]["modes"]:
var.USERS[trgt]["modes"].remove(change)
#completes a partial nickname or string from a list
def complete_match(string, matches):
num_matches = 0
bestmatch = string
for possible in matches:
if possible.startswith(string):
bestmatch = possible
num_matches += 1
if num_matches != 1:
return None
else:
return bestmatch
#wrapper around complete_match() used for roles
def get_victim(cli, nick, victim, self_in_list = False):
if not victim:
cli.notice(cli, nick, "Not enough parameters")
return
pl = [x for x in var.list_players() if x != nick or self_in_list]
pll = [x.lower() for x in pl]
tempvictim = complete_match(victim.lower(), pll)
if not tempvictim:
cli.notice(nick, "\u0002{0}\u0002 is currently not playing.".format(victim))
return
return pl[pll.index(tempvictim)] #convert back to normal casing
def mass_mode(cli, md):
""" Example: mass_mode(cli, (('+v', 'asdf'), ('-v','wobosd'))) """
lmd = len(md) # store how many mode changes to do
@ -2265,10 +2238,30 @@ def goat(cli, nick, chan, rest):
cli.notice(nick, 'This can only be done once per day.')
return
victim = get_victim(cli, nick, rest.split(' ')[0], True)
if not victim:
ul = list(var.USERS.keys())
ull = [x.lower() for x in ul]
rest = rest.split(' ')[0].strip().lower()
if not rest:
cli.notice(nick, 'Not enough parameters.')
return
matches = 0
for player in ull:
if rest == player:
victim = player
break
if player.startswith(rest):
victim = player
matches += 1
else:
if matches != 1:
pm(cli, nick, '\x02{}\x02 is not in this channel.'.format(rest))
return
victim = ul[ull.index(victim)]
goatact = random.choice(('kicks', 'headbutts'))
cli.msg(chan, '\x02{}\x02\'s goat walks by and {} \x02{}\x02.'.format(
@ -3251,7 +3244,7 @@ def vote(cli, nick, chan, rest):
if chan != botconfig.CHANNEL:
return
rest = re.split(" +",rest)[0].strip()
rest = re.split(" +",rest)[0].strip().lower()
if not rest:
show_votes(cli, nick, chan, rest)
@ -3278,9 +3271,23 @@ def vote(cli, nick, chan, rest):
if nick in var.NO_LYNCH:
var.NO_LYNCH.remove(nick)
voted = get_victim(cli, nick, rest, var.SELF_LYNCH_ALLOWED)
if not voted:
return
pl = var.list_players()
pl_l = [x.strip().lower() for x in pl]
matches = 0
for player in pl_l:
if rest == player:
target = player
break
if player.startswith(rest):
target = player
matches += 1
else:
if matches != 1:
pm(cli, nick, "\u0002{0}\u0002 is currently not playing.".format(rest))
return
voted = pl[pl_l.index(target)]
if not var.SELF_LYNCH_ALLOWED:
if nick == voted:
@ -3643,21 +3650,38 @@ def shoot(cli, nick, chan, rest):
"Please wait patiently for morning."))
return
if nick not in var.GUNNERS.keys() and nick not in var.WOLF_GUNNERS.keys():
cli.notice(nick, "You don't have a gun.")
pm(cli, nick, "You don't have a gun.")
return
elif ((nick in var.GUNNERS.keys() and not var.GUNNERS[nick]) or
(nick in var.WOLF_GUNNERS.keys() and not var.WOLF_GUNNERS[nick])):
cli.notice(nick, "You don't have any more bullets.")
pm(cli, nick, "You don't have any more bullets.")
return
elif nick in var.SILENCED:
pm(cli, nick, "You have been silenced, and are unable to use any special powers.")
return
victim = get_victim(cli, nick, rest.split(' ')[0])
victim = re.split(" +",rest)[0].strip().lower()
if not victim:
cli.notice(nick, "Not enough parameters")
return
pl = var.list_players()
pll = [x.lower() for x in pl]
matches = 0
for player in pll:
if victim == player:
target = player
break
if player.startswith(victim):
target = player
matches += 1
else:
if matches != 1:
pm(cli, nick, "\u0002{0}\u0002 is currently not playing.".format(victim))
return
victim = pl[pll.index(target)]
if victim == nick:
cli.notice(nick, "You are holding it the wrong way.")
return
# get actual victim
victim = choose_target(nick, victim)
@ -3776,35 +3800,61 @@ def kill(cli, nick, rest):
if role in ("wolf", "werecrow") and var.DISEASED_WOLVES:
pm(cli, nick, "You are feeling ill, and are unable to kill anyone tonight.")
return
pieces = rest.split(' ')
pieces = [p.strip().lower() for p in re.split(" +",rest)]
victim = pieces[0]
victim2 = None
if role in ("wolf", "werecrow") and var.ANGRY_WOLVES:
if len(pieces) > 1:
if len(pieces) > 2 and pieces[1] == "and":
try:
if pieces[1] == "and":
victim2 = pieces[2]
else:
victim2 = pieces[1]
else:
except IndexError:
victim2 = None
if not victim:
pm(cli, nick, "Not enough parameters")
return
if role == "werecrow": # Check if flying to observe
if var.OBSERVED.get(nick):
pm(cli, nick, ("You have already transformed into a crow; therefore, "+
"you are physically unable to kill a villager."))
return
pl = var.list_players()
allwolves = var.list_players(var.WOLFTEAM_ROLES)
allvills = []
for p in var.list_players():
for p in pl:
if p not in allwolves:
allvills.append(p)
victim = get_victim(cli, nick, victim)
if not victim:
return
if victim2 != None:
victim2 = get_victim(cli, nick, victim2)
if not victim2:
pll = [x.lower() for x in pl]
matches = 0
for player in pll:
if victim == player:
target = player
break
if player.startswith(victim):
target = player
matches += 1
else:
if matches != 1:
pm(cli, nick, "\u0002{0}\u0002 is currently not playing.".format(victim))
return
victim = pl[pll.index(target)]
if victim2 != None:
matches = 0
for player in pll:
if victim2 == player:
target = player
break
if player.startswith(victim2):
target = player
matches += 1
else:
if matches != 1:
pm(cli, nick, "\u0002{0}\u0002 is currently not playing.".format(victim2))
return
victim2 = pl[pll.index(target)]
if victim == nick or victim2 == nick:
if nick in var.VENGEFUL_GHOSTS.keys():
@ -3822,7 +3872,7 @@ def kill(cli, nick, rest):
return
if role in ("wolf", "werecrow"):
if victim in allwolves or victim2 in allwolves:
if victim in var.list_players(var.WOLFCHAT_ROLES) or victim2 in var.list_players(var.WOLFCHAT_ROLES):
pm(cli, nick, "You may only kill villagers, not other wolves.")
return
if var.ANGRY_WOLVES and victim2 != None:
@ -3883,13 +3933,28 @@ def guard(cli, nick, rest):
if nick in var.SILENCED:
pm(cli, nick, "You have been silenced, and are unable to use any special powers.")
return
victim = re.split(" +",rest)[0].strip().lower()
if not victim:
pm(cli, nick, "Not enough parameters")
return
if var.GUARDED.get(nick):
pm(cli, nick, "You are already protecting someone tonight.")
return
victim = get_victim(cli, nick, rest.split(' ')[0], role != "bodyguard" and var.GUARDIAN_ANGEL_CAN_GUARD_SELF)
if not victim:
return
pl = var.list_players()
pll = [x.lower() for x in pl]
matches = 0
for player in pll:
if victim == player:
target = player
break
if player.startswith(victim):
target = player
matches += 1
else:
if matches != 1:
pm(cli, nick, "\u0002{0}\u0002 is currently not playing.".format(victim))
return
victim = pl[pll.index(target)]
if role == "guardian angel" and var.LASTGUARDED.get(nick) == victim:
pm(cli, nick, ("You protected \u0002{0}\u0002 last night. " +
"You cannot protect the same person two nights in a row.").format(victim))
@ -3938,11 +4003,26 @@ def observe(cli, nick, rest):
if nick in var.SILENCED:
pm(cli, nick, "You have been silenced, and are unable to use any special powers.")
return
victim = get_victim(cli, nick, rest.split(' ')[0])
victim = re.split(" +", rest)[0].strip().lower()
if not victim:
pm(cli, nick, "Not enough parameters")
return
if victim == nick:
pl = var.list_players()
pll = [x.lower() for x in pl]
matches = 0
for player in pll:
if victim == player:
target = player
break
if player.startswith(victim):
target = player
matches += 1
else:
if matches != 1:
pm(cli, nick,"\u0002{0}\u0002 is currently not playing.".format(victim))
return
victim = pl[pll.index(target)]
if victim == nick.lower():
if role == "werecrow":
pm(cli, nick, "Instead of doing that, you should probably go kill someone.")
else:
@ -4004,10 +4084,25 @@ def investigate(cli, nick, rest):
if nick in var.INVESTIGATED:
pm(cli, nick, "You may only investigate one person per round.")
return
victim = get_victim(cli, nick, rest.split(' ')[0])
victim = re.split(" +", rest)[0].strip().lower()
if not victim:
pm(cli, nick, "Not enough parameters")
return
pl = var.list_players()
pll = [x.lower() for x in pl]
matches = 0
for player in pll:
if victim == player:
target = player
break
if player.startswith(victim):
target = player
matches += 1
else:
if matches != 1:
pm(cli, nick,"\u0002{0}\u0002 is currently not playing.".format(victim))
return
victim = pl[pll.index(target)]
if victim == nick:
pm(cli, nick, "Investigating yourself would be a waste.")
return
@ -4048,10 +4143,24 @@ def hvisit(cli, nick, rest):
pm(cli, nick, ("You are already spending the night "+
"with \u0002{0}\u0002.").format(var.HVISITED[nick]))
return
victim = get_victim(cli, nick, rest.split(' ')[0], True)
victim = re.split(" +",rest)[0].strip().lower()
if not victim:
pm(cli, nick, "Not enough parameters")
return
pll = [x.lower() for x in var.list_players()]
matches = 0
for player in pll:
if victim == player:
target = player
break
if player.startswith(victim):
target = player
matches += 1
else:
if matches != 1:
pm(cli, nick,"\u0002{0}\u0002 is currently not playing.".format(victim))
return
victim = var.list_players()[pll.index(target)]
if nick == victim: # Staying home
var.HVISITED[nick] = None
pm(cli, nick, "You have chosen to stay home for the night.")
@ -4061,9 +4170,8 @@ def hvisit(cli, nick, rest):
return
var.HVISITED[nick] = victim
pm(cli, nick, ("You are spending the night with \u0002{0}\u0002. "+
"Have a good time!").format(victim))
if nick != victim: #prevent exchange totem weirdness
pm(cli, victim, ("You are spending the night with \u0002{0}"+
"Have a good time!").format(var.HVISITED[nick]))
pm(cli, var.HVISITED[nick], ("You are spending the night with \u0002{0}"+
"\u0002. Have a good time!").format(nick))
var.LOGGER.logBare(var.HVISITED[nick], "VISITED", nick)
chk_nightdone(cli)
@ -4092,10 +4200,25 @@ def see(cli, nick, rest):
if nick in var.SEEN:
pm(cli, nick, "You may only have one vision per round.")
return
victim = get_victim(cli, nick, rest.split(' ')[0])
victim = re.split(" +",rest)[0].strip().lower()
pl = var.list_players()
pll = [x.lower() for x in pl]
if not victim:
pm(cli, nick, "Not enough parameters")
return
matches = 0
for player in pll:
if victim == player:
target = player
break
if player.startswith(victim):
target = player
matches += 1
else:
if matches != 1:
pm(cli, nick,"\u0002{0}\u0002 is currently not playing.".format(victim))
return
victim = pl[pll.index(target)]
if victim == nick:
pm(cli, nick, "Seeing yourself would be a waste.")
return
@ -4149,10 +4272,25 @@ def give(cli, nick, rest):
if nick in var.SHAMANS:
pm(cli, nick, "You have already given out your totem this round.")
return
victim = get_victim(cli, nick, rest.split(' ')[0], True)
victim = re.split(" +",rest)[0].strip().lower()
pl = var.list_players()
pll = [x.lower() for x in pl]
if not victim:
pm(cli, nick, "Not enough parameters")
return
matches = 0
for player in pll:
if victim == player:
target = player
break
if player.startswith(victim):
target = player
matches += 1
else:
if matches != 1:
pm(cli, nick,"\u0002{0}\u0002 is currently not playing.".format(victim))
return
victim = pl[pll.index(target)]
if nick in var.LASTGIVEN and var.LASTGIVEN[nick] == victim:
pm(cli, nick, "You gave your totem to \u0002{0}\u0002 last time, you must choose someone else.".format(victim))
return
@ -4268,23 +4406,51 @@ def choose(cli, nick, rest):
# no var.SILENCED check for night 1 only roles; silence should only apply for the night after
# but just in case, it also sucks if the one night you're allowed to act is when you are
# silenced, so we ignore it here anyway.
pieces = rest.split(' ')
pieces = [p.strip().lower() for p in re.split(" +",rest)]
victim = pieces[0]
if len(pieces) > 1:
if len(pieces) > 2 and pieces[1] == "and":
try:
if pieces[1] == "and":
victim2 = pieces[2]
else:
victim2 = pieces[1]
else:
except IndexError:
victim2 = None
victim = get_victim(cli, nick, victim, True)
if not victim:
return
victim2 = get_victim(cli, nick, victim2, True)
if not victim2:
if not victim or not victim2:
pm(cli, nick, "Not enough parameters")
return
pl = var.list_players()
pll = [x.lower() for x in pl]
matches = 0
for player in pll:
if victim == player:
target = player
break
if player.startswith(victim):
target = player
matches += 1
else:
if matches != 1:
pm(cli, nick, "\u0002{0}\u0002 is currently not playing.".format(victim))
return
victim = pl[pll.index(target)]
matches = 0
for player in pll:
if victim2 == player:
target = player
break
if player.startswith(victim2):
target = player
matches += 1
else:
if matches != 1:
pm(cli, nick, "\u0002{0}\u0002 is currently not playing.".format(victim2))
return
victim2 = pl[pll.index(target)]
if victim == victim2:
pm(cli, nick, "You must choose two different people.")
return
@ -4342,10 +4508,30 @@ def target(cli, nick, rest):
if nick in var.SILENCED:
pm(cli, nick, "You have been silenced, and are unable to use any special powers.")
return
victim = get_victim(cli, nick, rest.split(' ')[0])
pieces = [p.strip().lower() for p in re.split(" +",rest)]
victim = pieces[0]
if not victim:
pm(cli, nick, "Not enough parameters")
return
pl = var.list_players()
pll = [x.lower() for x in pl]
matches = 0
for player in pll:
if victim == player:
target = player
break
if player.startswith(victim):
target = player
matches += 1
else:
if matches != 1:
pm(cli, nick, "\u0002{0}\u0002 is currently not playing.".format(victim))
return
victim = pl[pll.index(target)]
if nick == victim:
pm(cli, nick, "You may not target yourself.")
return
@ -4378,10 +4564,30 @@ def hex(cli, nick, rest):
if nick in var.SILENCED:
pm(cli, nick, "You have been silenced, and are unable to use any special powers.")
return
victim = get_victim(cli, nick, rest.split(' ')[0])
pieces = [p.strip().lower() for p in re.split(" +",rest)]
victim = pieces[0]
if not victim:
pm(cli, nick, "Not enough parameters")
return
pl = var.list_players()
pll = [x.lower() for x in pl]
matches = 0
for player in pll:
if victim == player:
target = player
break
if player.startswith(victim):
target = player
matches += 1
else:
if matches != 1:
pm(cli, nick, "\u0002{0}\u0002 is currently not playing.".format(victim))
return
victim = pl[pll.index(target)]
if nick == victim:
pm(cli, nick, "You may not target yourself.")
return
@ -4427,10 +4633,30 @@ def clone(cli, nick, rest):
# but just in case, it also sucks if the one night you're allowed to act is when you are
# silenced, so we ignore it here anyway.
victim = get_victim(cli, nick, rest.split(' ')[0])
pieces = [p.strip().lower() for p in re.split(" +",rest)]
victim = pieces[0]
if not victim:
pm(cli, nick, "Not enough parameters")
return
pl = var.list_players()
pll = [x.lower() for x in pl]
matches = 0
for player in pll:
if victim == player:
target = player
break
if player.startswith(victim):
target = player
matches += 1
else:
if matches != 1:
pm(cli, nick, "\u0002{0}\u0002 is currently not playing.".format(victim))
return
victim = pl[pll.index(target)]
if nick == victim:
pm(cli, nick, "You may not target yourself.")
return
@ -6084,11 +6310,9 @@ def listroles(cli, nick, chan, rest):
#read game mode to get roles for
if len(rest[0]) and not rest[0].isdigit():
gamemode = rest[0]
if gamemode not in var.GAME_MODES.keys():
gamemode = complete_match(rest[0], var.GAME_MODES.keys() - ["roles"])
if gamemode in var.GAME_MODES.keys() and gamemode != "roles":
mode = var.GAME_MODES[gamemode][0]()
#check for valid game mode ("roles" gamemode is treated as invalid)
if rest[0] != "roles" and rest[0] in var.GAME_MODES.keys():
mode = var.GAME_MODES[rest[0]][0]()
if hasattr(mode, "ROLE_INDEX"):
roleindex = getattr(mode, "ROLE_INDEX")
if hasattr(mode, "ROLE_GUIDE"):
@ -6304,14 +6528,12 @@ def game_stats(cli, nick, chan, rest):
return
gamemode = var.CURRENT_GAMEMODE
rest = rest.split()
rest = rest.strip().split()
# Check for gamemode
if len(rest) and not rest[0].isdigit():
gamemode = rest[0]
if gamemode not in var.GAME_MODES.keys():
gamemode = complete_match(gamemode, var.GAME_MODES.keys())
if not gamemode:
cli.notice(nick, "{0} is not a valid game mode".format(rest[0]))
cli.notice(nick, "{0} is not a valid game mode".format(gamemode))
return
rest.pop(0)
# Check for invalid input
@ -6433,11 +6655,18 @@ def game(cli, nick, chan, rest):
return
if gamemode not in var.GAME_MODES.keys():
match = complete_match(gamemode, var.GAME_MODES.keys() - ["roles"])
if not match:
#players can vote by only using partial name
matches = 0
possiblegamemode = gamemode
for mode in var.GAME_MODES.keys():
if mode.startswith(gamemode) and mode != "roles":
possiblegamemode = mode
matches += 1
if matches != 1:
cli.notice(nick, "\002{0}\002 is not a valid game mode.".format(gamemode))
return
gamemode = match
else:
gamemode = possiblegamemode
if gamemode != "roles":
var.GAMEMODE_VOTES[cloak] = gamemode
@ -6612,15 +6841,20 @@ if botconfig.DEBUG_MODE or botconfig.ALLOWED_NORMAL_MODE_COMMANDS:
return
if rest:
rest = gamemode = rest.strip().lower()
rest = mode = rest.strip().lower()
if rest not in var.GAME_MODES.keys() and not rest.startswith("roles"):
rest = rest.split()[0]
gamemode = complete_match(rest, var.GAME_MODES.keys())
if not gamemode:
#players can vote by only using partial name
matches = 0
for gamemode in var.GAME_MODES.keys():
if gamemode.startswith(rest):
mode = gamemode
matches += 1
if matches != 1:
cli.notice(nick, "\002{0}\002 is not a valid game mode.".format(rest))
return
if cgamemode(cli, gamemode):
if cgamemode(cli, mode):
cli.msg(chan, ('\u0002{}\u0002 has changed the game settings '
'successfully.').format(nick))
var.FGAMED = True