X-Git-Url: https://git.realraum.at/?p=svn42.git;a=blobdiff_plain;f=play-sound-status.py;h=a60e0d6757eda8ceb6da0d775021d1cd96b046af;hp=25278ea8c704187a527789b69b5b752e5be5e684;hb=a34e51d757fe52cb19de2937ae1b211894167524;hpb=b8eaddd48a89fc3e762b590689322753ad123a15 diff --git a/play-sound-status.py b/play-sound-status.py index 25278ea..a60e0d6 100755 --- a/play-sound-status.py +++ b/play-sound-status.py @@ -6,15 +6,15 @@ import sys #import threading import logging import logging.handlers -import urllib import time import signal -import re -import socket import subprocess import types import ConfigParser import traceback +import random +import zmq +import zmq.utils.jsonapi as json logger = logging.getLogger() logger.setLevel(logging.INFO) @@ -28,38 +28,69 @@ class UWSConfig: def __init__(self,configfile=None): self.configfile=configfile self.config_parser=ConfigParser.ConfigParser() + #make option variable names case sensitive + self.config_parser.optionxform = str + self.config_parser.add_section('cmdlog') + self.config_parser.set('cmdlog','cmd',"logger %ARG%") + self.config_parser.set('cmdlog','timeout',"2.0") + self.config_parser.set('cmdlog','delay',"0.0") + self.config_parser.set('cmdlog','type',"shellcmd") + self.config_parser.add_section('slugplayfile') + self.config_parser.set('slugplayfile','remote_host',"root@slug.realraum.at") + self.config_parser.set('slugplayfile','remote_shell',"/home/playgst.sh %ARG%") + self.config_parser.set('slugplayfile','delay',"0.0") + self.config_parser.set('slugplayfile','type',"remotecmd") + self.config_parser.add_section('slugplaymp3') + self.config_parser.set('slugplaymp3','remote_host',"root@slug.realraum.at") + self.config_parser.set('slugplaymp3','remote_shell',"/home/playmp3.sh %ARG%") + self.config_parser.set('slugplaymp3','delay',"0.0") + self.config_parser.set('slugplaymp3','type',"remotecmd") self.config_parser.add_section('halflife2') - self.config_parser.set('halflife2','remote_cmd',"ssh -i /flash/tuer/id_rsa -o PasswordAuthentication=no -o StrictHostKeyChecking=no %RHOST% %RSHELL%") - self.config_parser.set('halflife2','remote_host',"root@slug.realraum.at") - self.config_parser.set('halflife2','remote_shell',"/home/playmp3.sh /home/half-life-door.mp3") - self.config_parser.set('halflife2','delay',"3.0") - self.config_parser.set('halflife2','type',"remotecmd") + self.config_parser.set('halflife2','arg',"/home/half-life-door.mp3") + self.config_parser.set('halflife2','type',"slugplaymp3") + self.config_parser.set('halflife2','delay',"0.2") self.config_parser.add_section('tardis') - self.config_parser.set('tardis','remote_cmd',"ssh -i /flash/tuer/id_rsa -o PasswordAuthentication=no -o StrictHostKeyChecking=no %RHOST% %RSHELL%") - self.config_parser.set('tardis','remote_host',"root@slug.realraum.at") - self.config_parser.set('tardis','remote_shell',"/home/playmp3.sh /home/tardis.mp3") - self.config_parser.set('tardis','delay',"3.0") - self.config_parser.set('tardis','type',"remotecmd") + self.config_parser.set('tardis','arg',"/home/tardis.mp3") + self.config_parser.set('tardis','type',"slugplaymp3") self.config_parser.add_section('sg1aliengreeting') - self.config_parser.set('sg1aliengreeting','remote_cmd',"ssh -i /flash/tuer/id_rsa -o PasswordAuthentication=no -o StrictHostKeyChecking=no %RHOST% %RSHELL%") - self.config_parser.set('sg1aliengreeting','remote_host',"root@slug.realraum.at") - self.config_parser.set('sg1aliengreeting','remote_shell',"/home/playmp3.sh /home/sg1aliengreeting.mp3") - self.config_parser.set('sg1aliengreeting','delay',"3.0") - self.config_parser.set('sg1aliengreeting','type',"remotecmd") + self.config_parser.set('sg1aliengreeting','arg',"/home/sg1aliengreeting.mp3") + self.config_parser.set('sg1aliengreeting','type',"slugplaymp3") self.config_parser.add_section('monkeyscream') - self.config_parser.set('monkeyscream','remote_cmd',"ssh -i /flash/tuer/id_rsa -o PasswordAuthentication=no -o StrictHostKeyChecking=no %RHOST% %RSHELL%") - self.config_parser.set('monkeyscream','remote_host',"root@slug.realraum.at") - self.config_parser.set('monkeyscream','remote_shell',"/home/playmp3.sh /home/monkeyscream.mp3") - self.config_parser.set('monkeyscream','delay',"3.0") - self.config_parser.set('monkeyscream','type',"remotecmd") + self.config_parser.set('monkeyscream','arg',"/home/monkeyscream.mp3") + self.config_parser.set('monkeyscream','delay',"1.5") + self.config_parser.set('monkeyscream','type',"slugplaymp3") + self.config_parser.add_section('gladosparty') + self.config_parser.set('gladosparty','arg',"/home/glados_party.mp3") + self.config_parser.set('gladosparty','type',"slugplaymp3") + self.config_parser.add_section('gladosbaked') + self.config_parser.set('gladosbaked','arg',"/home/glados_baked.mp3") + self.config_parser.set('gladosbaked','type',"slugplaymp3") + self.config_parser.add_section('gladoswelcome') + self.config_parser.set('gladoswelcome','arg',"/home/glados_welcome.mp3") + self.config_parser.set('gladoswelcome','type',"slugplaymp3") + self.config_parser.add_section('gladosreplaced') + self.config_parser.set('gladosreplaced','arg',"/home/glados_replaced_with_life_fire.mp3") + self.config_parser.set('gladosreplaced','type',"slugplaymp3") + self.config_parser.add_section('nothing') + self.config_parser.set('nothing','type',"nothing") + self.config_parser.add_section('randomset1') + self.config_parser.set('randomset1','type',"random") + self.config_parser.set('randomset1','one_of',"halflife2 gladosparty") self.config_parser.add_section('mapping') self.config_parser.set('mapping','DEFAULT',"halflife2") self.config_parser.set('mapping','PANIC',"monkeyscream") - self.config_parser.set('mapping','xro',"tardis") + self.config_parser.set('mapping','ERROR',"nothing") + self.config_parser.set('mapping','stratos',"tardis") + self.config_parser.set('mapping','xro',"gladosreplaced") + self.config_parser.set('mapping','equinox',"gladosparty") + self.config_parser.set('mapping','Rachel',"nothing") self.config_parser.add_section('debug') self.config_parser.set('debug','enabled',"False") + self.config_parser.add_section('broker') + self.config_parser.set('broker','uri',"tcp://torwaechter.realraum.at:4244") self.config_parser.add_section('tracker') - self.config_parser.set('tracker','socket',"/var/run/tuer/presence.socket") + self.config_parser.set('tracker','secs_movement_before_presence_to_launch_event','1') + self.config_parser.set('tracker','secs_presence_before_movement_to_launch_event','120') self.config_mtime=0 if not self.configfile is None: try: @@ -69,7 +100,7 @@ class UWSConfig: self.writeConfigFile() else: self.checkConfigUpdates() - + def checkConfigUpdates(self): global logger if self.configfile is None: @@ -94,7 +125,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) @@ -108,8 +139,11 @@ class UWSConfig: underscore_pos=name.find('_') if underscore_pos < 0: raise AttributeError + return self.getSectionValue(name[0:underscore_pos], name[underscore_pos+1:]) + + def getSectionValue(self, section, name): try: - return self.config_parser.get(name[0:underscore_pos], name[underscore_pos+1:]) + return self.config_parser.get(section,name) except (ConfigParser.NoOptionError, ConfigParser.NoSectionError): return None @@ -123,19 +157,26 @@ class UWSConfig: raise AttributeError +def runRandomAction(action_list,user,args=[]): + if not type(action_list) == types.ListType: + raise ValueError("runRandomAction: action_list must be a list") + return executeAction(random.choice(action_list),user,args) -def playRemoteSound(config): +def runRemoteCommand(remote_host,remote_shell,user,args=[]): global sshp,uwscfg sshp = None try: - cmd = uwscfg.getValue(config+"_remote_cmd").replace("%RHOST%",uwscfg.getValue(config+"_remote_host")).replace("%RSHELL%",uwscfg.getValue(config+"_remote_shell")).split(" ") - logging.debug("playRemoteSound: Executing: "+" ".join(cmd)) - sshp = subprocess.Popen(cmd, bufsize=1024, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, shell=False) - logging.debug("playRemoteSound: pid %d: running=%d" % (sshp.pid,sshp.poll() is None)) + cmd = "ssh -i /flash/tuer/id_rsa -o PasswordAuthentication=no -o StrictHostKeyChecking=no %RHOST% %RSHELL%" + cmd = cmd.replace("%RHOST%",remote_host).replace("%RSHELL%",remote_shell).replace("%ARG%", " ".join(args)).replace("%USER%", user) + logging.debug("runRemoteCommand: Executing: "+cmd) + sshp = subprocess.Popen(cmd.split(" "), bufsize=1024, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, shell=False) + logging.debug("runRemoteCommand: pid %d: running=%d" % (sshp.pid,sshp.poll() is None)) if not sshp.poll() is None: - raise Exception("playRemoteSound: subprocess %d not started ?, returncode: %d" % (sshp.pid,sshp.returncode)) + logging.error("runRemoteCommand: subprocess %d not started ?, returncode: %d" % (sshp.pid,sshp.returncode)) + return False + return True except Exception, ex: - logging.error("playRemoteSound: "+str(ex)) + logging.error("runRemoteCommand: "+str(ex)) traceback.print_exc(file=sys.stdout) if not sshp is None and sshp.poll() is None: if sys.hexversion >= 0x020600F0: @@ -144,28 +185,62 @@ def playRemoteSound(config): subprocess.call(["kill",str(sshp.pid)]) time.sleep(1.5) if sshp.poll() is None: - logging.error("playRemoteSound: subprocess still alive, sending SIGKILL to pid %d" % (sshp.pid)) + logging.error("runRemoteCommand: subprocess still alive, sending SIGKILL to pid %d" % (sshp.pid)) if sys.hexversion >= 0x020600F0: sshp.kill() else: subprocess.call(["kill","-9",str(sshp.pid)]) - time.sleep(5) + time.sleep(5) + return False + +def runShellCommand(cmd,ptimeout,stdinput,user,args=[]): + global uwscfg + cmd = cmd.replace("%ARG%"," ".join(args)).replace("%USER%", user) + if ptimeout is None or float(ptimeout) > 45: + ptimeout = 45 + else: + ptimeout = int(float(ptimeout)) + popenTimeout2(cmd,stdinput,ptimeout=ptimeout) -def playThemeOf(user): +def executeAction(action_name, user, args=[]): + if action_name is None: + logging.error("executeAction: action_name is None") + return False + action_type = uwscfg.getValue(action_name+"_type") + if action_type is None: + logging.error("executeAction: action %s not found or has no type" % action_name) + return False + action_delay=uwscfg.getValue(action_name+"_delay") + logging.info("executeAction %s of type %s for user %s with delay %s" % (action_name,action_type,user,action_delay)) + if not action_delay is None: + time.sleep(float(action_delay)) + + action_arg = uwscfg.getValue(action_name+"_arg") + if not action_arg is None: + args += [action_arg] + + #"registered" actions + if action_type == "remotecmd": + return runRemoteCommand(uwscfg.getSectionValue(action_name,"remote_host"), uwscfg.getSectionValue(action_name,"remote_shell"), user=user, args=args) + elif action_type == "shellcmd": + return runShellCommand(cmd=uwscfg.getSectionValue(action_name,"cmd"), ptimeout=uwscfg.getSectionValue(action_name,"timeout"), stdinput=uwscfg.getSectionValue(action_name,"stdinput"), user=user, args=args) + elif action_type == "nothing": + return True + elif action_type == "random": + return runRandomAction(action_list=uwscfg.getSectionValue(action_name,"one_of").split(" "),user=user,args=args) + else: + return executeAction(action_type,user=user,args=args) + +def playThemeOf(user,fallback_default): global uwscfg uwscfg.checkConfigUpdates() - type=None + if user is None: + user = "" config=uwscfg.getValue("mapping_"+str(user)) if config is None: - config=uwscfg.getValue("mapping_DEFAULT") - type=uwscfg.getValue(config+"_type") - if type is None: - type="remotecmd" - delay=uwscfg.getValue(config+"_delay") - if not delay is None: - time.sleep(float(delay)) - if type is "remotecmd": - playRemoteSound(config) + config=uwscfg.getValue("mapping_"+str(fallback_default)) + logging.debug("playThemeOf: action for user %s: %s" % (user,config)) + executeAction(config,user,[]) def popenTimeout1(cmd, pinput, returncode_ok=[0], ptimeout = 20.0, pcheckint = 0.25): logging.debug("popenTimeout1: starting: " + cmd) @@ -177,7 +252,7 @@ def popenTimeout1(cmd, pinput, returncode_ok=[0], ptimeout = 20.0, pcheckint = 0 time.sleep(pcheckint) timeout_counter -= pcheckint if not sppoo.poll() is None: - logging.debug("popenTimeout2: subprocess %d finished, returncode: %d" % (sppoo.pid,sppoo.returncode)) + logging.debug("popenTimeout1: subprocess %d finished, returncode: %d" % (sppoo.pid,sppoo.returncode)) return (sppoo.returncode in returncode_ok) #timeout reached logging.error("popenTimeout1: subprocess took too long (>%fs), sending SIGTERM to pid %d" % (ptimeout,sppoo.pid)) @@ -196,7 +271,7 @@ def popenTimeout1(cmd, pinput, returncode_ok=[0], ptimeout = 20.0, pcheckint = 0 except Exception, e: logging.error("popenTimeout1: "+str(e)) return False - + def popenTimeout2(cmd, pinput, returncode_ok=[0], ptimeout=21): logging.debug("popenTimeout2: starting: " + cmd) try: @@ -206,10 +281,11 @@ def popenTimeout2(cmd, pinput, returncode_ok=[0], ptimeout=21): else: old_shandler = signal.signal(signal.SIGALRM,lambda sn,sf: os.system("kill -9 %d" % sppoo.pid)) signal.alarm(ptimeout) #schedule alarm - sppoo.communicate(input=pinput) + if not pinput is None: + sppoo.communicate(input=pinput) sppoo.wait() signal.alarm(0) #disable pending alarms - signal.signal(signal.SIGALRM, old_shandler) + signal.signal(signal.SIGALRM, old_shandler) logging.debug("popenTimeout2: subprocess %d finished, returncode: %d" % (sppoo.pid,sppoo.returncode)) if sppoo.returncode < 0: logging.error("popenTimeout2: subprocess took too long (>%ds) and pid %d was killed" % (ptimeout,sppoo.pid)) @@ -217,23 +293,27 @@ def popenTimeout2(cmd, pinput, returncode_ok=[0], ptimeout=21): except Exception, e: logging.error("popenTimeout2: "+str(e)) try: - signal.signal(signal.SIGALRM, old_shandler) + signal.signal(signal.SIGALRM, old_shandler) except: pass return False +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("Update-Web-Status stopping") + logging.info("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) @@ -246,59 +326,64 @@ 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_REQUEST = re.compile(r'Request: (\w+) (?:(Card|Phone) )?(.+)') -last_who=None 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") - #sockhandle.send("status\n") + #Start zmq connection to publish / forward sensor data + zmqctx = zmq.Context() + zmqctx.linger = 0 + zmqsub = zmqctx.socket(zmq.SUB) + zmqsub.setsockopt(zmq.SUBSCRIBE, "DoorCommandEvent") + zmqsub.setsockopt(zmq.SUBSCRIBE, "PresenceUpdate") + zmqsub.setsockopt(zmq.SUBSCRIBE, "BoreDoomButtonPressEvent") +# zmqsub.setsockopt(zmq.SUBSCRIBE, "MovementSensorUpdate") + zmqsub.setsockopt(zmq.SUBSCRIBE, "DoorAjarUpdate") + zmqsub.setsockopt(zmq.SUBSCRIBE, "DoorProblemEvent") + zmqsub.connect(uwscfg.broker_uri) + last_status=None + last_user=None unixts_panic_button=None + unixts_last_movement=0 + unixts_last_presence=0 while True: - 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) - last_status=(status == "yes") + data = zmqsub.recv_multipart() + (structname, dictdata) = decodeR3Message(data) + logging.debug("Got data: " + structname + ":"+ str(dictdata)) + + #uwscfg.checkConfigUpdates() + + if structname == "PresenceUpdate" and "Present" in dictdata: + unixts_last_presence=time.time() + last_status=dictdata["Present"] unixts_panic_button=None - if last_status: - playThemeOf(user=last_who) + if ( time.time() - unixts_last_movement ) <= float(uwscfg.tracker_secs_movement_before_presence_to_launch_event): + unixts_last_movement=0 + if last_status: + playThemeOf(user=last_user, fallback_default="DEFAULT") + continue + elif structname == "DoorCommandEvent": + last_user = dictdata["Who"] continue - - m = RE_BUTTON.match(line) - if not m is None: - playThemeOf(user="PANIC") + elif structname == "BoreDoomButtonPressEvent": + playThemeOf(user="PANIC", fallback_default="nothing") + continue + elif structname == "MovementSensorUpdate" or structname == "DoorAjarUpdate": + unixts_last_movement=time.time() + if (time.time() - unixts_last_presence) <= float(uwscfg.tracker_secs_presence_before_movement_to_launch_event): + unixts_last_presence=0 + if last_status: + playThemeOf(user=last_user, fallback_default="DEFAULT") + continue + elif structname == "DoorProblemEvent" and "Severity" in dictdata: + playThemeOf(user="ERROR", fallback_default="nothing") continue - m = RE_REQUEST.match(line) - if not m is None: - last_who = m.group(3) - #last_how = m.group(2) - continue - except Exception, ex: - logging.error("main: "+str(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)