261 lines
8.8 KiB
Python
Executable file
261 lines
8.8 KiB
Python
Executable file
#!/usr/bin/env python
|
|
|
|
from os.path import exists
|
|
from datetime import datetime
|
|
from re import split, match, sub, IGNORECASE
|
|
from shutil import copyfile
|
|
|
|
from evennia import CmdSet
|
|
from evennia.utils import logger
|
|
|
|
from commands.command import Command
|
|
from typeclasses.characters import Character
|
|
from utils.word_list import routput
|
|
|
|
|
|
class Puppet(Character):
|
|
"""
|
|
Special character that if not puppetable, stays put.
|
|
Perhaps responding or logging information to the GM.
|
|
"""
|
|
def at_post_puppet(self, **kwargs):
|
|
"""
|
|
Called just after puppeting has been completed and all
|
|
Account<->Object links have been established.
|
|
|
|
Args:
|
|
**kwargs (dict): Arbitrary, optional arguments for users
|
|
overriding the call (unused by default).
|
|
Notes:
|
|
|
|
You can use `self.account` and `self.sessions.get()` to get
|
|
account and sessions at this point; the last entry in the
|
|
list from `self.sessions.get()` is the latest Session
|
|
puppeting this Object.
|
|
|
|
"""
|
|
self.msg(f"\nYou are puppeting |c{self.key}|n.")
|
|
self.msg((self.at_look(self.location), {"type": "look"}), options=None)
|
|
|
|
def at_pre_unpuppet(self, **kwargs):
|
|
"""
|
|
Reset our pose.
|
|
"""
|
|
super().at_pre_unpuppet()
|
|
self.execute_cmd("pose reset")
|
|
|
|
def at_post_unpuppet(self, account=None, session=None, **kwargs):
|
|
"""
|
|
Make the character object remain in the room.
|
|
|
|
Args:
|
|
account (DefaultAccount): The account object that just disconnected
|
|
from this object.
|
|
session (Session): Session controlling the connection that
|
|
just disconnected.
|
|
Keyword Args:
|
|
reason (str): If given, adds a reason for the unpuppet. This
|
|
is set when the user is auto-unpuppeted due to being link-dead.
|
|
**kwargs: Arbitrary, optional arguments for users
|
|
overriding the call (unused by default).
|
|
|
|
"""
|
|
self.msg(f"\nNo longer puppeting |c{self.key}|n.\n")
|
|
|
|
def get_display_desc(self, _, **kwargs):
|
|
"""
|
|
Return one of two appearances based on if it is being puppeted or not.
|
|
|
|
Set either or both:
|
|
|
|
@set me/desc_puppeted = "A clown bouncing up and down from a box."
|
|
@set me/desc_uppuppeted = "A colorful box with a closed lid."
|
|
|
|
If either is not specified, it uses the standard description.
|
|
"""
|
|
if self.tags.get(key='puppeted', category='account'):
|
|
return self.db.desc_puppeted if self.db.desc_puppeted else self.db.desc
|
|
else:
|
|
return self.db.desc_unpuppeted if self.db.desc_unpuppeted else self.db.desc
|
|
|
|
# ----------------------------------------------------------------------
|
|
# TRIGGERS
|
|
|
|
def get_character_label(self, character):
|
|
return character.get_display_name(self).split(' ')[-1]
|
|
|
|
def trigger_sequence(self, seq, character, *other_stuff):
|
|
self.delay_sequence(seq, 1, character.key, character.get_display_name(self),
|
|
self.get_character_label(character), *other_stuff)
|
|
|
|
def get_attribute_trigger(self, label, character):
|
|
"""
|
|
Return the attribute where 'key' is the label, and the
|
|
category can by either the 'character' or its display name.
|
|
"""
|
|
name = self.get_character_label(character)
|
|
|
|
return (self.attributes.get(key=label, category=character.key) or
|
|
self.attributes.get(key=label, category=name) or
|
|
self.attributes.get(key=label))
|
|
|
|
def move_triggers(self, label, character):
|
|
"""
|
|
Return a list of triggers matching 'label' and 'character'.
|
|
"""
|
|
seq = self.get_attribute_trigger(label, character)
|
|
if seq:
|
|
self.trigger_sequence(seq, character)
|
|
|
|
def other_arrive(self, character):
|
|
"""
|
|
Execute a command when a character arrives in the same location.
|
|
"""
|
|
self.move_triggers('arrive', character)
|
|
|
|
def other_leave(self, character):
|
|
"""
|
|
Execute a command when a character arrives in the same location.
|
|
"""
|
|
self.move_triggers('leave', character)
|
|
|
|
def say_triggers(self, label, character, speech):
|
|
trigs = self.get_attribute_trigger(label, character)
|
|
if trigs:
|
|
for _, trigger in enumerate(dict(trigs)):
|
|
seq = trigs[trigger]
|
|
if match(trigger, speech, IGNORECASE):
|
|
self.trigger_sequence(seq, character)
|
|
|
|
def other_say(self, speaker, speech):
|
|
self.say_triggers('say', speaker, speech)
|
|
|
|
def other_sayto(self, speaker, speech):
|
|
self.say_triggers('sayto', speaker, speech)
|
|
|
|
def other_given(self, giver, obj):
|
|
target = giver.get_display_name(self).split(' ')[-1]
|
|
self.execute_cmd(f"emote /Me says to /{target}, \"Thanks for the {obj.name}.\"")
|
|
|
|
|
|
class CmdShrubSay(Command):
|
|
"""Erase and write on the shrub's chalkboard.
|
|
"""
|
|
key = "say"
|
|
aliases = ["write"]
|
|
|
|
def func(self):
|
|
self.obj.write(self.args.strip())
|
|
|
|
|
|
class CmdSetShrubSay(CmdSet):
|
|
"""
|
|
The shrub's version of the 'say' command.
|
|
"""
|
|
def at_cmdset_creation(self):
|
|
super().at_cmdset_creation()
|
|
self.add(CmdShrubSay)
|
|
|
|
|
|
class Shrub(Puppet):
|
|
"""
|
|
The 'Shrub' has its own way of communicating.
|
|
"""
|
|
def at_object_creation(self):
|
|
"Called when a character is first created."
|
|
self.cmdset.add(CmdSetShrubSay, persistent=True)
|
|
|
|
def write(self, new_text):
|
|
"""
|
|
Change the readable message on the chalkboard.
|
|
|
|
This will also record the new message in the log.
|
|
"""
|
|
here = self.location
|
|
msg = routput("The shrub uses a branch to << brush ^ wipe >> << off ^ >> << the chalk from ^ >> its << small ^ >> chalkboard, and with another branch, << slowly ^ carefully ^ deliberately ^ >> starts to write << something ^ a message ^ >>.")
|
|
here.msg_contents(msg)
|
|
self.db.inside = f"The chalkboard reads: |w{new_text}|n"
|
|
self.record_msg(self.db.inside)
|
|
|
|
def find_actor(self, text, default=None):
|
|
"""
|
|
Extract the character performing the action from the text.
|
|
"""
|
|
if default:
|
|
return default
|
|
|
|
m = match(r".*\|b(.*?)\|n.*", text)
|
|
if m:
|
|
return m.group(1)
|
|
|
|
def capitalize_msg(self, text, msg_type=None):
|
|
"""
|
|
If text is lowercase, capitalize it.
|
|
|
|
Maybe prepend a 'The' to the front.
|
|
"""
|
|
if msg_type and msg_type in ('traverse', 'teleport'):
|
|
if match(r"^(\|[A-z])?[aeiou]", text):
|
|
return "An " + text
|
|
else:
|
|
return "A " + text
|
|
elif msg_type and msg_type == 'say' and match(r"^\|b[a-z]", text):
|
|
return "The " + text
|
|
elif match(r"^(\|[A-z])[a-z]", text) and len(text) > 2:
|
|
return text[0:1] + text[2].upper() + text[3:]
|
|
elif match(r"^(\|[A-z])?[a-z]", text) and len(text) > 2:
|
|
return text[0].upper() + text[1:]
|
|
return text
|
|
|
|
def to_html(self, text, msg_type=None):
|
|
"""
|
|
Convert the 'text' to an HTML formatted string.
|
|
"""
|
|
# Bold anything white:
|
|
text = sub(r"\|w(.*?)\|n", '<b>\\1</b>', text)
|
|
# Remove the rest:
|
|
text = sub(r"\|[A-z]", '', text)
|
|
# Remove initial or final carriage returns:
|
|
text = sub(r"^\n", '', text)
|
|
text = sub(r"\n$", '', text)
|
|
# Convert the carriage returns:
|
|
text = sub(r"\n", ' <br/>\n', text)
|
|
|
|
if msg_type and msg_type == 'look':
|
|
text = text[0].upper + text[1:]
|
|
return f"<div class=\"special\"><p>{text}</p></div>\n"
|
|
return f"<p>{text}</p>\n"
|
|
|
|
def record_msg(self, text, actor=None):
|
|
msg_type = None
|
|
|
|
if isinstance(text, tuple):
|
|
if text[1] and isinstance(text[1], dict):
|
|
msg_type = text[1]['type']
|
|
msg = self.capitalize_msg(text[0], msg_type)
|
|
else:
|
|
msg = self.capitalize_msg(str(text))
|
|
|
|
actor = self.find_actor(msg, actor)
|
|
msg = self.to_html(msg, msg_type)
|
|
filename = datetime.today().strftime('transcripts/%Y-%m-%d.html')
|
|
if not exists(filename):
|
|
copyfile("transcripts/header-template.html", filename)
|
|
|
|
with open(filename, "a") as myfile:
|
|
myfile.write(msg)
|
|
|
|
# Follow-up Actions ...
|
|
if match(r".* arrives .*", msg) and actor and \
|
|
msg_type in ('traverse', 'teleport'):
|
|
self.execute_cmd(f"look {actor}")
|
|
|
|
def msg(self, text=None, from_obj=None, session=None, **kwargs):
|
|
"""
|
|
Record everything that happens in the room for a transcript.
|
|
|
|
Some key events may trigger more actions to get more information.
|
|
"""
|
|
self.record_msg(text, from_obj)
|
|
super().msg(text, from_obj, session)
|