From 367eb1c60ed295b75b7d41f8e5d4955b8037bdad Mon Sep 17 00:00:00 2001 From: jesopo Date: Thu, 30 May 2019 15:34:39 +0100 Subject: Make labeled-resposnes it's own module, tag and track every sent line --- modules/line_handler/__init__.py | 20 ++++++++------------ 1 file changed, 8 insertions(+), 12 deletions(-) (limited to 'modules/line_handler/__init__.py') diff --git a/modules/line_handler/__init__.py b/modules/line_handler/__init__.py index 1d4881f9..a1b05426 100644 --- a/modules/line_handler/__init__.py +++ b/modules/line_handler/__init__.py @@ -2,10 +2,6 @@ import enum from src import EventManager, ModuleManager, utils from . import channel, core, ircv3, message, user -LABELED_BATCH = { - "draft/labeled-response": "draft/label" -} - class Module(ModuleManager.BaseModule): def _handle(self, server, line): hooks = self.events.on("raw.received").on(line.command).get_hooks() @@ -186,20 +182,20 @@ class Module(ModuleManager.BaseModule): if modifier == "+": batch_type = event["args"][1] - event["server"].batches[identifier] = utils.irc.IRCRecvBatch( - identifier, batch_type, event["tags"]) + batch = utils.irc.IRCRecvBatch(identifier, batch_type, + event["tags"]) + event["server"].batches[identifier] = batch + + self.events.on("received.batch.start").call(batch=batch, + server=event["server"]) else: batch = event["server"].batches[identifier] del event["server"].batches[identifier] - add_tags = {} - if batch.type in LABELED_BATCH.keys(): - tag_name = LABELED_BATCH[batch.type] - add_tags[tag_name] = batch.tags[tag_name] + self.events.on("received.batch.end").call(batch=batch, + server=event["server"]) for line in batch.lines: - if add_tags: - line.tags.update(add_tags) self._handle(event["server"], line) # IRCv3 CHGHOST, a user's username and/or hostname has changed -- cgit v1.3.1-10-gc9f91