def _messagetests(self, d): ''' Run tests on a message to see if a specific one has been passed. If so, mark the test passed. :param bytes d: a data packet from the queue ''' if 'TERM' in str(d): printM('Got TERM message...', sender=self.sender) t.TEST['c_TERM'][1] = True self.alive = False elif 'ALARM' in str(d): printM('Got ALARM message with time %s' % (helpers.fsec(helpers.get_msg_time(d))), sender=self.sender) t.TEST['c_ALARM'][1] = True elif 'RESET' in str(d): printM('Got RESET message with time %s' % (helpers.fsec(helpers.get_msg_time(d))), sender=self.sender) t.TEST['c_RESET'][1] = True elif 'IMGPATH' in str(d): printM('Got IMGPATH message with time %s' % (helpers.fsec(helpers.get_msg_time(d))), sender=self.sender) printM('and path %s' % (helpers.get_msg_path(d)), sender=self.sender) t.TEST['c_IMGPATH'][1] = True
def _when_alarm(self, d): ''' Send a telegram in an alert scenario. :param bytes d: queue message ''' event_time = helpers.fsec(helpers.get_msg_time(d)) self.last_event_str = '%s' % (event_time.strftime(self.fmt)[:22]) message = '%s %s UTC - %s' % (self.message0, self.last_event_str, self.livelink) response = None try: printM('Sending alert...', sender=self.sender) response = self.telegram.sendMessage(chat_id=self.chat_id, text=message) printM('Sent Telegram: %s' % (message), sender=self.sender) except Exception as e: printE('Could not send alert - %s' % (e)) try: printE('Waiting 5 seconds and trying to send again...', sender=self.sender, spaces=True) time.sleep(5) self.auth() response = self.telegram.sendMessage(chat_id=self.chat_id, text=message) printM('Sent Telegram: %s' % (message), sender=self.sender) except Exception as e: printE('Could not send alert - %s' % (e)) response = None
def _when_alarm(self, d): ''' Send a tweet when you get an ``ALARM`` message. :param bytes d: queue message ''' event_time = helpers.fsec(helpers.get_msg_time(d)) self.last_event_str = '%s' % (event_time.strftime(self.fmt)[:22]) message = '%s %s UTC%s - %s' % (self.message0, self.last_event_str, self.extra_text, self.livelink) response = None try: printM('Tweet: %s' % (message), sender=self.sender) if not self.testing: response = self.twitter.update_status( status=message, lat=rs.inv[0][0].latitude, long=rs.inv[0][0].longitude, geo_enabled=True, display_coordinates=True) # location will only stick to tweets on accounts that have location enabled in Settings url = 'https://twitter.com/%s/status/%s' % ( response['user']['screen_name'], response['id_str']) printM('Tweet URL: %s' % url) if self.testing: TEST['c_tweet'][1] = True except Exception as e: printE('could not send alert tweet - %s' % (e)) try: printE('Waiting 5 seconds and trying to send tweet again...', sender=self.sender, spaces=True) time.sleep(5.1) printM('Tweet: %s' % (message), sender=self.sender) if not self.testing: self.auth() response = self.twitter.update_status( status=message, lat=rs.inv[0][0].latitude, long=rs.inv[0][0].longitude, geo_enabled=True, display_coordinates=True) # location will only stick to tweets on accounts that have location enabled in Settings url = 'https://twitter.com/%s/status/%s' % ( response['user']['screen_name'], response['id_str']) printM('Tweet URL: %s' % url) except Exception as e: printE('could not send alert tweet - %s' % (e)) response = None self.last_message = message
def getq(self): ''' Get data from the queue and test for whether it has certain strings. ALARM and TERM both trigger specific behavior. ALARM messages cause the event counter to increment, and if :py:data:`screencap==True` then aplot image will be saved when the event is :py:data:`self.save_pct` of the way across the plot. ''' d = self.queue.get() self.queue.task_done() if 'TERM' in str(d): plt.close() if 'SELF' in str(d): printM('Plot has been closed, plot thread will exit.', self.sender) self.alive = False rs.producer = False elif 'ALARM' in str(d): self.events += 1 # add event to count self.save_timer -= 1 # don't push the save time forward if there are a large number of alarm events event = [ self.save_timer + int(self.save_pct * self.pkts_in_period), helpers.fsec(helpers.get_msg_time(d)) ] # event = [save after count, datetime] self.last_event_str = '%s UTC' % ( event[1].strftime('%Y-%m-%d %H:%M:%S.%f')[:22]) printM('Event time: %s' % (self.last_event_str), sender=self.sender) # show event time in the logs if self.screencap: printM( 'Saving png in about %i seconds' % (self.save_pct * (self.seconds)), self.sender) self.save.append(event) # append self.fig.suptitle( '%s.%s live output - detected events: %s' # title % (self.net, self.stn, self.events), fontsize=14, color=self.fgcolor, x=0.52) self.fig.canvas.set_window_title( '(%s) %s.%s - Raspberry Shake Monitor' % (self.events, self.net, self.stn)) if rs.getCHN(d) in self.chans: self.raw = rs.update_stream(stream=self.raw, d=d, fill_value='latest') return True else: return False
def _when_img(self, d): ''' Send a tweet with an image in when you get an ``IMGPATH`` message. :param bytes d: queue message ''' if self.tweet_images: imgpath = helpers.get_msg_path(d) imgtime = helpers.fsec(helpers.get_msg_time(d)) message = '%s %s UTC' % (self.message1, imgtime.strftime(self.fmt)[:22]) response = None if os.path.exists(imgpath): with open(imgpath, 'rb') as image: try: printM('Uploading image to Twitter %s' % (imgpath), self.sender) response = self.twitter.upload_media(media=image) time.sleep(5.1) printM('Sending tweet...', sender=self.sender) response = self.twitter.update_status(status=message, media_ids=response['media_id'], lat=rs.inv[0][0].latitude, long=rs.inv[0][0].longitude, geo_enabled=True, display_coordinates=True) # location will only stick to tweets on accounts that have location enabled in Settings printM('Tweeted with image: %s' % (message), sender=self.sender) url = 'https://twitter.com/%s/status/%s' % (response['user']['screen_name'], response['id_str']) printM('Tweet URL: %s' % url) except Exception as e: printE('could not send multimedia tweet - %s' % (e)) try: printM('Waiting 5 seconds and trying to send tweet again...', sender=self.sender) time.sleep(5.1) self.auth() printM('Uploading image to Twitter (2nd try) %s' % (imgpath), self.sender) response = self.twitter.upload_media(media=image) time.sleep(5.1) printM('Sending tweet...', sender=self.sender) response = self.twitter.update_status(status=message, media_ids=response['media_id'], lat=rs.inv[0][0].latitude, long=rs.inv[0][0].longitude, geo_enabled=True, display_coordinates=True) # location will only stick to tweets on accounts that have location enabled in Settings printM('Tweeted with image: %s' % (message), sender=self.sender) url = 'https://twitter.com/%s/status/%s' % (response['user']['screen_name'], response['id_str']) printM('Tweet URL: %s' % url) except Exception as e: printE('could not send multimedia tweet (2nd try) - %s' % (e)) response = None else: printM('Could not find image: %s' % (imgpath), sender=self.sender)
def _when_alarm(self, d): ''' Send a telegram in an alert scenario. :param bytes d: queue message ''' event_time = helpers.fsec(helpers.get_msg_time(d)) self.last_event_str = '%s' % (event_time.strftime(self.fmt)[:22]) message = '%s %s UTC%s - %s' % (self.message0, self.last_event_str, self.extra_text, self.livelink) response = None try: printM('Sending alert...', sender=self.sender) printM('Telegram message: %s' % (message), sender=self.sender) if not self.testing: response = self.telegram.sendMessage(chat_id=self.chat_id, text=message) else: TEST['c_telegram'][1] = True except Exception as e: printE('Could not send alert - %s' % (e)) try: printE('Waiting 5 seconds and trying to send again...', sender=self.sender, spaces=True) time.sleep(5) self.auth() printM('Telegram message: %s' % (message), sender=self.sender) if not self.testing: response = self.telegram.sendMessage(chat_id=self.chat_id, text=message) else: # if you are here in testing mode, there is a problem TEST['c_telegram'][1] = False except Exception as e: printE('Could not send alert - %s' % (e)) response = None self.last_message = message