add comment with URL of web colors
[chrisu/seepark.git] / 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
12
13 #May 27 21:32 /sys/bus/w1/devices/28-0416a1bab9ff
14 #May 27 21:33 /sys/bus/w1/devices/28-0416a1ac66ff
15 #May 27 21:35 /sys/bus/w1/devices/28-0516a207a4ff
16 #May 27 21:38 /sys/bus/w1/devices/28-0316a2193bff
17 #May 27 21:38 /sys/bus/w1/devices/28-0316a21383ff
18
19 # Eine Funktion:
20 # einen Sensor auslesen
21 #/sys/bus/w1/devices/
22
23 class ReadsensorError(RuntimeError):
24     pass
25     
26     
27 def readsensor(sensor_id):
28     sensorfile = "/sys/bus/w1/devices/28-{}/w1_slave".format(sensor_id)
29     file = open(sensorfile)
30
31     # Inhalt des Sensors:
32     # 64 01 4b 46 7f ff 0c 10 01 : crc=01 YES
33     # 64 01 4b 46 7f ff 0c 10 01 t=22250
34
35     # Suche nach YES
36     linecrc = file.readline()
37     match = re.search(": crc=[0-9a-f]{2} (YES|NO)",linecrc)
38
39     yesno = match.group(1)
40     if yesno != "YES":
41         raise ReadsensorError('Could not evaluate sensorfile "{}"'.format(sensorfile))
42         
43     # Temperatur raus
44     linetemp = file.readline()
45     match = re.search(" t=([-0-9]+)",linetemp)
46
47     temp_raw = match.group(1)
48     temp = float(temp_raw)/1000
49
50     file.close()
51     jetzt = datetime.datetime.now()
52
53     return jetzt, temp_raw, temp
54
55
56 def writesensordatacsv(config, sensor_id, sensor_name, timestamp, value_type, value_raw, value):
57     # Schreiben des csv-files
58     file = open(os.path.expanduser(config.get("csv", "filename")), "a", newline = "")
59     writer = csv.writer(file, dialect = "excel")
60     writer.writerow([timestamp.strftime("%Y-%m-%d %H:%M"), sensor_id, sensor_name, "{:.1f}".format(value)])
61
62     file.close()
63
64
65 def writesensordatadb(config, sensor_id, sensor_name, timestamp, value_type, value_raw, value):
66     # Schreiben des db-files
67     user = config.get('database', 'user')
68     pwd = config.get('database','password')
69     host = config.get('database','hostname')
70     db = config.get('database','database')
71     
72     engine = create_engine('mysql+mysqldb://{}:{}@{}/{}'.format(user, pwd, host, db), echo=False)
73     conn = engine.connect()
74     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)
75     conn.close()
76
77
78 def main(configfile):
79     config = configparser.ConfigParser()
80     config.read(configfile)
81     
82     value_type = "Wassertemperatur"
83     for sensor_id, sensor_name in config.items('temperature'):
84         timestamp, value_raw, value = readsensor(sensor_id)
85         writesensordatacsv(config, sensor_id, sensor_name, timestamp, value_type, value_raw, value)
86         writesensordatadb(config, sensor_id, sensor_name, timestamp, value_type, value_raw, value)
87         logging.info('Sensor {}: {:.1f}°C'.format(sensor_id, value))
88
89
90 if __name__ == '__main__':
91     default_config_file = os.path.expanduser('~/seewasser.ini')
92     parser = argparse.ArgumentParser(description='Read sensor data')
93     parser.add_argument('--config', default=default_config_file, help='configuration file')
94     args = parser.parse_args()
95     main(args.config)
96