to github
[svn42.git] / play-sound-status.py
index afee171..a60e0d6 100755 (executable)
@@ -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)
@@ -35,6 +35,11 @@ class UWSConfig:
     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%")
@@ -43,7 +48,7 @@ class UWSConfig:
     self.config_parser.add_section('halflife2')
     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.set('halflife2','delay',"0.2")
     self.config_parser.add_section('tardis')
     self.config_parser.set('tardis','arg',"/home/tardis.mp3")
     self.config_parser.set('tardis','type',"slugplaymp3")
@@ -68,6 +73,9 @@ class UWSConfig:
     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")
@@ -78,8 +86,11 @@ class UWSConfig:
     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:
@@ -89,7 +100,7 @@ class UWSConfig:
         self.writeConfigFile()
       else:
         self.checkConfigUpdates()
-    
+
   def checkConfigUpdates(self):
     global logger
     if self.configfile is None:
@@ -114,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)
@@ -128,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
 
@@ -143,6 +157,10 @@ 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 runRemoteCommand(remote_host,remote_shell,user,args=[]):
   global sshp,uwscfg
@@ -158,7 +176,7 @@ def runRemoteCommand(remote_host,remote_shell,user,args=[]):
       return False
     return True
   except Exception, ex:
-    logging.error("runRemoteCommand: "+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:
@@ -173,41 +191,46 @@ def runRemoteCommand(remote_host,remote_shell,user,args=[]):
         else:
           subprocess.call(["kill","-9",str(sshp.pid)])
     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
-  popenTimeout2(cmd,stdinput,ptimeout=float(ptimeout))
+  else:
+    ptimeout = int(float(ptimeout))
+  popenTimeout2(cmd,stdinput,ptimeout=ptimeout)
 
 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") 
+  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.debug("executeAction, action_name=%s, action_type=%s, action_delay=%s" % (action_name,action_type,action_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.getValue(action_name+"_remote_host"), uwscfg.getValue(action_name+"_remote_shell"), user=user, args=args)
+    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.getValue(action_name+"_cmd"), ptimeout=uwscfg.getValue(action_name+"_timeout"), stdinput=uwscfg.getValue(action_name+"_stdinput"), user=user, args=args)
+    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()
@@ -248,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:
@@ -262,7 +285,7 @@ def popenTimeout2(cmd, pinput, returncode_ok=[0], ptimeout=21):
       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))
@@ -270,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")
-  try:
-    conn.close()
-  except:
-    pass
+  logging.info("stopping")
   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)
@@ -299,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) )?(.+)')
-RE_ERROR = re.compile(r'Error: (.+)')
-
 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)
-      
+      data = zmqsub.recv_multipart()
+      (structname, dictdata) = decodeR3Message(data)
+      logging.debug("Got data: " + structname + ":"+ str(dictdata))
+
       #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")
+
+      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=m.group(3), fallback_default="DEFAULT")
+        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:
+      elif structname == "BoreDoomButtonPressEvent":
         playThemeOf(user="PANIC", fallback_default="nothing")
         continue
-
-      m = RE_ERROR.match(line)
-      if not m is None:
+      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
-                
+
   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)