439 lines
16 KiB
Python
439 lines
16 KiB
Python
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
"""Cameteo module
|
|
|
|
Data classes for GPS, BME280 sensors and some system stuff
|
|
Function to handle data files rotation and set internal RTC date from other
|
|
(more reliable) sources.
|
|
"""
|
|
|
|
import os
|
|
import time
|
|
import rtc
|
|
import microcontroller
|
|
import board
|
|
from busio import I2C, UART
|
|
from analogio import AnalogIn
|
|
from digitalio import DigitalInOut, Direction
|
|
from adafruit_bme280 import Adafruit_BME280_I2C
|
|
from adafruit_gps import GPS
|
|
|
|
TIME_FORMAT = "{:04}/{:02}/{:02}_{:02}:{:02}:{:02}" # Date/time format
|
|
|
|
# Set the pin N°13 to use the onboard LED as read-only/read-write indicator
|
|
# (RED = read-only, no data recorded)
|
|
LED13 = DigitalInOut(board.D13)
|
|
LED13.direction = Direction.OUTPUT
|
|
LED13.value = False
|
|
|
|
###########
|
|
# Classes #
|
|
###########
|
|
|
|
class Data:
|
|
"""Class for handling data"""
|
|
|
|
def __init__(self,
|
|
name,
|
|
update_interval=10,
|
|
write_interval=300,
|
|
send_interval=60,
|
|
debug=False):
|
|
|
|
self.name = name
|
|
self.data = {}
|
|
self.debug = debug
|
|
self.update_interval = update_interval
|
|
self.write_interval = write_interval
|
|
self.send_interval = send_interval
|
|
self._last_update = 0
|
|
self._last_backup = 0
|
|
self._last_send = 0
|
|
self._last_file_rotation = time.localtime()
|
|
|
|
# Check if data directories exists if we need to use them
|
|
if self.write_interval >= 0:
|
|
try:
|
|
if 'data' not in os.listdir():
|
|
os.mkdir('data')
|
|
os.mkdir('data/hourly')
|
|
os.mkdir('data/daily')
|
|
elif 'hourly' not in os.listdir('data'):
|
|
os.mkdir('data/hourly')
|
|
elif 'daily' not in os.listdir('data'):
|
|
os.mkdir('data/daily')
|
|
except OSError as err:
|
|
print("Err {}: readonly".format(err))
|
|
self.write_interval = -1 # to avoid trying again till next reset
|
|
# Turn onboard led on to indicate read-only error
|
|
LED13.value = True
|
|
|
|
def __str__(self):
|
|
"""Serialize data for visualization on serial console"""
|
|
output = self.name + ":\n"
|
|
for item in self.data.items():
|
|
output += "\t{0}: {1}\n".format(*item)
|
|
return output
|
|
|
|
def __repr__(self):
|
|
return self.json
|
|
|
|
def update(self, current, verbose=True):
|
|
"""Read the data from sensors and update the data dict variable"""
|
|
if self.debug:
|
|
print("Update {} : Current : {} | Last : {} | Interval : {}".format(
|
|
self.name,
|
|
current,
|
|
self._last_update,
|
|
self.update_interval))
|
|
if current - self._last_update >= self.update_interval:
|
|
self._last_update = current
|
|
self._update()
|
|
if verbose:
|
|
print(self)
|
|
|
|
def _update(self):
|
|
"""Here comes specific update code"""
|
|
pass
|
|
|
|
@property
|
|
def json(self):
|
|
"""Serialized data to compact json-formatted string"""
|
|
output = '{"' + self.name + '":{'
|
|
first_data = True
|
|
for name, value in self.data.items():
|
|
if first_data:
|
|
comma = ""
|
|
first_data = False
|
|
else:
|
|
comma = ","
|
|
output = '{}{}"{}":"{}"'.format(output, comma, name, value)
|
|
output = output + '}}'
|
|
return output
|
|
|
|
def csv(self, header=False):
|
|
"""Serialized data values (or keys if header=True) as CSV line"""
|
|
if header:
|
|
return ";".join([str(x) for x in self.data.keys()])
|
|
|
|
return ";".join([str(x) for x in self.data.values()])
|
|
|
|
def write_on_flash(self, current, verbose=True):
|
|
"""Save the current data as csv file on SPI flash"""
|
|
if self.write_interval >= 0:
|
|
if self.debug:
|
|
print("Write {} : Current : {} | Last : {} | Interval : {}".format(
|
|
self.name,
|
|
current,
|
|
self._last_backup,
|
|
self.write_interval))
|
|
if current - self._last_backup >= self.write_interval:
|
|
self._last_backup = current
|
|
#First check if a rotation is needed
|
|
self._file_rotation()
|
|
|
|
file_name = "{}.csv".format(self.name)
|
|
file_path = "data"
|
|
try:
|
|
#Check if the file exists. If not, creates it with CSV header
|
|
if file_name not in os.listdir(file_path):
|
|
with open("/".join((file_path, file_name)), "w") as csv_file:
|
|
csv_file.write(self.csv(header=True))
|
|
csv_file.write("\n")
|
|
if verbose:
|
|
print("File created : {}".format(file_name))
|
|
|
|
with open("/".join((file_path, file_name)), "a") as csv_file:
|
|
csv_file.write(self.csv())
|
|
csv_file.write("\n")
|
|
if verbose:
|
|
print("Written data : \n{}".format(self.csv()))
|
|
|
|
except OSError as err:
|
|
print("Err {}: readonly".format(err))
|
|
# to avoid trying again till next reset
|
|
self.write_interval = -1
|
|
# Turn onboard led on to indicate read-only filesystem
|
|
LED13.value = True
|
|
|
|
def send_json(self, current, uart, verbose=True):
|
|
"""Send JSON string over UART"""
|
|
if self.debug:
|
|
print("Send {} : Current : {} | Last : {} | Interval : {}".format(
|
|
self.name,
|
|
current,
|
|
self._last_send,
|
|
self.send_interval))
|
|
if current - self._last_send >= self.send_interval:
|
|
self._last_send = current
|
|
uart.write(self.json)
|
|
uart.write("\n")
|
|
if verbose:
|
|
print(self.json)
|
|
#print("\n")
|
|
|
|
def _file_rotation(self):
|
|
"""Check if files need to rotate
|
|
=> every new hour
|
|
=> every new day
|
|
"""
|
|
current_time = time.localtime()
|
|
if self.debug:
|
|
print(self._last_file_rotation[3])
|
|
print(current_time[3])
|
|
|
|
# If the hour changed : copy current csv to hourly directory
|
|
if current_time[3] != self._last_file_rotation[3]:
|
|
print("Time to move hourly data !")
|
|
os.rename("data/{}.csv".format(self.name),
|
|
"data/hourly/{}_{:02}.csv".format(
|
|
self.name,
|
|
self._last_file_rotation[3]))
|
|
|
|
# If the day changed : copy content of hourly to daily directories
|
|
if current_time[2] != self._last_file_rotation[2]:
|
|
print("Time to move daily data !")
|
|
# Create new dir for the date of "yesterday"
|
|
newdir = "{}{:02}{:02}".format(*self._last_file_rotation[0:3])
|
|
if newdir not in os.listdir("data/daily/"):
|
|
os.mkdir("data/daily/" + newdir)
|
|
# Move each "hourly file" to the new directory
|
|
for file in os.listdir('data/hourly'):
|
|
# Move only file which name is beginning with self.name
|
|
if file.find(self.name) == 0:
|
|
print("Move {} to {}".format(file, newdir))
|
|
os.rename("data/hourly/{}".format(file), "{}/{}".format(newdir, file))
|
|
|
|
#Finally : remember last rotation check
|
|
self._last_file_rotation = current_time
|
|
|
|
return
|
|
|
|
class SysData(Data):
|
|
"""Subclass for Feather board data"""
|
|
|
|
def __init__(self, name="SYS", update_interval=10, write_interval=300,
|
|
send_interval=60, debug=False):
|
|
|
|
Data.__init__(self, name=name,
|
|
update_interval=update_interval,
|
|
write_interval=write_interval,
|
|
send_interval=send_interval,
|
|
debug=debug)
|
|
|
|
self.data = {'time': "2000/01/01_00:00:00",
|
|
'vbat': int(),
|
|
'cput': float()}
|
|
|
|
# Battery voltage
|
|
self.vbat = AnalogIn(board.VOLTAGE_MONITOR)
|
|
self._rtc = rtc.RTC()
|
|
|
|
def _update(self):
|
|
""" Update data from Feather board"""
|
|
self.data['time'] = TIME_FORMAT.format(*self._rtc.datetime[0:6])
|
|
self.data['cput'] = round(microcontroller.cpu.temperature, 2)
|
|
|
|
# Note about v_bat calculation :
|
|
# 0.000100708 = 2*3.3/65536 with
|
|
# 2 : voltage is divided by 2
|
|
# 3.3 : Vref = 3.3V
|
|
# 65536 : 16bit ADC
|
|
val = 0
|
|
samples = 10
|
|
for sample in range(samples):
|
|
val += self.vbat.value
|
|
time.sleep(0.01)
|
|
self.data['vbat'] = round(val/samples*0.000100708, 3)
|
|
|
|
|
|
class BME280Data(Data):
|
|
"""Subclass for BME280 sensor"""
|
|
|
|
def __init__(self, name="BME", update_interval=10, write_interval=300,
|
|
send_interval=60, debug=False):
|
|
|
|
Data.__init__(self, name=name,
|
|
update_interval=update_interval,
|
|
write_interval=write_interval,
|
|
send_interval=send_interval,
|
|
debug=debug)
|
|
|
|
self.data = {'time' : "2000/01/01_00:00:00",
|
|
'temp': float(),
|
|
'hum': int(),
|
|
'press': float()}
|
|
# BME280 sensors (I2C)
|
|
# i2c addresses for BME280 breakout :
|
|
# 0x77 = adafruit breakout board
|
|
# 0x76 = tiny cheap chinese board
|
|
self.bme280 = Adafruit_BME280_I2C(I2C(board.SCL, board.SDA),
|
|
address=0x76)
|
|
|
|
def _update(self):
|
|
"""Update data from BME280"""
|
|
self.data['time'] = TIME_FORMAT.format(*time.localtime()[0:6])
|
|
self.data['temp'] = round(self.bme280.temperature, 1)
|
|
self.data['hum'] = int(self.bme280.humidity)
|
|
self.data['press'] = round(self.bme280.pressure, 2)
|
|
|
|
def rgb(self, neopixel_max=70):
|
|
"""Convert atmospheric data from BME280 sensor into NeoPixel color as
|
|
a tuple (RED, BLUE, GREEN):
|
|
* RED => temperature : max = 35degC, min =10degC (range 25°C)
|
|
* BLUE => humidity : max= 100%, mini=0%
|
|
* GREEN => pression : mini=960hPa, maxi = 1030hPa (range 70hPa)
|
|
"""
|
|
# RED componant calculation from temperature data
|
|
# 10 is the min temperature, 25 is the range
|
|
# (10+25=35°C = max temperature)
|
|
red = int((self.data['temp']-10)*neopixel_max/25)
|
|
if red > neopixel_max:
|
|
red = neopixel_max
|
|
if red < 0:
|
|
red = 0
|
|
|
|
# BLUE componant calculation: very simple! By definition relative
|
|
# humidity cannot be more than 100 or less than 0, physically
|
|
blue = int(self.data['hum']*neopixel_max/100)
|
|
|
|
# GREEN component calculation : 960 is the minimum pressure and 70 is
|
|
# the range (960+70 = 1030hPa = max pressure)
|
|
green = int((self.data['press']-960)*neopixel_max/70)
|
|
if green > neopixel_max:
|
|
green = neopixel_max
|
|
if green < 0:
|
|
green = 0
|
|
|
|
if self.debug:
|
|
print("Col:{}".format((red, green, blue)))
|
|
|
|
return (red, green, blue)
|
|
|
|
|
|
class GPSData(Data):
|
|
"""Sub class for GPS"""
|
|
|
|
def __init__(self, name="GPS",
|
|
update_interval=0, write_interval=60, send_interval=60,
|
|
enable=True, enable_pin=DigitalInOut(board.A5),
|
|
debug=False):
|
|
|
|
Data.__init__(self, name=name,
|
|
update_interval=update_interval,
|
|
write_interval=write_interval,
|
|
send_interval=send_interval,
|
|
debug=debug)
|
|
|
|
self.data = {'time': "2000/01/01_00:00:00",
|
|
'lat': float(),
|
|
'lon': float(),
|
|
'alt': float(),
|
|
'qual': int(),
|
|
'age': int()}
|
|
self._enable = enable
|
|
self._gps_last_fix = int()
|
|
self._gps_current_fix = int()
|
|
|
|
# Set the pin to control the power to GPS module
|
|
self._gps_en_pin = enable_pin
|
|
self._gps_en_pin.direction = Direction.OUTPUT
|
|
|
|
self._gps = GPS(UART(board.TX, board.RX, baudrate=9600, timeout=3000)) #, debug=self.debug)
|
|
|
|
if self._enable:
|
|
self.enable()
|
|
else:
|
|
self.disable()
|
|
|
|
def _update(self):
|
|
self._gps.update()
|
|
|
|
if self._enable:
|
|
self._gps_current_fix = int(time.monotonic())
|
|
if self._gps.timestamp_utc is not None:
|
|
self.data['time'] = TIME_FORMAT.format(*self._gps.timestamp_utc[0:6])
|
|
else:
|
|
self.data['time'] = TIME_FORMAT.format(2018, 9, 10, 10, 0, 0)
|
|
if self._gps.has_fix:
|
|
self._gps_last_fix = self._gps_current_fix
|
|
self.data['lat'] = self._gps.latitude
|
|
self.data['lon'] = self._gps.longitude
|
|
self.data['alt'] = self._gps.altitude_m
|
|
self.data['qual'] = self._gps.fix_quality
|
|
self.data['age'] = 0
|
|
else:
|
|
self.data['age'] = int(self._gps_current_fix - self._gps_last_fix)
|
|
else:
|
|
self.data = {}
|
|
|
|
def enable(self):
|
|
"""Enable GPS module"""
|
|
# Set GPS module on FeatherWing board
|
|
|
|
if not self._enable:
|
|
self._enable = True
|
|
self._gps_en_pin.value = not self._enable # Set enable pin high to disable GPS module
|
|
time.sleep(0.1)
|
|
|
|
# Turn on the basic GGA and RMC info
|
|
self._gps.send_command('PMTK314,0,1,0,1,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0')
|
|
self._gps.send_command('PMTK220,1000') # 1000 ms refresh rate
|
|
|
|
def disable(self):
|
|
"""Disable GPS module"""
|
|
if self._enable:
|
|
self._enable = False
|
|
self._gps_en_pin.value = not self._enable # Set enable pin high to disable GPS module
|
|
|
|
@property
|
|
def datetime(self):
|
|
"""Wrapper to feed rtc.set_time_source()"""
|
|
return self._gps.timestamp_utc
|
|
|
|
|
|
#############
|
|
# Functions #
|
|
#############
|
|
|
|
def set_clock_from_localtime(clock, threshold=2.0, debug=False):
|
|
"""
|
|
Compare internal RTC and date-time from time.localtime() and set
|
|
the RTC date/time to this value if the difference is more or equal to
|
|
threshold (in seconds).
|
|
The source of time.localtime() can be set with rtc.set_time_source()
|
|
"""
|
|
|
|
if debug:
|
|
print("Clock date/time : " + TIME_FORMAT.format(*clock.datetime[0:6]))
|
|
print("Local date/time : " + TIME_FORMAT.format(*time.localtime()[0:6]))
|
|
|
|
# Check if time.localtime() is valid and can be converted in seconds
|
|
try:
|
|
local_seconds = time.mktime(time.localtime())
|
|
except OverflowError as err:
|
|
print(err)
|
|
print("Local time source : " + TIME_FORMAT.format(*time.localtime()[0:6]))
|
|
print("Wait some time to have proper time from time source...")
|
|
return False
|
|
|
|
# Max difference between GPS and internal RTC (in seconds):
|
|
if abs(local_seconds - time.mktime(clock.datetime)) >= threshold:
|
|
print("Clock difference with GPS!")
|
|
print("Previous date/time : " + TIME_FORMAT.format(*clock.datetime[0:6]))
|
|
clock.datetime = time.localtime() # Trust localtime if there is a bias
|
|
print("Clocks synced !")
|
|
|
|
return True
|