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

1
.gitignore vendored
View File

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

View File

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

View File

@ -49,3 +49,4 @@ Optional: Upload der Aufzeichnungen mit SFTP an einem Server (beinhaltet Archivi
# TODOS
* 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)

View File

@ -18,31 +18,34 @@ except ImportError:
from config import SensorType
from config import settings
class ConnectionSetup(object):
def __init__(self, log):
self.log = log
def setupConnection(self, host):
def setup_connection(self, host):
ipcon = IPConnection()
try:
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):
hostname = host['name']
port = host['port']
ipcon = IPConnection()
ipcon.connect(hostname, port)
sensorSetup = SensorSetup(ipcon, sensors, cb_generic, self.log)
connectedSensors = sensorSetup.setupSensors()
return (ipcon, connectedSensors)
def setup_connection_and_sensors(self, host, sensors, cb_generic):
ipcon = self.setup_connection(host)
sensor_setup = SensorSetup(ipcon, sensors, cb_generic, self.log)
connected_sensors = sensor_setup.setup_sensors()
return ipcon, connected_sensors
def disconnectAny(self, connections):
def disconnect_any(self, connections):
for connection in connections:
if not connection.get_connection_state() is IPConnection.CONNECTION_STATE_DISCONNECTED:
self.log.debug("disconnecting (%s)", connection)
connection.disconnect()
class SensorSetup(object):
class SensorSetup(object):
def __init__(self, connection, sensors, cb_generic, log):
self.connection = connection
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)
def __setupSensor__(self, callback, id, cbtime, var):
obj = None
if id in self._previous_sensors:
self.log.debug("reusing instance for %s", id)
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
return obj
def genericSensorSetup(self, name, sensor):
def generic_sensor_setup(self, name, sensor):
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]
obj = None
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())
return obj
def setupSensors(self):
def setup_sensors(self):
connected = []
for name in self.sensors:
sensor = self.sensors[name]
obj = self.genericSensorSetup(name, sensor)
obj = self.generic_sensor_setup(name, sensor)
connected.append(obj)
return connected

View File

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

33
main.py
View File

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

13
move.py
View File

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

View File

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