X-Git-Url: https://zdv2.bktei.com/gitweb/EVA-2020-02-2.git/blobdiff_plain/4f8716dc13aa14bf586dd78002d6bf4d331bc656..0cf10e5bb8354458b6eac7d04d4aa6235bedcf73:/library/envirophatplus/gas.py?ds=sidebyside

diff --git a/library/envirophatplus/gas.py b/library/envirophatplus/gas.py
index 9b213ed..4f61940 100644
--- a/library/envirophatplus/gas.py
+++ b/library/envirophatplus/gas.py
@@ -1,9 +1,10 @@
 """Read the MICS6812 via an ads1015 ADC"""
 
+import atexit
 import ads1015
 import RPi.GPIO as GPIO
 
-MICS6812_EN_PIN = 24
+MICS6812_HEATER_PIN = 24
 
 
 ads1015.I2C_ADDRESS_DEFAULT = ads1015.I2C_ADDRESS_ALTERNATE
@@ -40,11 +41,18 @@ def setup():
 
     GPIO.setwarnings(False)
     GPIO.setmode(GPIO.BCM)
-    GPIO.setup(MICS6812_EN_PIN, GPIO.OUT)
-    GPIO.output(MICS6812_EN_PIN, 1)
+    GPIO.setup(MICS6812_HEATER_PIN, GPIO.OUT)
+    GPIO.output(MICS6812_HEATER_PIN, 1)
+    atexit.register(cleanup)
+
+
+def cleanup():
+    GPIO.output(MICS6812_HEATER_PIN, 0)
 
 
 def read_all():
+    """Return gas resistence for oxidising, reducing and NH3"""
+    setup()
     ox = adc.get_voltage('in0/gnd')
     red = adc.get_voltage('in1/gnd')
     nh3 = adc.get_voltage('in2/gnd')