EvaluationEfficiencyOptimizationWind/davisAPI/davisAPI.py

import gc
import logging
import time
from datetime import datetime, timedelta
from pprint import pprint
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, DateStamp  FROM weather_data ORDER BY DateStamp DESC LIMIT 1")
    result = cursor.fetchone()

    if result is None:
        return None, None, None, None

    sun_rise, sun_set, wind_dir, datestamp = result
    return sun_rise, sun_set, wind_dir, datestamp


def save_prediction_to_db(predictions):
    try:

        sun_rise, sun_set, wind_dir, datestamp = get_previous_values(cursor)

        fields = ['DateStamp', 'SunRise', 'SunSet', 'WindDir'] + list(predictions.keys())
        placeholders = ', '.join(['%s'] * len(fields))
        field_names = ', '.join(fields)

        values = [datestamp + timedelta(minutes = 1), sun_rise, sun_set, wind_dir] + list(predictions.values())
        pprint(dict(zip(fields, 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
while True:
    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(60)
    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)
        time.sleep(60)


#todo переписать под influx, для линухи приколы сделать