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

Implement client-to-server BATCHes

Browse source
This commit is contained in:
jesopo 2019-06-16 09:52:54 +01:00
parent 055eed24fd
commit 919a488e86
1 changed files with 18 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -1,4 +1,4 @@
import fnmatch, json, string, re, typing
import fnmatch, json, string, re, typing, uuid
from src import IRCLine, utils
from . import protocol
@ -259,17 +259,32 @@ 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]={}):
tags: typing.Dict[str, str]=None):
self.identifier = identifier
self.type = batch_type
self.args = args
self.tags = tags
self.tags = tags or {}
self._lines = [] # type: typing.List[IRCLine.ParsedLine]
def add_line(self, line: IRCLine.ParsedLine):
self._lines.append(line)
def get_lines(self) -> typing.List[IRCLine.ParsedLine]:
return self._lines
class IRCSendBatch(IRCBatch):
def __init__(self, batch_type: str, args: typing.List[str],
tags: typing.Dict[str, str]=None):
IRCBatch.__init__(self, str(uuid.uuid4()), batch_type, args, tags)
def get_lines(self) -> typing.List[IRCLine.ParsedLine]:
lines = []
for line in self._lines:
line.add_tag("batch", self.identifier)
lines.append(line)
lines.insert(0, IRCLine.ParsedLine("BATCH",
["+%s" % self.identifier, self.type]))
lines.append(IRCLine.ParsedLine("BATCH", ["-%s" % self.identifier]))
return lines
class Capability(object):
def __init__(self, name: typing.Optional[str], draft_name: str=None):
self._caps = set([name, draft_name])