Add initial event listener

salt/utils/event.py

@@ -0,0 +1,62 @@
+'''
+Manage event listeners.
+'''
+
+# Import Python libs
+import os
+
+# Import Third Party libs
+import zmq
+
+
+class SaltEvent(object):
+    '''
+    The base class used to manage salt events
+    '''
+    def __init__(self, opts, node):
+        self.opts = opts
+        self.poller = zmq.Poller()
+        self.cpub = False
+        if node == 'master':
+            self.puburi = os.path.join(
+                    self.opts['sock_dir'],
+                    'master_event_pub.ipc'
+                    )
+            self.pulluri = os.path.join(
+                    self.opts['sock_dir'],
+                    'master_event_pull.ipc'
+                    )
+        else:
+            self.puburi = os.path.join(
+                    self.opts['sock_dir'],
+                    'minion_event_pub.ipc'
+                    )
+            self.pulluri = os.path.join(
+                    self.opts['sock_dir'],
+                    'minion_event_pull.ipc'
+                    )
+
+    def connect_pub(self):
+        '''
+        Establish the publish connection
+        '''
+        self.context = zmq.context()
+        self.sub = context.socket(zmq.SUB)
+        self.sub.connect(self.puburi)
+        self.poller.register(self.pub, zmq.POLLIN)
+        self.cpub = True
+
+    def get_event(self, wait=5, tag=''):
+        '''
+        Get a single publication
+        '''
+        if not self.cpub:
+            self.connect_pub()
+        self.sub.setsockopt(zmq.SUBSCRIBE, tag)
+        start = time.time()
+        while True:
+            socks = dict(self.poller.poll())
+            if self.pub in socks and socks[self.pub] == zmq.POLLIN:
+                return self.sub.recv()
+            if (time.time() - start) > wait:
+                return None