X-Git-Url: https://git.realraum.at/?p=svn42.git;a=blobdiff_plain;f=switch-power.py;h=94b0deeae9bd1460bb30684770a59f75094679d5;hp=5a33f19fcd58ea9f87c592567e3af51fb5cca112;hb=a34e51d757fe52cb19de2937ae1b211894167524;hpb=85b29d83b79b660075a7aa02efb3f696c25e11f3 diff --git a/switch-power.py b/switch-power.py index 5a33f19..94b0dee 100755 --- a/switch-power.py +++ b/switch-power.py @@ -3,16 +3,13 @@ import os import os.path import sys -#import threading import logging import logging.handlers import urllib import time import signal -import re -import socket -import subprocess -import types +import json +import zmq import ConfigParser import datetime import random @@ -31,22 +28,26 @@ class UWSConfig: self.configfile=configfile self.config_parser=ConfigParser.ConfigParser() self.config_parser.add_section('powerswitching') - self.config_parser.set('powerswitching','min_secs_periodical_event','59') + self.config_parser.set('powerswitching','secs_movement_before_presence_to_launch_event','1') + self.config_parser.set('powerswitching','secs_presence_before_movement_to_launch_event','120') self.config_parser.set('powerswitching','max_secs_since_movement','600') self.config_parser.add_section('slug') self.config_parser.set('slug','cgiuri','http://slug.realraum.at/cgi-bin/switch.cgi?id=%ID%&power=%ONOFF%') + self.config_parser.set('slug','lightleveluri','http://slug.realraum.at/cgi-bin/lightlevel.cgi') self.config_parser.set('slug','ids_logo','logo') self.config_parser.set('slug','ids_present_day_bright_room','ymhpoweron werkzeug ymhcd') self.config_parser.set('slug','ids_present_day_dark_room','ymhpoweron decke werkzeug ymhcd') - self.config_parser.set('slug','ids_present_night','ymhpoweron werkzeug schreibtisch idee labor ymhcd') - self.config_parser.set('slug','ids_panic','idee ymhmute labor werkzeug deckevorne deckehinten') + self.config_parser.set('slug','ids_present_night','ymhpoweron werkzeug schreibtisch spots1 labor ymhcd') + self.config_parser.set('slug','ids_panic','spots1 ymhmute labor werkzeug deckevorne deckehinten') + self.config_parser.set('slug','ids_decke','deckevorne deckehinten') self.config_parser.set('slug','ids_nonpresent_off','ymhvoldown ymhvoldown ymhvoldown ymhvoldown ymhvoldown ymhvoldown ymhvoldown ymhvoldown ymhvoldown ymhvoldown ymhvoldown ymhvoldown lichter ymhpoweroff lichter') self.config_parser.set('slug','light_threshold_brightness','400') + self.config_parser.set('slug','light_difference_decke','100') #self.config_parser.set('slug','time_day','6:00-17:00') self.config_parser.add_section('debug') self.config_parser.set('debug','enabled',"False") - self.config_parser.add_section('tracker') - self.config_parser.set('tracker','socket',"/var/run/tuer/presence.socket") + self.config_parser.add_section('broker') + self.config_parser.set('broker','uri',"tcp://torwaechter.realraum.at:4244") self.config_mtime=0 if not self.configfile is None: try: @@ -56,7 +57,7 @@ class UWSConfig: self.writeConfigFile() else: self.checkConfigUpdates() - + def checkConfigUpdates(self): global logger if self.configfile is None: @@ -81,7 +82,7 @@ class UWSConfig: def writeConfigFile(self): if self.configfile is None: return - logging.debug("Writing Configfile "+self.configfile) + logging.debug("Writing Configfile "+self.configfile) try: cf_handle = open(self.configfile,"w") self.config_parser.write(cf_handle) @@ -118,9 +119,16 @@ def switchPower(powerid,turn_on=False): onoff="off" touchURL(uwscfg.slug_cgiuri.replace("%ID%",powerid).replace("%ONOFF%",onoff)) +def getLightValueNow(): + lvalue = touchURL(uwscfg.slug_lightleveluri) + try: + return int(lvalue) + except: + return None + def haveDaylight(): dawn_per_month = {1:8, 2:7, 3:6, 4:6, 5:5, 6:5, 7:5, 8:6, 9:7, 10:8, 11:8, 12:8} - dusk_per_month = {1:16, 2:17, 3:18, 4:20, 5:20, 6:21, 7:21, 8:20, 9:19, 10:18, 11:16, 12:16} + dusk_per_month = {1:15, 2:16, 3:17, 4:19, 5:20, 6:20, 7:20, 8:19, 9:17, 10:17, 11:16, 12:15} hour = datetime.datetime.now().hour month = datetime.datetime.now().month return (hour >= dawn_per_month[month] and hour < dusk_per_month[month]) @@ -129,11 +137,32 @@ def isWolfHour(): hour = datetime.datetime.now().hour return (hour >= 2 and hour < 6) -######### EVENTS ############### +######### ALGOS ############### + +def switchLogo(status_presence): + logo_action=None + if status_presence: + logo_action=True + else: + if haveDaylight(): + logo_action=False + else: + if isWolfHour(): + logo_action=False + else: + logo_action=True + if not logo_action is None: + logging.info("switchLogo: logo_action:%s" % str(logo_action)) + for id in uwscfg.slug_ids_logo.split(" "): + switchPower(id,logo_action) + +######### EVENTS ############### unixts_last_movement=0 -status_presense=None +unixts_last_presence=0 +status_presence=None room_is_bright=None + def eventRoomGotBright(): global room_is_bright logging.debug("eventRoomGotBright()") @@ -144,48 +173,27 @@ def eventRoomGotDark(): logging.debug("eventRoomGotDark()") room_is_bright=False -def eventDaylightStart(): - logging.debug("eventDaylightStart()") - for id in uwscfg.slug_ids_logo.split(" "): - switchPower(id,False) - -def eventDaylightStop(): - logging.debug("eventDaylightStop()") - if not isWolfHour(): - for id in uwscfg.slug_ids_logo.split(" "): - switchPower(id,True) - -def eventWolfHourStart(): - logging.debug("eventWolfHourStart()") - for id in uwscfg.slug_ids_logo.split(" "): - switchPower(id,False) - -def eventWolfHourStop(): - logging.debug("eventWolfHourStop()") - if haveDaylight(): - for id in uwscfg.slug_ids_logo.split(" "): - switchPower(id,True) - def eventMovement(): - global unixts_last_movement + global unixts_last_movement, unixts_last_presence unixts_last_movement=time.time() + if (time.time() - unixts_last_presence) <= float(uwscfg.powerswitching_secs_presence_before_movement_to_launch_event): + eventPresentAndMoved() + unixts_last_presence=0 # so that eventPresentAndMoved will only launch once per presence event (i.e. supress multiple movement events) -def eventPeriodical(): - pass # global unixts_last_movement -# if status_presense is True and unixts_last_movement + int(uwscfg.powerswitching_max_secs_since_movement) >= time.time(): +# if status_presence is True and unixts_last_movement + int(uwscfg.powerswitching_max_secs_since_movement) >= time.time(): # presumed_state=not (haveDaylight() or isWolfHour()) -# logging.debug("event: periodical event") # for id in uwscfg.slug_ids_logo.split(" "): # switchPower(id,not presumed_state) # time.sleep(1); # switchPower(id,presumed_state) def eventPresent(): - global status_presense,room_is_bright + global status_presence,room_is_bright,unixts_last_movement,uwscfg,unixts_last_presence + unixts_last_presence=time.time() logging.debug("eventPresent()"); - status_presense=True + status_presence=True if haveDaylight(): if room_is_bright is False: present_ids=uwscfg.slug_ids_present_day_dark_room @@ -196,15 +204,31 @@ def eventPresent(): logging.info("event: someone present, switching on: "+present_ids) for id in present_ids.split(" "): switchPower(id,True) + switchLogo(status_presence) + if ( time.time() - unixts_last_movement ) <= float(uwscfg.powerswitching_secs_movement_before_presence_to_launch_event): + unixts_last_movement=0 + eventPresentAndMoved() + +def eventPresentAndMoved(): + global status_presence,room_is_bright + pass def eventNobodyHere(): - global status_presense + global status_presence logging.debug("eventNobodyHere()"); - status_presense=False + status_presence=False present_ids=uwscfg.slug_ids_nonpresent_off logging.info("event: noone here, switching off: "+present_ids) - for id in present_ids.split(" "): - time.sleep(0.2) + present_id_list=present_ids.split(" ") + for id in present_id_list: + time.sleep(0.15) + switchPower(id,False) + present_id_list.reverse() + time.sleep(0.15) + switchLogo(status_presence) + time.sleep(2) + for id in present_id_list: + time.sleep(0.15) switchPower(id,False) ##def eventPanic(): @@ -216,46 +240,67 @@ def eventNobodyHere(): ## for delay in map(lambda e: (40-e)/33.0,range(10,33)): ## e = random.choice(lst2) ## e[1]=not e[1] -## switchPower(e[0],e[1]) +## switchPower(e[0],e[1]) ## time.sleep(delay) ## random.shuffle(lst1) ## for id in lst1: ## switchPower(id,False) ## time.sleep(1.2) ## eventPresent() - + def eventPanic(): + global light_value logging.info("eventPanic(): switching around: "+uwscfg.slug_ids_panic) lst1 = uwscfg.slug_ids_panic.split(" ") lst2 = map(lambda e:[e,False], lst1) + deckenlicht_ids = uwscfg.slug_ids_decke.split(" ") + ceiling_light_was_on = False + #guess main light state: + if len(list(set(deckenlicht_ids) & set(lst1))) > 0: + light_value_before = light_value + for id in deckenlicht_ids: + switchPower(id,False) + time.sleep(2.8) + light_value_after = getLightValueNow() + if not (light_value_before is None or light_value_after is None): + ceiling_light_was_on = ((light_value_before - light_value_after) > int(uwscfg.slug_light_difference_decke)) + logging.debug("eventPanic: light_value_before: %d, light_value_after: %d, ceiling_light_was_on: %s" % (light_value_before,light_value_after,str(ceiling_light_was_on))) for id in lst1: switchPower(id,False) - for times in range(1,6) + for times in range(1,6): delay = random.choice([0.3,1.4,0.9,0.5,0.3,1.4,0.9,0.5,2.2]) time.sleep(delay) - for e in lst1: + for e in lst2: e[1]=not e[1] - switchPower(e[0],e[1]) + switchPower(e[0],e[1]) for id in lst1: switchPower(id,False) time.sleep(1.2) eventPresent() - + #we can only test if it was on, we don't any other informatino + if ceiling_light_was_on: + for id in deckenlicht_ids: + switchPower(id,True) + ######################## +def decodeR3Message(multipart_msg): + try: + return (multipart_msg[0], json.loads(multipart_msg[1])) + except Exception, e: + logging.debug("decodeR3Message:"+str(e)) + return ("",{}) + def exitHandler(signum, frame): logging.info("Power Switch Daemon stopping") try: - conn.close() - except: - pass - try: - sockhandle.close() + zmqsub.close() + zmqctx.destroy() except: pass sys.exit(0) - + #signals proapbly don't work because of readline #signal.signal(signal.SIGTERM, exitHandler) signal.signal(signal.SIGINT, exitHandler) @@ -268,71 +313,41 @@ if len(sys.argv) > 1: else: uwscfg = UWSConfig() -#socket.setdefaulttimeout(10.0) #affects all new Socket Connections (urllib as well) -RE_PRESENCE = re.compile(r'Presence: (yes|no)(?:, (opened|closed), (.+))?') -RE_BUTTON = re.compile(r'PanicButton|button\d?') -RE_MOVEMENT = re.compile(r'movement') -RE_PHOTO = re.compile(r'photo\d: [^0-9]*?(\d+)') + daylight=None wolfhour=None -unixts_last_periodical=0 +light_value=0 while True: try: - if not os.path.exists(uwscfg.tracker_socket): - logging.debug("Socketfile '%s' not found, waiting 5 secs" % uwscfg.tracker_socket) - time.sleep(5) - continue - sockhandle = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) - sockhandle.connect(uwscfg.tracker_socket) - conn = os.fdopen(sockhandle.fileno()) - #sockhandle.send("listen\n") + #Start zmq connection to publish / forward sensor data + zmqctx = zmq.Context() + zmqctx.linger = 0 + zmqsub = zmqctx.socket(zmq.SUB) + zmqsub.setsockopt(zmq.SUBSCRIBE, "") + zmqsub.connect(uwscfg.broker_uri) + while True: - if haveDaylight() != daylight: - daylight = haveDaylight() - if daylight: - eventDaylightStart() - else: - eventDaylightStop() + data = zmqsub.recv_multipart() + (structname, dictdata) = decodeR3Message(data) + logging.debug("Got data: " + structname + ":"+ str(dictdata)) - if isWolfHour() != wolfhour: - wolfhour = isWolfHour() - if wolfhour: - eventWolfHourStart() - else: - eventWolfHourStop() - - if unixts_last_periodical + int(uwscfg.powerswitching_min_secs_periodical_event) <= time.time(): - unixts_last_periodical = time.time() - eventPeriodical() - - line = conn.readline() - logging.debug("Got Line: " + line) - uwscfg.checkConfigUpdates() - - if line == "": - raise Exception("EOF on Socket, daemon seems to have quit") - - m = RE_PRESENCE.match(line) - if not m is None: - status = m.group(1) - if status == "yes": + + if structname == "PresenceUpdate" and "Present" in dictdata: + if dictdata["Present"]: eventPresent() else: eventNobodyHere() continue - m = RE_BUTTON.match(line) - if not m is None: + elif structname == "BoreDoomButtonPressEvent": eventPanic() continue - m = RE_MOVEMENT.match(line) - if not m is None: + elif structname == "MovementSensorUpdate" or structname == "DoorAjarUpdate": eventMovement() continue - m = RE_PHOTO.match(line) - if not m is None: - light_value = int(m.group(1)) + elif structname == "IlluminationSensorUpdate" and "Value" in dictdata: + light_value = dictdata["Value"] light_threshold = int(uwscfg.slug_light_threshold_brightness) #logging.debug("photo value: %d threshold: %s" % (light_value,light_threshold)) if light_value >= light_threshold: @@ -340,14 +355,13 @@ while True: else: eventRoomGotDark() continue - + except Exception, ex: logging.error("main: "+str(ex)) traceback.print_exc(file=sys.stdout) try: - sockhandle.close() + zmqsub.close() + zmqctx.destroy() except: pass - conn=None - sockhandle=None time.sleep(5)