Optimize imports.

[chrisu/seepark.git] / seewasser.py

diff --git a/seewasser.py b/seewasser.py
index be5f149ea49e8afd6660e5113756ca1c91b4b616..7b58c0531fbfb23ef74c50806ecf2e335594e353 100755 (executable)
--- a/seewasser.py
+++ b/seewasser.py
@@ -3,6 +3,7 @@
 import argparse
 import logging
 import datetime
 import argparse
 import logging
 import datetime

+import time

 import re
 import sys
 import csv
 import re
 import sys
 import csv


@@ -12,6 +13,8 @@ from sqlalchemy import create_engine
 import warnings
 import MySQLdb.cursors
 
 import warnings
 import MySQLdb.cursors
 

+logging.basicConfig(format='%(levelname)s: %(message)s', level=logging.INFO)
+

 #May 27 21:32 /sys/bus/w1/devices/28-0416a1bab9ff
 #May 27 21:33 /sys/bus/w1/devices/28-0416a1ac66ff
 #May 27 21:35 /sys/bus/w1/devices/28-0516a207a4ff
 #May 27 21:32 /sys/bus/w1/devices/28-0416a1bab9ff
 #May 27 21:33 /sys/bus/w1/devices/28-0416a1ac66ff
 #May 27 21:35 /sys/bus/w1/devices/28-0516a207a4ff


@@ -54,6 +57,13 @@ def readsensor(sensor_id):
 
     return jetzt, temp_raw, temp
 
 
     return jetzt, temp_raw, temp
 

+def read_buggy_sensor(sensor_id):
+    for i in range(120):
+        jetzt, temp_raw, temp = readsensor(sensor_id)
+        if temp < 30:
+            return jetzt, temp_raw, temp
+        time.sleep(1)
+    raise ReadsensorError('Zu hoch')

 
 def writesensordatacsv(config, sensor_id, sensor_name, timestamp, value_type, value_raw, value):
     # Schreiben des csv-files
 
 def writesensordatacsv(config, sensor_id, sensor_name, timestamp, value_type, value_raw, value):
     # Schreiben des csv-files


@@ -79,6 +89,8 @@ def writesensordatadb(config, sensor_id, sensor_name, timestamp, value_type, val
     with warnings.catch_warnings():
         # ignore _mysql_exceptions.Warning: Duplicate entry '0316a2193bff-2018-08-29 14:30:00' for key 'sensorid_timestamp'
         warnings.simplefilter("ignore", category=MySQLdb.cursors.Warning)
     with warnings.catch_warnings():
         # ignore _mysql_exceptions.Warning: Duplicate entry '0316a2193bff-2018-08-29 14:30:00' for key 'sensorid_timestamp'
         warnings.simplefilter("ignore", category=MySQLdb.cursors.Warning)

+        if args.verbose:
+            logging.info("Writing to database: sensor_id={}, sensor_name={}, value_type={}, value_raw={}, value={}, timestamp={}".format(sensor_id, sensor_name, value_type, value_raw, value, timestamp))

         conn.execute("insert ignore 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)
     conn.close()
 
         conn.execute("insert ignore 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)
     conn.close()
 


@@ -104,10 +116,11 @@ def main(configfile, fromcsvfile):
         return
 
     for sensor_id, sensor_name in sensors:
         return
 
     for sensor_id, sensor_name in sensors:

-        timestamp, value_raw, value = readsensor(sensor_id)
+        timestamp, value_raw, value = read_buggy_sensor(sensor_id)

         writesensordatacsv(config, sensor_id, sensor_name, timestamp, value_type, value_raw, value)
         writesensordatadb(config, sensor_id, sensor_name, timestamp, value_type, value_raw, value)
         writesensordatacsv(config, sensor_id, sensor_name, timestamp, value_type, value_raw, value)
         writesensordatadb(config, sensor_id, sensor_name, timestamp, value_type, value_raw, value)

-        logging.info('Sensor {}: {:.1f}°C'.format(sensor_id, value))
+        if args.verbose:
+            logging.info('Sensor {}: {:.1f}°C'.format(sensor_id, value))

 
 
 if __name__ == '__main__':
 
 
 if __name__ == '__main__':


@@ -115,6 +128,7 @@ if __name__ == '__main__':
     parser = argparse.ArgumentParser(description='Read sensor data')
     parser.add_argument('--config', default=default_config_file, help='configuration file')
     parser.add_argument('--fromcsvfile', help='write values from csv file to database')
     parser = argparse.ArgumentParser(description='Read sensor data')
     parser.add_argument('--config', default=default_config_file, help='configuration file')
     parser.add_argument('--fromcsvfile', help='write values from csv file to database')

+    parser.add_argument("--verbose", help='write some output', action='store_true')

     args = parser.parse_args()
     main(args.config, args.fromcsvfile)
 
     args = parser.parse_args()
     main(args.config, args.fromcsvfile)