added a (very) basic tcpircbot client class

eevans · eevans · commit 4c83310406dd · 2016-10-20T19:31:56.000-05:00
diff --git a/cassandra/tools/__init__.py b/cassandra/tools/__init__.py
@@ -5,4 +5,5 @@
 
 from .instances import get_instances
 
+
 __all__ = ["get_instances"]
diff --git a/cassandra/tools/tcpircbot.py b/cassandra/tools/tcpircbot.py
@@ -0,0 +1,29 @@
+
+
+import logging
+import socket
+
+
+# Cribbed from: https://phabricator.wikimedia.org/diffusion/MSCA/browse/master/scap/log.py;d407eaf
+class IrcBot(object):
+    def __init__(self, host, port, timeout=1.0):
+        self.host = host
+        self.port = port
+        self.timeout = timeout
+
+    def log(self, msg, *args, **kwargs):
+        sock = None
+        try:
+            sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+            sock.settimeout(self.timeout)
+            sock.connect((self.host, self.port))
+            sock.sendall(IrcBot.format_message(msg, *args, **kwargs))
+        except (socket.timeout, socket.error, socket.gaierror), err:
+            logging.error("Unable to send to logmsgbot (SAL): %s", err.message)
+        finally:
+            if sock:
+                sock.close()
+
+    @classmethod
+    def format_message(cls, msg, *args, **kwargs):
+        return "!log " + msg.format(*args, **kwargs)

Original file line number	Diff line number	Diff line change
`@@ -5,4 +5,5 @@`
`5`	`5`
`6`	`6`	`from .instances import get_instances`
`7`	`7`
	`8`	`+`
`8`	`9`	`__all__ = ["get_instances"]`