alertringer.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import json
import time
import rpi_ws281x
import redis
import sys
import signal
import logging


# LED strip configuration:
LED_COUNT      = 24      # Number of LED pixels per ring.
LED_RINGS      = 5       # Number of LED rings
LED_PIN        = 18      # GPIO pin connected to the pixels (must support PWM!).
LED_FREQ_HZ    = 800000  # LED signal frequency in hertz (usually 800khz)
LED_DMA        = 10      # DMA channel to use for generating signal (try 5)
LED_BRIGHTNESS = 30      # Set to 0 for darkest and 255 for brightest
LED_INVERT     = False   # True to invert the signal (when using NPN transistor level shift)
LED_WAIT       = 0.01    # wait for rolling in SEC
LED_FLASH      = 1       # wait for flasing in SEC

leds = (LED_COUNT * LED_RINGS)
red        = rpi_ws281x.Color(255,0,0)
green      = rpi_ws281x.Color(0,255,0)
blue       = rpi_ws281x.Color(0,0,255)

yellow     = rpi_ws281x.Color(255,255,0)
black      = rpi_ws281x.Color(0,0,0)
halfgreen  = rpi_ws281x.Color(0,150,0)
halfyellow = rpi_ws281x.Color(150,150,0)


logfile = '/var/log/alertringer.log'
debug = False

logger = logging.getLogger("ringer")
logger.setLevel(logging.INFO)
formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
handler = logging.FileHandler(logfile)
handler.setFormatter(formatter)
logger.addHandler(handler)


def signal_handler(signal, frame):
  logger.info('shutting down')
  for led in range(leds):
    strip.setPixelColor(led, black)
  strip.show()
  exit()


def flashing(strip, chart, wait):
  """ flashing, heartbeat has stopped """
  #reset the things b4 
  for led in range(leds):
    strip.setPixelColor(led, black)
  strip.show()
  time.sleep(wait)
  for led in range(leds):
    strip.setPixelColor(led, yellow)
  strip.show()
  time.sleep(wait)


def rolling(strip, chart, wait):
  """ attempt for rolling x number of stuff """
  global paint
  #reset the things b4 
  for led in range(leds):
    strip.setPixelColor(led, black)

  # set the correct color
  for ring in range(LED_RINGS):
    if chart[ring]['red'] == 0 and chart[ring]['yellow'] == 0:
      for greenled in range(LED_COUNT):
        strip.setPixelColor((ring * LED_COUNT + (int(paint) - greenled + 1) % LED_COUNT), halfgreen)
    else:  
      # ringid * ledar/ring + ledid
      for redled in range(chart[ring]['red']):
        strip.setPixelColor((ring * LED_COUNT + (int(paint) - (redled + 1)) % LED_COUNT), red)
      for yellowled in range(chart[ring]['yellow']):
        strip.setPixelColor((ring * LED_COUNT + (int(paint) - (yellowled + 1) - (redled + 1)) % LED_COUNT), yellow)
  strip.show()
  time.sleep(wait)
  paint = (paint + 1) % LED_COUNT


# Main program logic follows:
if __name__ == '__main__':
  logger.info('starting up')
  paint = 0
  signal.signal(signal.SIGTERM, signal_handler)
  r = redis.StrictRedis(host='localhost', port=6379, db=0)  
  # Create NeoPixel object with appropriate configuration.
  strip = rpi_ws281x.Adafruit_NeoPixel((LED_COUNT * LED_RINGS), LED_PIN, LED_FREQ_HZ, LED_DMA, LED_INVERT, LED_BRIGHTNESS)
  # Intialize the library (must be called once before other functions)
  strip.begin()
  while True:
    teams = {'access': 0, 'core': 0, 'powerpatrol': 0, 'services': 0, 'wifi': 0, 'heartbeat': 0}
    for key in r.scan_iter():
      val = r.get(key)
      if val is None:
        if debug:
          print(key)
      else:
        team = json.loads(val.decode('utf-8'))['team'] 
        teams[team] = teams[team] +1

    chart = {0: {'yellow': 0, 'red': teams['services']},
             1: {'yellow': 0, 'red': teams['core']},
             2: {'yellow': 0, 'red': teams['access']},
             3: {'yellow': 0, 'red': teams['wifi']},
             4: {'yellow': 0, 'red': teams['powerpatrol']}}

    if debug:
      print('heartbeat {}'.format(teams['heartbeat']))
      print('services {}'.format(teams['services']))
      print('core {}'.format(teams['core']))
      print('access {}'.format(teams['access']))
      print('wifi {}'.format(teams['wifi']))
      print('powerpatrol {}'.format(teams['powerpatrol']))
    if teams['heartbeat'] == 0:
      flashing(strip, chart, LED_FLASH) # Paint all the stuff
    else:
      rolling(strip, chart, LED_WAIT) # Paint all the stuff