Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

Allow passing source Hostmask to IRCBatch

Browse source
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
Show stats Download patch file Download diff file Expand all files Collapse all files

2
modules/line_handler/__init__.py
View file

@ -179,7 +179,7 @@ class Module(ModuleManager.BaseModule):
args = event["line"].args[2:] args = event["line"].args[2:]
batch = utils.irc.IRCBatch(identifier, batch_type, args, batch = utils.irc.IRCBatch(identifier, batch_type, args,
event["line"].tags) event["line"].tags, source=event["line"].source)
event["server"].batches[identifier] = batch event["server"].batches[identifier] = batch
self.events.on("received.batch.start").call(batch=batch, self.events.on("received.batch.start").call(batch=batch,

3
src/utils/irc/__init__.py
View file

@ -258,11 +258,12 @@ def parse_ctcp(s: str) -> typing.Optional[CTCPMessage]:
class IRCBatch(object): class IRCBatch(object):
def __init__(self, identifier: str, batch_type: str, args: typing.List[str], 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.identifier = identifier
self.type = batch_type self.type = batch_type
self.args = args self.args = args
self.tags = tags or {} self.tags = tags or {}
self.source = source
self._lines = [] # type: typing.List[IRCLine.ParsedLine] self._lines = [] # type: typing.List[IRCLine.ParsedLine]
def add_line(self, line: IRCLine.ParsedLine): def add_line(self, line: IRCLine.ParsedLine):
self._lines.append(line) self._lines.append(line)