merge web/owm.py into master

[chrisu/seepark.git] / seewasser.py

diff --git a/seewasser.py b/seewasser.py
index 40d2455dc0da2bfcef63d410fba0fe3f04954672..ef7f401eeb8a29ec2ae869066879777cfc7bb770 100755 (executable)
--- a/seewasser.py
+++ b/seewasser.py
@@ -1,5 +1,6 @@
 #! /usr/bin/python3
 
+import argparse
 import logging
 import datetime
 import re
@@ -54,9 +55,9 @@ def readsensor(sensor_id):
 
 def writesensordatacsv(config, sensor_id, sensor_name, timestamp, value_type, value_raw, value):
     # Schreiben des csv-files
-    file = open(config.get("csv", "filename"), "a", newline = "")
+    file = open(os.path.expanduser(config.get("csv", "filename")), "a", newline = "")
     writer = csv.writer(file, dialect = "excel")
-    writer.writerow([timestamp.strftime("%Y-%m-%d %H:%M"), sensor_id, "{:.1f}".format(value)])
+    writer.writerow([timestamp.strftime("%Y-%m-%d %H:%M"), sensor_id, sensor_name, "{:.1f}".format(value)])
 
     file.close()
 
@@ -74,9 +75,9 @@ def writesensordatadb(config, sensor_id, sensor_name, timestamp, value_type, val
     conn.close()
 
 
-def main():
+def main(configfile):
     config = configparser.ConfigParser()
-    config.read(os.path.expanduser('~/seewasser.ini'))
+    config.read(configfile)
     
     value_type = "Wassertemperatur"
     for sensor_id, sensor_name in config.items('temperature'):
@@ -86,4 +87,10 @@ def main():
         logging.info('Sensor {}: {:.1f}°C'.format(sensor_id, value))
 
 
-main()
+if __name__ == '__main__':
+    default_config_file = os.path.expanduser('~/seewasser.ini')
+    parser = argparse.ArgumentParser(description='Read sensor data')
+    parser.add_argument('--config', default=default_config_file, help='configuration file')
+    args = parser.parse_args()
+    main(args.config)
+