103 lines
3.5 KiB
Python
103 lines
3.5 KiB
Python
|
from bot import loaded_modules_dict
|
||
|
|
from bot.constants import TELNET_TIMEOUT_EXTENDED
|
||
|
|
from os import path, pardir
|
||
|
|
from time import time, sleep
|
||
|
|
import random
|
||
|
|
import re
|
||
|
|
|
||
|
|
module_name = path.basename(path.normpath(path.join(path.abspath(__file__), pardir, pardir)))
|
||
|
|
action_name = path.basename(path.abspath(__file__))[:-3]
|
||
|
|
|
||
|
|
|
||
|
|
def kill_entity(module, event_data, dispatchers_steamid=None):
|
||
|
|
timeout = TELNET_TIMEOUT_EXTENDED
|
||
|
|
timeout_start = time()
|
||
|
|
event_data[1]["action_identifier"] = action_name
|
||
|
|
|
||
|
|
entity_to_be_killed = event_data[1].get("entity_id", None)
|
||
|
|
|
||
|
|
command = "kill {}".format(entity_to_be_killed)
|
||
|
|
if not module.telnet.add_telnet_command_to_queue(command):
|
||
|
|
module.callback_fail(callback_fail, module, event_data, dispatchers_steamid)
|
||
|
|
return
|
||
|
|
|
||
|
|
poll_is_finished = False
|
||
|
|
# Modern format - no datetime/stardate prefix
|
||
|
|
regex = (
|
||
|
|
r"Entity\s(?P<zombie_name>.*)\s" + str(entity_to_be_killed) + r"\skilled"
|
||
|
|
)
|
||
|
|
number_of_attempts = 0
|
||
|
|
while not poll_is_finished and (time() < timeout_start + timeout):
|
||
|
|
number_of_attempts += 1
|
||
|
|
telnet_buffer_copy = (module.telnet.telnet_buffer + '.')[:-1]
|
||
|
|
for match in re.finditer(regex, telnet_buffer_copy, re.DOTALL):
|
||
|
|
return match
|
||
|
|
|
||
|
|
sleep(1)
|
||
|
|
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
def main_function(module, event_data, dispatchers_steamid):
|
||
|
|
action = event_data[1].get("action", None)
|
||
|
|
dataset = event_data[1].get("dataset")
|
||
|
|
entity_id = event_data[1].get("entity_id")
|
||
|
|
entity_name = event_data[1].get("entity_name")
|
||
|
|
|
||
|
|
if action is not None:
|
||
|
|
if action == "kill":
|
||
|
|
match = kill_entity(module, event_data, dispatchers_steamid)
|
||
|
|
if match is not False:
|
||
|
|
if entity_name == "zombieScreamer":
|
||
|
|
module.callback_success(callback_success, module, event_data, dispatchers_steamid, match)
|
||
|
|
|
||
|
|
module.callback_fail(callback_fail, module, event_data, dispatchers_steamid)
|
||
|
|
|
||
|
|
|
||
|
|
def callback_success(module, event_data, dispatchers_steamid, match=None):
|
||
|
|
action = event_data[1].get("action", None)
|
||
|
|
entity_id = event_data[1].get("entity_id")
|
||
|
|
entity_name = event_data[1].get("entity_name")
|
||
|
|
|
||
|
|
if all([
|
||
|
|
action is not None
|
||
|
|
]):
|
||
|
|
if entity_name == "zombieScreamer":
|
||
|
|
possible_maybes = [
|
||
|
|
"hopefully",
|
||
|
|
"probably dead, yes!",
|
||
|
|
"i think",
|
||
|
|
"i'm almost certain",
|
||
|
|
"yeah. definitely!!"
|
||
|
|
]
|
||
|
|
event_data = ['say_to_all', {
|
||
|
|
'message': '[CCFFCC]Screamer ([FFFFFF]{entity_id}[CCFFCC]) killed[-], [FFFFFF]{maybe}[-]...'.format(
|
||
|
|
entity_id=entity_id,
|
||
|
|
maybe=random.choice(possible_maybes)
|
||
|
|
)
|
||
|
|
}]
|
||
|
|
module.trigger_action_hook(module, event_data=event_data)
|
||
|
|
else:
|
||
|
|
event_data = ['say_to_all', {
|
||
|
|
'message': '[CCFFCC]entity ([FFFFFF]{entity_id}[CCFFCC]) killed[-]'.format(
|
||
|
|
entity_id=entity_id
|
||
|
|
)
|
||
|
|
}]
|
||
|
|
module.trigger_action_hook(module, event_data=event_data)
|
||
|
|
|
||
|
|
|
||
|
|
def callback_fail(module, event_data, dispatchers_steamid):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
action_meta = {
|
||
|
|
"description": "manages entity entries",
|
||
|
|
"main_function": main_function,
|
||
|
|
"callback_success": callback_success,
|
||
|
|
"callback_fail": callback_fail,
|
||
|
|
"requires_telnet_connection": False,
|
||
|
|
"enabled": True
|
||
|
|
}
|
||
|
|
|
||
|
|
loaded_modules_dict["module_" + module_name].register_action(action_name, action_meta)
|