def __init__(self, bot): self.bot = bot self.banphrases = [] self.enabled_banphrases = [] self.db_session = DBManager.create_session(expire_on_commit=False) if self.bot: self.bot.socket_manager.add_handler("banphrase.update", self.on_banphrase_update) self.bot.socket_manager.add_handler("banphrase.remove", self.on_banphrase_remove)
def __init__(self, socket_manager=None, module_manager=None, bot=None): UserDict.__init__(self) self.db_session = DBManager.create_session() self.internal_commands = {} self.db_commands = {} self.module_commands = {} self.data = {} self.bot = bot self.module_manager = module_manager if socket_manager: socket_manager.add_handler("module.update", self.on_module_reload) socket_manager.add_handler("command.update", self.on_command_update) socket_manager.add_handler("command.remove", self.on_command_remove)
def enable(self, bot): if not bot: return HandlerManager.add_handler("on_message", self.on_message, priority=100) HandlerManager.add_handler("on_commit", self.on_commit) if self.db_session is not None: self.db_session.commit() self.db_session.close() self.db_session = None self.db_session = DBManager.create_session() self.blacklisted_links = [] for link in self.db_session.query(BlacklistedLink): self.blacklisted_links.append(link) self.whitelisted_links = [] for link in self.db_session.query(WhitelistedLink): self.whitelisted_links.append(link)