Allow passing source Hostmask to IRCBatch

This commit is contained in:
jesopo 2019-09-19 18:16:10 +01:00
parent c7409c91be
commit efc0e197e5
2 changed files with 3 additions and 2 deletions

View file

@ -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,

View file

@ -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)