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

Insert label tag to lines in labeled BATCHes (line_handler.py)

Browse source
This commit is contained in:
jesopo 2019-02-11 22:19:20 +00:00
parent 2f5a232c4e
commit 98b82c7f2c
1 changed files with 13 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
modules/line_handler.py
View file

@ -8,7 +8,11 @@ CAPABILITIES = {"multi-prefix", "chghost", "invite-notify", "account-tag",
"account-notify", "extended-join", "away-notify", "userhost-in-names", "account-notify", "extended-join", "away-notify", "userhost-in-names",
"draft/message-tags-0.2", "draft/message-tags-0.3", "server-time", "draft/message-tags-0.2", "draft/message-tags-0.3", "server-time",
"cap-notify", "batch", "draft/labeled-response", "draft/rename"} "cap-notify", "batch", "draft/labeled-response", "draft/rename"}
LABELED_BATCH = ["draft/labeled-response", "labeled-response"]
LABELED_BATCH = {
"labeled-response": "label",
"draft/labeled-response": "draft/label"
}
class Direction(enum.Enum): class Direction(enum.Enum):
SEND = 0 SEND = 0
@ -563,7 +567,15 @@ class Module(ModuleManager.BaseModule):
else: else:
batch = event["server"].batches[identifier] batch = event["server"].batches[identifier]
del 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]
for line in batch.lines: for line in batch.lines:
if add_tags:
line.tags.update(add_tags)
self._handle(line) self._handle(line)
# IRCv3 CHGHOST, a user's username and/or hostname has changed # IRCv3 CHGHOST, a user's username and/or hostname has changed