def __init__(self, token, mac):
super(SlackMonitorWorker, self).__init__()
self.stoprequest = threading.Event()
self.token = token
self.mac = mac
self.slack_json = self.slack_auth()
self.userid = self.slack_json['self']['id']
#print self.slack_json
self.ws = websocket.create_connection(self.slack_json['url'])
self.alert_q = Queue.Queue()
self.cw = ColorificWorker(self.alert_q, mac)
self.cw.start()
def main():
notification_q = Queue.Queue()
cw = ColorificWorker(notification_q)
cw.start()
cw.alert_red()
time.sleep(5)
cw.color_shift(cw.BLUE)
time.sleep(5)
cw.alert_red()
time.sleep(20)
cw.join()
class SlackMonitorWorker(threading.Thread):
def __init__(self, token, mac):
super(SlackMonitorWorker, self).__init__()
self.stoprequest = threading.Event()
self.token = token
self.mac = mac
self.slack_json = self.slack_auth()
self.userid = self.slack_json['self']['id']
#print self.slack_json
self.ws = websocket.create_connection(self.slack_json['url'])
self.alert_q = Queue.Queue()
self.cw = ColorificWorker(self.alert_q, mac)
self.cw.start()
def run(self):
while not self.stoprequest.isSet():
try:
opcode, data = self.recv()
except websocket.WebSocketConnectionClosedException:
try:
self.close()
except:
pass
return
msg = None
if opcode in OPCODE_DATA:
msg = CleanMsgUnicode(data)
self.log_msg(msg)
if msg:
#print(msg)
try:
msgj = json.loads(msg)
if msgj['type'] == 'message':
if 'text' in msgj and self.userid in str(msgj['text']):
self.alert_q.put('alert_mention')
self.log_msg('ALERT_MENTION')
if 'text' in msgj and 'user' in msgj and CHANNELALERT in str(msgj['text']) and self.userid not in str(msgj['user']):
self.alert_q.put('alert_channel')
self.log_msg('ALERT_CHANNEL')
if 'channel' in msgj and 'user' in msgj and self.is_channel_im(str(msgj['channel'])) and self.userid not in str(msgj['user']):
self.alert_q.put('alert_im')
self.log_msg('ALERT_IM')
except UnicodeDecodeError:
print('Unicode Decode Error while searching msg for userid.')
print('userid: {0}'.format(self.userid))
print('msg: {0}'.format(msg))
continue
try:
if CHANNELALERT in msg:
self.alert_q.put('alert_red')
except UnicodeDecodeError:
print('Unicode Decode Error while searching msg for CHANNELALERT.')
continue
def close(self, timeout=None):
self.log_msg('SERVICE CLOSED')
self.ws.close()
self.cw.close()
self.stoprequest.set()
super(SlackMonitorWorker, self).join(timeout)
def slack_auth(self):
url = 'https://slack.com/api/rtm.start?'
params = {'pretty':'1','token':self.token}
urlparams = urllib.urlencode(params)
response = urllib2.urlopen(url+urlparams)
data = response.read()
startrsp = data.decode('utf-8')
return json.loads(startrsp)
def is_channel_im(self, channelid):
for im in self.slack_json['ims']:
if channelid == im['id']: return True
return False
def log_msg(self, msg):
try:
with open('slackmsg.log','a') as f:
curtime = strftime("%Y%m%d-%H:%M:%S - ", time.localtime())
f.write(curtime + msg + '\n')
except:
pass
def recv(self):
try:
frame = self.ws.recv_frame()
if not frame:
raise websocket.WebSocketException("Not a valid frame {0}".format(frame))
elif frame.opcode in OPCODE_DATA:
return (frame.opcode, frame.data)
elif frame.opcode == websocket.ABNF.OPCODE_CLOSE:
self.ws.send_close()
return (frame.opcode, None)
elif frame.opcode == websocket.ABNF.OPCODE_PING:
self.ws.pong("Hi!")
except websocket.SSLError:
self.log_msg('ERROR: SSLError while attempting to receive frame.')
return None, None
