from sense_hat import SenseHat
from time import sleep
import time
sense = SenseHat()

def temp_h():
  th = sense.get_temperature_from_humidity()
  th = round(th, 1)
  th = str(th)
  return th

def temp_p():
  tp = sense.get_temperature_from_pressure()
  tp = round(tp, 1)
  tp = str(tp)
  return tp

def humidity():
  h = int(sense.get_humidity())
  h = str(h)

  return h

def date():
  d = time.strftime("%Y-%m-%d; %H:%M:%S")

  return d

def pressure():

  p = int(sense.get_pressure())
  p = str(p)

  return p

def compass():
  M = sense.get_compass_raw()
  X = M['x']
  Y = M['y']
  Z = M['z']

  X = round(X, 0)
  Y = round(Y, 0)
  Z = round(Z, 0)

  return "{}; {}; {}".format(X, Y, Z)
def acceleration():
  acc = sense.get_accelerometer_raw()
  x = round(int(acc['x']), 8)
  y = round(int(acc['y']), 8)
  z = round(int(acc['z']), 8)

  return "{}; {}; {}".format(x, y, z)

def gyroscope():
  gyro_only = sense.get_gyroscope()
  p = round(gyro_only['pitch'], 8)
  r = round(gyro_only['roll'], 8)
  y = round(gyro_only['yaw'], 8)

  return "{}; {}; {}".format(p,r,y)



_ = (0, 0 ,0)
g = (0, 50, 0)
r = (50, 0, 0)
w = (50, 50, 50)
def good_pixel():
  sense.set_pixel(0, 0, w)
  sense.set_pixel(7, 0, w)
  sense.set_pixel(0, 7, w)
  sense.set_pixel(7, 7, w)
def bad_pixel():
  sense.set_pixel(0, 0, r)
  sense.set_pixel(7, 0, r)
  sense.set_pixel(0, 7, r)
  sense.set_pixel(7, 7, r)
def stop_pixel():
  sense.set_pixel(0, 0, _)
  sense.set_pixel(7, 0, _)
  sense.set_pixel(0, 7, _)
  sense.set_pixel(7, 7, _)
def Temp_good():
  sense.set_pixel(0, 1, g)
  sense.set_pixel(1, 1, g)
  sense.set_pixel(2, 1, g)
  sense.set_pixel(1, 2, g)
  sense.set_pixel(1, 3, g)
  sense.set_pixel(1, 4, g)
def Temp_bad():
  sense.set_pixel(0, 1, r)
  sense.set_pixel(1, 1, r)
  sense.set_pixel(2, 1, r)
  sense.set_pixel(1, 2, r)
  sense.set_pixel(1, 3, r)
  sense.set_pixel(1, 4, r)
def Hum_good():
   sense.set_pixel(5, 1, g)
   sense.set_pixel(5, 2, g)
   sense.set_pixel(5, 3, g)
   sense.set_pixel(5, 4, g)
   sense.set_pixel(6, 2, g)
   sense.set_pixel(6, 3, g)
   sense.set_pixel(7, 1, g)
   sense.set_pixel(7, 2, g)
   sense.set_pixel(7, 3, g)
   sense.set_pixel(7, 4, g)
def Hum_bad():
   sense.set_pixel(5, 1, r)
   sense.set_pixel(5, 2, r)
   sense.set_pixel(5, 3, r)
   sense.set_pixel(5, 4, r)
   sense.set_pixel(6, 2, r)
   sense.set_pixel(6, 3, r)
   sense.set_pixel(7, 1, r)
   sense.set_pixel(7, 2, r)
   sense.set_pixel(7, 3, r)
   sense.set_pixel(7, 4, r)
def Press_good():
  sense.set_pixel(2, 5, g)
  sense.set_pixel(2, 6, g)
  sense.set_pixel(2, 7, g)
  sense.set_pixel(3, 4, g)
  sense.set_pixel(4, 5, g)
  sense.set_pixel(3, 6, g)
def Press_bad():
  sense.set_pixel(2, 5, r)
  sense.set_pixel(2, 6, r)
  sense.set_pixel(2, 7, r)
  sense.set_pixel(3, 4, r)
  sense.set_pixel(4, 5, r)
  sense.set_pixel(3, 6, r)

def flash_bad():
  bad_pixel()
  sleep(1)
  stop_pixel()
  sleep(1)

def flash_good():
  good_pixel()
  sleep(1)
  stop_pixel()
  sleep(1)

with open( "team_data.csv" , "w" ) as f:
  row = "Date; Time; Temperature_hum; Temperature_pres; Humidity; Pressure; Acceleration X; Acceleration Y; Acceleration Z; Gyroscope P; Gyroscope R; Gyroscope Y; Magnetometer X; Magnetometer Y; Magnetometer Z\r\n"
  f.write(row)

while True:
  h = sense.get_humidity()
  t = sense.get_temperature_from_humidity()
  p = sense.get_pressure()
  if t <= 25 and h <=65 and p <=1050:
   flash_good()
   Temp_good()
   Press_good()
   Hum_good()
  else:
    flash_bad()
  if h > 65:
    Hum_bad()
  if t > 25:
    Temp_bad()
  if p > 1050:
    Press_bad()

  else:
    flash_bad()
  if h < 55:
    Hum_bad()
  if t < 15:
    Temp_bad()
  if p < 990:
    Press_bad()



  row = "{}; {}; {}; {}; {}; {}; {}; {}\r\n".format(date(), temp_h(), temp_p(), humidity(), pressure(), acceleration(), gyroscope(), compass())
  with open( "team_data.csv" , "a" ) as f:
    f.write(row)
  sleep(5)