125 lines
3.7 KiB
Python
125 lines
3.7 KiB
Python
import gc
|
|
import logging
|
|
import time
|
|
from datetime import datetime
|
|
|
|
import mariadb
|
|
import serial.tools.list_ports
|
|
|
|
from PyWeather.weather.stations.davis import VantagePro
|
|
from prediction import run_prediction_module
|
|
|
|
logging.basicConfig(filename="Stations.log",
|
|
format='%(asctime)s %(message)s',
|
|
filemode='a')
|
|
logger = logging.getLogger('davis_api')
|
|
logger.setLevel(logging.DEBUG)
|
|
|
|
console_handler = logging.StreamHandler()
|
|
console_handler.setLevel(logging.DEBUG)
|
|
console_handler.setFormatter(logging.Formatter('%(asctime)s %(message)s'))
|
|
logger.addHandler(console_handler)
|
|
|
|
|
|
def write_data(device, station, send=True):
|
|
try:
|
|
# device.parse()
|
|
data = device.fields
|
|
logger.info(data)
|
|
if len(data) < 1:
|
|
return
|
|
else:
|
|
logger.info(data)
|
|
fields = ['BarTrend', 'CRC', 'DateStamp', 'DewPoint', 'HeatIndex', 'ETDay', 'HeatIndex',
|
|
'HumIn', 'HumOut', 'Pressure', 'RainDay', 'RainMonth', 'RainRate', 'RainStorm',
|
|
'RainYear', 'SunRise', 'SunSet', 'TempIn', 'TempOut', 'WindDir', 'WindSpeed',
|
|
'WindSpeed10Min']
|
|
|
|
if send:
|
|
placeholders = ', '.join(['%s'] * len(fields))
|
|
field_names = ', '.join(fields)
|
|
sql = f"INSERT INTO weather_data ({field_names}) VALUES ({placeholders})"
|
|
values = [data[field] for field in fields]
|
|
cursor.execute(sql, values)
|
|
conn.commit()
|
|
else:
|
|
logger.info(data)
|
|
|
|
del data
|
|
del fields
|
|
gc.collect()
|
|
except Exception as e:
|
|
logger.error(str(e))
|
|
raise e
|
|
|
|
|
|
def get_previous_values(cursor):
|
|
cursor.execute("SELECT SunRise, SunSet, WindDir FROM weather_data ORDER BY DateStamp DESC LIMIT 1")
|
|
result = cursor.fetchone()
|
|
|
|
if result is None:
|
|
return None, None, None
|
|
|
|
sun_rise, sun_set, wind_dir = result
|
|
return sun_rise, sun_set, wind_dir
|
|
|
|
|
|
def save_prediction_to_db(predictions):
|
|
try:
|
|
current_timestamp = datetime.now()
|
|
|
|
sun_rise, sun_set, wind_dir = get_previous_values(cursor)
|
|
|
|
fields = ['DateStamp', 'SunRise', 'SunSet', 'WindDir'] + list(predictions.keys())
|
|
placeholders = ', '.join(['%s'] * len(fields))
|
|
field_names = ', '.join(fields)
|
|
|
|
values = [current_timestamp, sun_rise, sun_set, wind_dir] + list(predictions.values())
|
|
|
|
sql = f"INSERT INTO weather_data ({field_names}) VALUES ({placeholders})"
|
|
# cursor.execute(sql, values)
|
|
# conn.commit()
|
|
logger.info("Save prediction to db success!")
|
|
except Exception as e:
|
|
logger.error(str(e))
|
|
raise e
|
|
|
|
|
|
try:
|
|
conn = mariadb.connect(
|
|
user="wind",
|
|
password="wind",
|
|
host="193.124.203.110",
|
|
port=3306,
|
|
database="wind_towers"
|
|
)
|
|
cursor = conn.cursor()
|
|
except mariadb.Error as e:
|
|
logger.error('DB_ERR: ' + str(e))
|
|
raise e
|
|
|
|
try:
|
|
ports = serial.tools.list_ports.comports()
|
|
available_ports = {}
|
|
|
|
for port in ports:
|
|
if port.serial_number == '0001':
|
|
available_ports[port.name] = port.vid
|
|
|
|
devices = [VantagePro(port) for port in available_ports.keys()]
|
|
|
|
while True:
|
|
for i in range(1):
|
|
if len(devices) != 0:
|
|
logger.info(devices)
|
|
# write_data(devices[i], 'st' + str(available_ports[list(available_ports.keys())[i]]), True)
|
|
else:
|
|
raise Exception("Can't connect to device")
|
|
time.sleep(1)
|
|
except Exception as e:
|
|
logger.error('Device_error: ' + str(e))
|
|
predictions = run_prediction_module()
|
|
logger.info(predictions)
|
|
if predictions is not None:
|
|
save_prediction_to_db(predictions)
|