145 lines
4.4 KiB
Python
145 lines
4.4 KiB
Python
import re
|
|
import random
|
|
|
|
import src.settings as var
|
|
from src.utilities import *
|
|
from src import debuglog, errlog, plog
|
|
from src.decorators import cmd, event_listener
|
|
from src.messages import messages
|
|
from src.events import Event
|
|
|
|
SEEN = set()
|
|
KILLS = {}
|
|
SICK = {}
|
|
LYCANS = {}
|
|
|
|
@cmd("see", chan=False, pm=True, playing=True, silenced=True, phases=("night",), roles=("doomsayer",))
|
|
def see(cli, nick, chan, rest):
|
|
"""Use your paranormal senses to determine a player's doom."""
|
|
role = get_role(nick)
|
|
if nick in SEEN:
|
|
pm(cli, nick, messages["seer_fail"])
|
|
return
|
|
victim = get_victim(cli, nick, re.split(" +",rest)[0], False)
|
|
if not victim:
|
|
return
|
|
|
|
if victim == nick:
|
|
pm(cli, nick, messages["no_see_self"])
|
|
return
|
|
if in_wolflist(nick, victim):
|
|
pm(cli, nick, messages["no_see_wolf"])
|
|
return
|
|
|
|
evt = Event("targeted_command", {"target": victim, "misdirection": True, "exchange": True})
|
|
evt.dispatch(cli, var, "see", nick, victim, frozenset({"detrimental", "immediate"}))
|
|
if evt.prevent_default:
|
|
return
|
|
victim = evt.data["target"]
|
|
victimrole = get_role(victim)
|
|
|
|
mode = random.randint(1, 3)
|
|
if mode == 1:
|
|
mode = "DEATH"
|
|
pm(cli, nick, messages["doomsayer_death"].format(victim))
|
|
KILLS[nick] = victim
|
|
elif mode == 2:
|
|
mode = "LYCAN"
|
|
pm(cli, nick, messages["doomsayer_lycan"].format(victim))
|
|
LYCANS[nick] = victim
|
|
elif mode == 3:
|
|
mode = "SICK"
|
|
pm(cli, nick, messages["doomsayer_sick"].format(victim))
|
|
SICK[nick] = victim
|
|
|
|
debuglog("{0} ({1}) SEE: {2} ({3}) - {4}".format(nick, role, victim, victimrole, mode))
|
|
relay_wolfchat_command(cli, nick, messages["doomsayer_wolfchat"].format(nick, victim), ("doomsayer",), is_wolf_command=True)
|
|
|
|
SEEN.add(nick)
|
|
chk_nightdone(cli)
|
|
|
|
@event_listener("rename_player")
|
|
def on_rename(evt, cli, var, prefix, nick):
|
|
if prefix in SEEN:
|
|
SEEN.remove(prefix)
|
|
SEEN.add(nick)
|
|
for dictvar in (KILLS, SICK, LYCANS):
|
|
kvp = []
|
|
for a, b in dictvar.items():
|
|
if a == prefix:
|
|
a = nick
|
|
if b == prefix:
|
|
b = nick
|
|
kvp.append((a, b))
|
|
dictvar.update(kvp)
|
|
if prefix in dictvar:
|
|
del dictvar[prefix]
|
|
|
|
@event_listener("night_acted")
|
|
def on_acted(evt, cli, var, nick, sender):
|
|
if nick in SEEN:
|
|
evt.data["acted"] = True
|
|
|
|
@event_listener("exchange_roles")
|
|
def on_exchange(evt, cli, var, actor, nick, actor_role, nick_role):
|
|
if actor_role == "doomsayer" and nick_role != "doomsayer":
|
|
SEEN.discard(actor)
|
|
if actor in KILLS:
|
|
del KILLS[actor]
|
|
elif actor in SICK:
|
|
del SICK[actor]
|
|
elif actor in LYCANS:
|
|
del LYCANS[actor]
|
|
elif nick_role == "doomsayer" and actor_role != "doomsayer":
|
|
SEEN.discard(nick)
|
|
if nick in KILLS:
|
|
del KILLS[nick]
|
|
elif nick in SICK:
|
|
del SICK[nick]
|
|
elif nick in LYCANS:
|
|
del LYCANS[nick]
|
|
|
|
@event_listener("del_player")
|
|
def on_del_player(evt, cli, var, nick, nickrole, nicktpls, death_triggers):
|
|
SEEN.discard(nick)
|
|
for dictvar in (KILLS, SICK, LYCANS):
|
|
for k, v in list(dictvar.items()):
|
|
if nick == k or nick == v:
|
|
del dictvar[k]
|
|
|
|
@event_listener("chk_nightdone")
|
|
def on_chk_nightdone(evt, cli, var):
|
|
evt.data["actedcount"] += len(SEEN)
|
|
evt.data["nightroles"].extend(get_roles("doomsayer"))
|
|
|
|
@event_listener("transition_day", priority=2)
|
|
def on_transition_day(evt, cli, var):
|
|
for k, d in list(KILLS.items()):
|
|
evt.data["victims"].append(d)
|
|
# even though doomsayer is a wolf, remove from onlybywolves since
|
|
# that particular item indicates that they were the target of a wolf !kill.
|
|
# If doomsayer doesn't remove this, roles such as harlot or monster will not
|
|
# die if they are the target of a doomsayer !see that ends up killing the target.
|
|
evt.data["onlybywolves"].discard(d)
|
|
evt.data["killers"][d].append(k)
|
|
|
|
@event_listener("begin_day")
|
|
def on_begin_day(evt, cli, var):
|
|
var.DISEASED.update(SICK.values())
|
|
var.SILENCED.update(SICK.values())
|
|
var.LYCANTHROPES.update(LYCANS.values())
|
|
|
|
SEEN.clear()
|
|
KILLS.clear()
|
|
SICK.clear()
|
|
LYCANS.clear()
|
|
|
|
@event_listener("reset")
|
|
def on_reset(evt, var):
|
|
SEEN.clear()
|
|
KILLS.clear()
|
|
SICK.clear()
|
|
LYCANS.clear()
|
|
|
|
# vim: set sw=4 expandtab:
|