From 114d1054a5d50ac12171b7eae2592b3717b833b6 Mon Sep 17 00:00:00 2001
From: jesopo
Date: Tue, 19 Feb 2019 15:41:17 +0000
Subject: Add `.message(` and `.notice(` to IRCSendBatch

---
 src/utils/irc/__init__.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'src/utils')

diff --git a/src/utils/irc/__init__.py b/src/utils/irc/__init__.py
index 85410e5d..691414f5 100644
--- a/src/utils/irc/__init__.py
+++ b/src/utils/irc/__init__.py
@@ -326,7 +326,13 @@ class IRCBatch(object):
 class IRCRecvBatch(IRCBatch):
     pass
 class IRCSendBatch(IRCBatch):
-    pass
+    def _add_line(self, line: IRCParsedLine):
+        line.tags["batch"] = self.id
+        self.lines.append(line)
+    def message(self, target: str, message: str, tags: dict=None):
+        self._add_line(utils.irc.protocol.message(target, message, tags))
+    def notice(self, target: str, message: str, tags: dict=None):
+        self._add_line(utls.irc.protocol.notice(target, message, tags))
 
 def trailing(s: str) -> str:
     if s[0] == ":" or " " in s:
-- 
cgit v1.3.1-10-gc9f91