diff --git a/modules/line_handler/__init__.py b/modules/line_handler/__init__.py index 0df30109..714bc23b 100644 --- a/modules/line_handler/__init__.py +++ b/modules/line_handler/__init__.py @@ -179,7 +179,7 @@ class Module(ModuleManager.BaseModule): args = event["line"].args[2:] batch = utils.irc.IRCBatch(identifier, batch_type, args, - event["line"].tags) + event["line"].tags, source=event["line"].source) event["server"].batches[identifier] = batch self.events.on("received.batch.start").call(batch=batch, diff --git a/src/utils/irc/__init__.py b/src/utils/irc/__init__.py index 57c65fdb..bd7c09ec 100644 --- a/src/utils/irc/__init__.py +++ b/src/utils/irc/__init__.py @@ -258,11 +258,12 @@ def parse_ctcp(s: str) -> typing.Optional[CTCPMessage]: class IRCBatch(object): def __init__(self, identifier: str, batch_type: str, args: typing.List[str], - tags: typing.Dict[str, str]=None): + tags: typing.Dict[str, str]=None, source: IRCLine.Hostmask=None): self.identifier = identifier self.type = batch_type self.args = args self.tags = tags or {} + self.source = source self._lines = [] # type: typing.List[IRCLine.ParsedLine] def add_line(self, line: IRCLine.ParsedLine): self._lines.append(line)