From f9f637e084b54f1b800db66499d44f7ec30d604b Mon Sep 17 00:00:00 2001
From: jesopo
Date: Sun, 2 Jun 2019 10:13:51 +0100
Subject: Remove IRCSendBatch, fix some batch-related type hints

---
 src/utils/irc/__init__.py | 16 +++++-----------
 1 file changed, 5 insertions(+), 11 deletions(-)

(limited to 'src/utils/irc')

diff --git a/src/utils/irc/__init__.py b/src/utils/irc/__init__.py
index 3be9b279..67eb6b7e 100644
--- a/src/utils/irc/__init__.py
+++ b/src/utils/irc/__init__.py
@@ -263,17 +263,11 @@ class IRCBatch(object):
         self.id = identifier
         self.type = batch_type
         self.tags = tags
-        self.lines = [] # type: typing.List[IRCLine.ParsedLine]
-class IRCRecvBatch(IRCBatch):
-    pass
-class IRCSendBatch(IRCBatch):
-    def _add_line(self, line: IRCLine.ParsedLine):
-        line.tags["batch"] = self.id
-        self.lines.append(line)
-    def message(self, target: str, message: str, tags: dict={}):
-        self._add_line(utils.irc.protocol.message(target, message, tags))
-    def notice(self, target: str, message: str, tags: dict={}):
-        self._add_line(utils.irc.protocol.notice(target, message, tags))
+        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 Capability(object):
     def __init__(self, name, draft_name=None):
-- 
cgit v1.3.1-10-gc9f91