agp8x
/
weatherstation
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

reformatted using pyCharm

master
agp8x 2015-08-30 18:22:55 +02:00
parent bb840d46cf
commit a508fb7269
8 changed files with 450 additions and 433 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -4,6 +4,7 @@ arch/
locks/ locks/
logs/ logs/
records/ records/
.idea/
bin/ bin/
bin/* bin/*
*~ *~

27
Logger.py
View File

@ -1,8 +1,6 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import time
import sys
import traceback import traceback
import xml.etree.ElementTree as ET import xml.etree.ElementTree as ET
import os import os
@ -11,6 +9,7 @@ from timeFunctions import *
from config import SensorType, setup_data_echo, setup_data_log from config import SensorType, setup_data_echo, setup_data_log
from config import settings from config import settings
class Logger(object): class Logger(object):
def __init__(self, log): def __init__(self, log):
self.temp_prev_default = settings.prev_temps_default self.temp_prev_default = settings.prev_temps_default
@ -33,17 +32,15 @@ class Logger(object):
def temp_rise(self, new, sensor): def temp_rise(self, new, sensor):
old = self.prev_temps[sensor] old = self.prev_temps[sensor]
if(old == self.temp_prev_default): if old == self.temp_prev_default:
return True return True
if(((old-new) > self.temp_max_diff) or ((new-old) > self.temp_max_diff)): if ((old - new) > self.temp_max_diff) or ((new - old) > self.temp_max_diff):
self.log.error('error checking ' + sensor + ';prev(' + str(old) + ');cur(' + str(new) + ');') self.log.error('error checking ' + sensor + ';prev(' + str(old) + ');cur(' + str(new) + ');')
return False return False
else: else:
return True return True
########################################## # common function to write value to file
# common function to write value to file #
##########################################
def write_value(self, value, sensor): def write_value(self, value, sensor):
if settings.legacy_record: if settings.legacy_record:
valuename = self.records + "/" + sensor + "_" + preptime() valuename = self.records + "/" + sensor + "_" + preptime()
@ -52,15 +49,13 @@ class Logger(object):
valuelog.close() valuelog.close()
self.datalog.info('%s;%s;%s', value, int(time.time()), sensor) self.datalog.info('%s;%s;%s', value, int(time.time()), sensor)
########################################## # generic callback
# generic callback #
##########################################
def cb_generic(self, value, sensor, type, supress=False): def cb_generic(self, value, sensor, type, supress=False):
if type == SensorType.temp: if type == SensorType.temp:
if self.temp_rise(value, sensor): if self.temp_rise(value, sensor):
self.write_value(value, sensor) self.write_value(value, sensor)
self.prev_temps[sensor] = value self.prev_temps[sensor] = value
elif (type == SensorType.none): elif type == SensorType.none:
return return
else: else:
self.write_value(value, sensor) self.write_value(value, sensor)
@ -68,17 +63,15 @@ class Logger(object):
if not supress: if not supress:
self.dataecho.info(sensor + ': ' + str(value / unit[1]) + ' ' + unit[2]) self.dataecho.info(sensor + ': ' + str(value / unit[1]) + ' ' + unit[2])
########################################### # exception logging
# exception logging # def print_exception(self, inst):
###########################################
def printException(self, inst):
tree = ET.parse(os.path.join(settings.log, settings.exceptionlog)) tree = ET.parse(os.path.join(settings.log, settings.exceptionlog))
root = tree.getroot() root = tree.getroot()
new = ET.Element('exception', {'class':str( type(inst) ).split("'")[1], 'date':str( time.ctime() ), 'time':str( int(time.time()) ), 'type':str(inst)}) new = ET.Element('exception', {'class': str(type(inst)).split("'")[1], 'date': str(time.ctime()),
'time': str(int(time.time())), 'type': str(inst)})
new.text = traceback.format_exc() new.text = traceback.format_exc()
root.append(new) root.append(new)
tree.write(settings.exceptionlog) tree.write(settings.exceptionlog)
message = 'an Exception happend @' + time.ctime() + "\n" message = 'an Exception happend @' + time.ctime() + "\n"
self.log.error(message) self.log.error(message)
print(message) print(message)

1
README.md
View File

@ -49,3 +49,4 @@ Optional: Upload der Aufzeichnungen mit SFTP an einem Server (beinhaltet Archivi
# TODOS # TODOS
* TODOS ausdenken * TODOS ausdenken
* rugged approach [http://www.tinkerforge.com/en/doc/Tutorials/Tutorial_Rugged/Tutorial.html](http://www.tinkerforge.com/en/doc/Tutorials/Tutorial_Rugged/Tutorial.html)

39
Setup.py
View File

@ -18,31 +18,34 @@ except ImportError:
from config import SensorType from config import SensorType
from config import settings from config import settings
class ConnectionSetup(object): class ConnectionSetup(object):
def __init__(self, log): def __init__(self, log):
self.log = log self.log = log
def setupConnection(self, host): def setup_connection(self, host):
ipcon = IPConnection() ipcon = IPConnection()
try:
ipcon.connect(host['name'], host['port']) ipcon.connect(host['name'], host['port'])
return (ipcon) self.log.info("connection to '%s:%s' established", host['name'], host['port'])
except ConnectionError:
self.log.error("connection to '%s:%s' failed", host['name'], host['port'])
return ipcon
def setupConnectionAndSensors(self, host, sensors, cb_generic): def setup_connection_and_sensors(self, host, sensors, cb_generic):
hostname = host['name'] ipcon = self.setup_connection(host)
port = host['port'] sensor_setup = SensorSetup(ipcon, sensors, cb_generic, self.log)
ipcon = IPConnection() connected_sensors = sensor_setup.setup_sensors()
ipcon.connect(hostname, port) return ipcon, connected_sensors
sensorSetup = SensorSetup(ipcon, sensors, cb_generic, self.log)
connectedSensors = sensorSetup.setupSensors()
return (ipcon, connectedSensors)
def disconnectAny(self, connections): def disconnect_any(self, connections):
for connection in connections: for connection in connections:
if not connection.get_connection_state() is IPConnection.CONNECTION_STATE_DISCONNECTED: if not connection.get_connection_state() is IPConnection.CONNECTION_STATE_DISCONNECTED:
self.log.debug("disconnecting (%s)", connection)
connection.disconnect() connection.disconnect()
class SensorSetup(object):
class SensorSetup(object):
def __init__(self, connection, sensors, cb_generic, log): def __init__(self, connection, sensors, cb_generic, log):
self.connection = connection self.connection = connection
self.sensors = sensors self.sensors = sensors
@ -94,11 +97,10 @@ class SensorSetup(object):
} }
} }
def parametrizedCallback(self, name, type): def parametrized_callback(self, name, type):
return partial(self.cb_generic, sensor=name, type=type) return partial(self.cb_generic, sensor=name, type=type)
def __setupSensor__(self, callback, id, cbtime, var): def __setupSensor__(self, callback, id, cbtime, var):
obj = None
if id in self._previous_sensors: if id in self._previous_sensors:
self.log.debug("reusing instance for %s", id) self.log.debug("reusing instance for %s", id)
obj = self._previous_sensors[id] # restore instance for another callback obj = self._previous_sensors[id] # restore instance for another callback
@ -111,9 +113,9 @@ class SensorSetup(object):
obj.register_callback(var['cb'], callback) # register callback obj.register_callback(var['cb'], callback) # register callback
return obj return obj
def genericSensorSetup(self, name, sensor): def generic_sensor_setup(self, name, sensor):
status = "setup device " + sensor[0] + " (" + name + "):" status = "setup device " + sensor[0] + " (" + name + "):"
callback = self.parametrizedCallback(name, type=sensor[1]) callback = self.parametrized_callback(name, type=sensor[1])
cbtime = settings.sensor_properties[sensor[1]][0] cbtime = settings.sensor_properties[sensor[1]][0]
obj = None obj = None
if sensor[1] is SensorType.rain: if sensor[1] is SensorType.rain:
@ -129,11 +131,10 @@ class SensorSetup(object):
self.log.error("%s FAIL:: %s (%s)", status, e, traceback.format_exc()) self.log.error("%s FAIL:: %s (%s)", status, e, traceback.format_exc())
return obj return obj
def setupSensors(self): def setup_sensors(self):
connected = [] connected = []
for name in self.sensors: for name in self.sensors:
sensor = self.sensors[name] sensor = self.sensors[name]
obj = self.genericSensorSetup(name, sensor) obj = self.generic_sensor_setup(name, sensor)
connected.append(obj) connected.append(obj)
return connected return connected

16
config.py
View File

@ -5,6 +5,7 @@ import json
import logging import logging
import os import os
class SensorType: class SensorType:
none = 0 none = 0
temp = 1 # temperature bricklet temp = 1 # temperature bricklet
@ -15,6 +16,7 @@ class SensorType:
iram = 6 # temperature ir bricklet, ambient iram = 6 # temperature ir bricklet, ambient
irob = 7 # temperature ir bricklet, object irob = 7 # temperature ir bricklet, object
DEFAULTS = { DEFAULTS = {
"hosts": { "hosts": {
"HOSTDESC_i": { "HOSTDESC_i": {
@ -51,15 +53,21 @@ DEFAULTS ={
"dataecho": "info", "dataecho": "info",
"legacy_record": True "legacy_record": True
} }
class Settings(dict): class Settings(dict):
def __getattr__(self, name): def __getattr__(self, name):
if name in self: if name in self:
return self[name] return self[name]
return DEFAULTS[name] return DEFAULTS[name]
def __setattr__(self, name, value): def __setattr__(self, name, value):
self[name] = value self[name] = value
def __delattr__(self, name): def __delattr__(self, name):
del self[name] del self[name]
def load_json(filename="config.json"): def load_json(filename="config.json"):
values = json.load(open(filename), object_hook=Settings) values = json.load(open(filename), object_hook=Settings)
sensors_name = "hosts" sensors_name = "hosts"
@ -75,7 +83,9 @@ def load_json(filename="config.json"):
new_units[eval(unit)] = values[sensor_props][unit] new_units[eval(unit)] = values[sensor_props][unit]
values[sensor_props] = new_units values[sensor_props] = new_units
return values return values
def setupLogger():
def setup_logger():
level = getattr(logging, settings.loglevel.upper(), logging.INFO) level = getattr(logging, settings.loglevel.upper(), logging.INFO)
log = logging.getLogger("weatherstation") log = logging.getLogger("weatherstation")
log.setLevel(level) log.setLevel(level)
@ -89,6 +99,7 @@ def setupLogger():
log.addHandler(fh) log.addHandler(fh)
return log return log
def setup_data_log(): def setup_data_log():
level = getattr(logging, settings.datalog.upper(), logging.INFO) level = getattr(logging, settings.datalog.upper(), logging.INFO)
log = logging.getLogger("weatherstation.datalog") log = logging.getLogger("weatherstation.datalog")
@ -100,6 +111,7 @@ def setup_data_log():
log.propagate = False log.propagate = False
return log return log
def setup_data_echo(): def setup_data_echo():
level = getattr(logging, settings.dataecho.upper(), logging.INFO) level = getattr(logging, settings.dataecho.upper(), logging.INFO)
log = logging.getLogger("weatherstation.data") log = logging.getLogger("weatherstation.data")
@ -111,6 +123,7 @@ def setup_data_echo():
log.propagate = False log.propagate = False
return log return log
settings = load_json() settings = load_json()
""" """
0: { 0: {
@ -129,4 +142,3 @@ settings=load_json()
"humi2": ["9V5", SensorType.humi], "humi2": ["9V5", SensorType.humi],
} }
},""" },"""

33
main.py
View File

@ -6,11 +6,12 @@ import time
from Logger import Logger from Logger import Logger
from Setup import ConnectionSetup from Setup import ConnectionSetup
from config import settings, setupLogger from config import settings, setup_logger
log = setupLogger() log = setup_logger()
lockpath = os.path.join(settings.locks, settings.lockname) lockpath = os.path.join(settings.locks, settings.lockname)
def check_dirs_and_files(): def check_dirs_and_files():
# log # log
if not os.path.exists(settings.logs): if not os.path.exists(settings.logs):
@ -26,7 +27,8 @@ def check_dirs_and_files():
if not os.path.exists(settings.records): if not os.path.exists(settings.records):
os.mkdir(settings.records, 0o000755) os.mkdir(settings.records, 0o000755)
def obtainLock():
def obtain_lock():
if not os.path.exists(lockpath): if not os.path.exists(lockpath):
lock = open(lockpath, 'w') lock = open(lockpath, 'w')
lock.write(str(time.time())) lock.write(str(time.time()))
@ -34,13 +36,16 @@ def obtainLock():
return True return True
return False return False
def freeLock():
def free_lock():
if os.path.exists(lockpath): if os.path.exists(lockpath):
os.remove(lockpath) os.remove(lockpath)
def formatHost(host):
def format_host(host):
return "%s:%d" % (host['name'], host['port']) return "%s:%d" % (host['name'], host['port'])
if __name__ == "__main__": if __name__ == "__main__":
try: try:
input = raw_input input = raw_input
@ -53,30 +58,30 @@ if __name__ == "__main__":
try: try:
log.info("setting up all sensors") log.info("setting up all sensors")
while True: while True:
if obtainLock(): if obtain_lock():
logger = Logger(log) logger = Logger(log)
for con in settings.hosts: for con in settings.hosts:
try: try:
log.info("connecting to host '" + str(con) + "'") log.info("connecting to host '" + str(con) + "'")
con = settings.hosts[con] con = settings.hosts[con]
connection, sensors = conSetup.setupConnectionAndSensors(con['host'], con['sensors'], logger.cb_generic) connection, sensors = conSetup.setup_connection_and_sensors(con['host'], con['sensors'], logger.cb_generic)
connections.append(connection) connections.append(connection)
connectedSensors.append(sensors) connectedSensors.append(sensors)
log.info("started logging at " + formatHost(con['host'])) log.info("started logging at " + format_host(con['host']))
except Exception as inst: except Exception as inst:
# connection failed, log and exit # connection failed, log and exit
#TODO: logger.printException(inst) # TODO: logger.print_exception(inst)
log.error("connection failed: " + str(inst)) log.error("connection failed: " + str(inst))
# noinspection PyUnboundLocalVariable
input("Press key to restart\n") input("Press key to restart\n")
log.info("stop logging... @" + time.ctime() + "\n") log.info("stop logging... @" + time.ctime() + "\n")
conSetup.disconnectAny(connections) conSetup.disconnect_any(connections)
freeLock() free_lock()
else: else:
log.critical("lock collision: lock 'all' active") log.critical("lock collision: lock 'all' active")
log.info("wait for retry (" + str(settings.waitDelay) + ")") log.info("wait for retry (" + str(settings.waitDelay) + ")")
time.sleep(settings.waitDelay) time.sleep(settings.waitDelay)
except KeyboardInterrupt: except KeyboardInterrupt:
log.info("keyboard-interrupt happened, cleaning up") log.info("keyboard-interrupt happened, cleaning up")
conSetup.disconnectAny(connections) conSetup.disconnect_any(connections)
freeLock() free_lock()

13
move.py
View File

@ -1,7 +1,6 @@
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import time
import os import os
import logging import logging
from shutil import move from shutil import move
@ -9,7 +8,8 @@ from shutil import move
from timeFunctions import * from timeFunctions import *
from config import settings from config import settings
def setupLogger():
def setup_logger():
log = logging.getLogger("weatherstation.move") log = logging.getLogger("weatherstation.move")
log.setLevel(logging.INFO) log.setLevel(logging.INFO)
ch = logging.StreamHandler() ch = logging.StreamHandler()
@ -22,14 +22,16 @@ def setupLogger():
log.addHandler(fh) log.addHandler(fh)
return log return log
log = setupLogger()
log = setup_logger()
checkfile = os.path.join(settings.locks, settings.movelock) checkfile = os.path.join(settings.locks, settings.movelock)
if not os.path.exists(checkfile): if not os.path.exists(checkfile):
open(checkfile, 'w').close() open(checkfile, 'w').close()
if not os.path.exists(settings.arch): if not os.path.exists(settings.arch):
os.mkdir(arch, 0o000755) os.mkdir(settings.arch, 0o000755)
def mycopy(keep): def mycopy(keep):
names = os.listdir(settings.records) names = os.listdir(settings.records)
@ -38,6 +40,7 @@ def mycopy(keep):
continue continue
move(os.path.join(settings.records, name), settings.arch) move(os.path.join(settings.records, name), settings.arch)
check = open(checkfile, 'r') check = open(checkfile, 'r')
temp = check.read() temp = check.read()
if len(temp) < 1: if len(temp) < 1:
@ -49,7 +52,7 @@ if len(temp)<1:
else: else:
last = time.gmtime(float(temp)) last = time.gmtime(float(temp))
now = time.gmtime() now = time.gmtime()
if(prevday(last,now)): if prevday(last, now):
log.info("moving records") log.info("moving records")
mycopy(preptime()) mycopy(preptime())
check.close() check.close()

13
timeFunctions.py
View File

@ -3,14 +3,16 @@
import time import time
def prevday(then, now): def prevday(then, now):
# ist "then" gestern (oder noch älter)? # ist "then" gestern (oder noch älter)?
greaterDay = (then.tm_yday < now.tm_yday) and (then.tm_year == now.tm_year) greater_day = (then.tm_yday < now.tm_yday) and (then.tm_year == now.tm_year)
if greaterDay: if greater_day:
newYear = False new_year = False
else: else:
newYear = then.tm_year < now.tm_year new_year = then.tm_year < now.tm_year
return greaterDay or newYear return greater_day or new_year
def preptime(): def preptime():
now = time.localtime() now = time.localtime()
@ -26,4 +28,3 @@ def preptime():
else: else:
month = str(month) month = str(month)
return month + "." + day + "." + year return month + "." + day + "." + year