Minidodo
d3461ef462
Allowed viewing the most recent account logs by using !account log recent Allowed removing of chatbots Moved the so far hardcoded org prefixing into the DB, which allows dynamically changing them. Fix for orgrosters; org leaves are being detected now.
74 lines
3.4 KiB
Python
74 lines
3.4 KiB
Python
from core.aochat.BaseModule import BaseModule
|
|
from core.buddy_service import BuddyService
|
|
from core.chat_blob import ChatBlob
|
|
from core.command_alias_service import CommandAliasService
|
|
from core.command_param_types import Const, Character
|
|
from core.db import DB
|
|
from core.decorators import instance, command
|
|
from core.logger import Logger
|
|
from core.lookup.pork_service import PorkService
|
|
from core.text import Text
|
|
from core.igncore import IgnCore
|
|
from core.util import Util
|
|
|
|
|
|
# noinspection DuplicatedCode
|
|
@instance()
|
|
class BotController(BaseModule):
|
|
bots = []
|
|
threads = {}
|
|
|
|
def inject(self, registry):
|
|
self.logger = Logger(__name__)
|
|
self.bot: IgnCore = registry.get_instance("bot")
|
|
self.db: DB = registry.get_instance("db")
|
|
self.text: Text = registry.get_instance("text")
|
|
self.util: Util = registry.get_instance("util")
|
|
self.pork: PorkService = registry.get_instance("pork_service")
|
|
self.org_pork: PorkService = registry.get_instance("org_pork_service")
|
|
self.command_alias_service: CommandAliasService = registry.get_instance("command_alias_service")
|
|
self.buddy_service: BuddyService = registry.get_instance("buddy_service")
|
|
self.online = registry.get_instance("buddy_service")
|
|
|
|
def pre_start(self):
|
|
self.db.exec("CREATE TABLE IF NOT EXISTS org_bots(char_id int primary key not null, org_id int not null)")
|
|
|
|
def start(self):
|
|
pass
|
|
|
|
@command(command="orgs", params=[Const("bots"), Const("add"), Character("botname")], access_level="admin",
|
|
description="Add an bot to the bot list", sub_command="bots_mng")
|
|
def cmd_orgs_bots_add(self, request, _, _1, bot):
|
|
player = self.pork.request_char_info(bot.name, 5)
|
|
if self.db.exec("INSERT IGNORE INTO org_bots(char_id, org_id) VALUES(?, ?)", [bot.char_id, player.org_id]) == 0:
|
|
request.reply(f"The bot <highlight>{bot.name}<end> is already marked as a chatbot.")
|
|
else:
|
|
self.db.exec("DELETE FROM online where char_id=?", [bot.char_id])
|
|
request.reply(f"Successfully marked <highlight>{bot.name}<end> as a chatbot.")
|
|
|
|
@command(command="orgs", params=[Const("bots"), Const("rem"), Character("botname")], access_level="admin",
|
|
description="Add an bot to the bot list", sub_command="bots_mng")
|
|
def cmd_orgs_bots_rem(self, request, _, _1, bot):
|
|
|
|
if self.db.exec("DELETE FROM org_bots WHERE char_id=?", [bot.char_id]) == 0:
|
|
request.reply(f"<highlight>{bot.name}<end> is not a chatbot.")
|
|
else:
|
|
request.reply(f"Successfully removed <highlight>{bot.name}<end> as a chatbot. The character will show up in !online again after its next login.")
|
|
|
|
@command(command="orgs",
|
|
params=[Const("bots")],
|
|
access_level="moderator",
|
|
description="show all orgbots",
|
|
sub_command="bots_show")
|
|
def cmd_orgs_bots_show(self, _, _1):
|
|
def format_row(query):
|
|
bud = self.buddy_service.is_online(query["char_id"])
|
|
buddy = "<green>O<end>" if bud == 1 else "<red>O<end>" if bud == 0 else "<grey>U<end>"
|
|
return "- [{status}] <highlight>{name}<end> (<notice>{org_name}<end>)\n".format(**query, status=buddy)
|
|
|
|
data = self.db.query("SELECT * FROM org_bots o LEFT JOIN player p on o.char_id = p.char_id order by p.org_name")
|
|
blob = ""
|
|
for row in data:
|
|
blob += format_row(row)
|
|
return ChatBlob("Our Orgbots", blob)
|