class Led():
def __init__(self, name):
#so we can have many different leds
self.name = name
self.led = BlinkM()
self.delay(2)
self.led.reset()
self.delay(1)
self.led.set_fade_speed(32)
self.delay(1)
def delay(self, seconds):
time.sleep(seconds)
def show_status_light(self):
self.led.fade_to_hex("0000ff")
self.delay(1)
def show_thinking_light(self):
#self.led.play_script(Scripts.WHITE_FLASH)
self.led.fade_to_hex("ffffff")
self.delay(1)
def show_positive_light(self):
#self.led.stop_script()
self.led.fade_to_hex("00ff00")
self.delay(1)
def show_negative_light(self):
#self.led.stop_script()
self.led.fade_to_hex("ff0000")
self.delay(1)
class AquaPi:
DEBUG = False
HOST = 'http://aqua.willandchi.com'
ENDPOINT_POLL = '/v1/poll'
ENDPOINT_EVENT = '/v1/events'
timeout = 5
led_fade_speed = 64
def __init__(self):
self.log('AquaPi initializing...')
GPIO.setwarnings(False)
GPIO.setmode(GPIO.BCM)
GPIO.setup(PIN_POWER, GPIO.OUT)
self.led = BlinkM()
self.led.reset()
self.led.set_fade_speed(self.led_fade_speed)
# self.led.write_script_line(Scripts.TRANSFER, 0, 10, 'c', 0xff, 0xff, 0xff)
# self.led.write_script_line(Scripts.TRANSFER, 1, 10, 'c', 0x00, 0x00, 0x00)
self.spi = spidev.SpiDev()
self.spi.open(0, SPI_ADC)
self.sio = serial.Serial('/dev/ttyAMA0', 9600, serial.EIGHTBITS, serial.PARITY_NONE, serial.STOPBITS_ONE, 1)
self.sensor_temp = TempSensor(self.spi, ADC_TEMP)
self.sensor_light = AnalogSensor(self.spi, ADC_LIGHT)
self.sensor_liquid = SerialSensor(self.sio)
self.metro_sensor_sample = Metro(500)
self.metro_sensor_send = Metro(30000)
self.metro_poll = Metro(6000)
self.metro_health = Metro(180000)
self.events = deque()
self.running = False
self.current_color = False
self.happy()
def loop(self):
self.log("Loop...")
if self.metro_health.check():
self.log("Health checks are failing. I'm sad :(")
self.sad()
self.event('error', 'health-check-failure')
if self.metro_sensor_sample.check():
self.log("Sampling sensors...")
self.sensor_temp.read()
self.sensor_light.read()
self.sensor_liquid.read()
if self.metro_sensor_send.check():
self.log("Creating sensor events...")
# Temperature Sensor
self.event('temp', round(self.sensor_temp.fahrenheit(), 2))
# Light Sensor
self.event('light', self.sensor_light.value())
# Ultrasonic Range Finder
self.event('liquid', round((URF_EMPTY - self.sensor_liquid.value()) / (URF_EMPTY - URF_FULL) * 100.0, 2))
if self.metro_poll.check():
self.poll()
self.send_events()
time.sleep(0.1)
def poll(self):
self.log('Polling...')
self.led_script(Scripts.TRANSFER)
self.current_color = colors.white
try:
r = requests.get(self.HOST + self.ENDPOINT_POLL,
auth=(credentials.username, credentials.password),
timeout=self.timeout)
self.log(r.text)
if r.status_code == 200:
data = r.json()
if len(data):
self.power(data['power'] == 'on')
self.metro_health.reset()
self.happy()
else:
self.log('Polling non-200 response')
self.event('error', 'poll-non-200')
self.sad()
except requests.RequestException:
self.log('Polling failed')
self.event('error', 'poll-exception')
self.sad()
def send_events(self):
if len(self.events) > 0:
self.log('Sending event...')
event = self.events.popleft()
self.log(json.dumps(event))
self.led_script(Scripts.TRANSFER)
self.current_color = colors.white
try:
r = requests.post(self.HOST + self.ENDPOINT_EVENT,
data=event,
auth=(credentials.username, credentials.password),
timeout=self.timeout)
if r.status_code == 200:
self.metro_health.reset()
self.happy()
else:
self.log('Sending event non-200 response')
self.event('error', 'event-non-200')
self.sad()
except requests.RequestException:
self.events.appendleft(event)
self.log('Sending event failed')
self.event('error', 'event-exception')
self.sad()
def power(self, on):
if on:
GPIO.output(PIN_POWER, True)
self.running = True
self.happy()
self.event('power', 'on')
else:
GPIO.output(PIN_POWER, False)
self.running = False
self.happy()
self.event('power', 'off')
def event(self, name, data):
self.events.append({
'event': name,
'data': data
})
def sad(self):
if self.current_color != colors.red:
self.led_script(Scripts.RED_FLASH)
self.current_color = colors.red
def happy(self):
if self.running:
if self.current_color != colors.blue:
self.led_script(Scripts.BLUE_FLASH)
self.current_color = colors.blue
else:
if self.current_color != colors.green:
self.led_color(colors.green)
self.current_color = colors.green
def led_script(self, script):
try:
self.led.reset()
self.led.play_script(script)
except IOError:
self.log('Playing led script failed')
self.event('led', 'error')
def led_color(self, color):
try:
self.led.reset()
self.led.go_to_hex(color)
except IOError:
self.log('Changing led color failed')
self.event('led', 'error')
def log(self, msg):
if self.DEBUG:
print "[{}] {}".format(time.strftime('%Y-%m-%d %H:%M:%S'), msg)