seewasser.py

   1 #! /usr/bin/python3
   2
   3 import argparse
   4 import logging
   5 import datetime
   6 import re
   7 import sys
   8 import csv
   9 import os
  10 import configparser
  11 from sqlalchemy import create_engine, exc
  12 import warnings
  13 import MySQLdb.cursors
  14
  15 #May 27 21:32 /sys/bus/w1/devices/28-0416a1bab9ff
  16 #May 27 21:33 /sys/bus/w1/devices/28-0416a1ac66ff
  17 #May 27 21:35 /sys/bus/w1/devices/28-0516a207a4ff
  18 #May 27 21:38 /sys/bus/w1/devices/28-0316a2193bff
  19 #May 27 21:38 /sys/bus/w1/devices/28-0316a21383ff
  20
  21 # Eine Funktion:
  22 # einen Sensor auslesen
  23 #/sys/bus/w1/devices/
  24
  25 class ReadsensorError(RuntimeError):
  26     pass
  27
  28
  29 def readsensor(sensor_id):
  30     sensorfile = "/sys/bus/w1/devices/28-{}/w1_slave".format(sensor_id)
  31     file = open(sensorfile)
  32
  33     # Inhalt des Sensors:
  34     # 64 01 4b 46 7f ff 0c 10 01 : crc=01 YES
  35     # 64 01 4b 46 7f ff 0c 10 01 t=22250
  36
  37     # Suche nach YES
  38     linecrc = file.readline()
  39     match = re.search(": crc=[0-9a-f]{2} (YES|NO)",linecrc)
  40
  41     yesno = match.group(1)
  42     if yesno != "YES":
  43         raise ReadsensorError('Could not evaluate sensorfile "{}"'.format(sensorfile))
  44
  45     # Temperatur raus
  46     linetemp = file.readline()
  47     match = re.search(" t=([-0-9]+)",linetemp)
  48
  49     temp_raw = match.group(1)
  50     temp = float(temp_raw)/1000
  51
  52     file.close()
  53     jetzt = datetime.datetime.now()
  54
  55     return jetzt, temp_raw, temp
  56
  57
  58 def writesensordatacsv(config, sensor_id, sensor_name, timestamp, value_type, value_raw, value):
  59     # Schreiben des csv-files
  60     file = open(os.path.expanduser(config.get("csv", "filename")), "a", newline = "")
  61     writer = csv.writer(file, dialect = "excel")
  62     writer.writerow([timestamp.strftime("%Y-%m-%d %H:%M"), sensor_id, sensor_name, "{:.1f}".format(value)])
  63
  64     file.close()
  65
  66
  67 def readcsvfile(csvfile):
  68     file = open(csvfile, "r", newline="")
  69     reader = csv.DictReader(file, dialect = "excel", fieldnames=("timestamp", "sensor_id", "sensor_name", "value"))
  70     records = []
  71     for row in reader:
  72         records.append(row)
  73
  74     file.close()
  75     return records
  76
  77
  78 def writesensordatadb(config, sensor_id, sensor_name, timestamp, value_type, value_raw, value):
  79     # Schreiben des db-files
  80     user = config.get('database', 'user')
  81     pwd = config.get('database','password')
  82     host = config.get('database','hostname')
  83     db = config.get('database','database')
  84
  85     engine = create_engine('mysql+mysqldb://{}:{}@{}/{}'.format(user, pwd, host, db), echo=False)
  86     conn = engine.connect()
  87     with warnings.catch_warnings():
  88         # ignore _mysql_exceptions.IntegrityError: (1062, "Duplicate entry '0316a2193bff-2018-08-29 14:30:00' for key 'sensorid_timestamp'")
  89         # TODO: the following doesn't work
  90         warnings.simplefilter("ignore", category=MySQLdb.cursors.Warning)
  91         # TODO: this does but …?!
  92         try:
  93             conn.execute("insert into sensors (sensor_id, sensor_name, value_type, value_raw, value, timestamp) values (%s,%s,%s,%s,%s,%s)", sensor_id, sensor_name, value_type, value_raw, value, timestamp)
  94         except exc.IntegrityError as e:
  95             if "1062" in str(e):
  96                 pass
  97             else:
  98                 raise
  99     conn.close()
 100
 101
 102 def main(configfile, fromcsvfile):
 103     config = configparser.ConfigParser()
 104     config.read(configfile)
 105
 106     value_type = "Wassertemperatur"
 107     for sensor_id, sensor_name in config.items('temperature'):
 108         if fromcsvfile:
 109             # "timestamp", "sensor_id", "sensor_name", "value"
 110             for record in readcsvfile(fromcsvfile):
 111                 timestamp, sensor_id, sensor_name, value = (record["timestamp"], record["sensor_id"], record["sensor_name"], float(record["value"]))
 112                 value_raw = None
 113                 writesensordatadb(config, sensor_id, sensor_name, timestamp, value_type, value_raw, value)
 114         else:
 115             timestamp, value_raw, value = readsensor(sensor_id)
 116             writesensordatacsv(config, sensor_id, sensor_name, timestamp, value_type, value_raw, value)
 117             writesensordatadb(config, sensor_id, sensor_name, timestamp, value_type, value_raw, value)
 118         logging.info('Sensor {}: {:.1f}°C'.format(sensor_id, value))
 119
 120
 121 if __name__ == '__main__':
 122     default_config_file = os.path.expanduser('~/seewasser.ini')
 123     parser = argparse.ArgumentParser(description='Read sensor data')
 124     parser.add_argument('--config', default=default_config_file, help='configuration file')
 125     parser.add_argument('--fromcsvfile', help='write values from csv file to database')
 126     args = parser.parse_args()
 127     main(args.config, args.fromcsvfile)
 128