X-Git-Url: https://git.realraum.at/?a=blobdiff_plain;f=play-sound-status.py;h=6eb433a06879cfa719a99b6d917d6a5614ad0e75;hb=9b57ad36fa772a94f8f38321a88f2b1df1818a51;hp=25278ea8c704187a527789b69b5b752e5be5e684;hpb=b8eaddd48a89fc3e762b590689322753ad123a15;p=svn42.git diff --git a/play-sound-status.py b/play-sound-status.py index 25278ea..6eb433a 100755 --- a/play-sound-status.py +++ b/play-sound-status.py @@ -6,7 +6,6 @@ import sys #import threading import logging import logging.handlers -import urllib import time import signal import re @@ -15,6 +14,7 @@ import subprocess import types import ConfigParser import traceback +import random logger = logging.getLogger() logger.setLevel(logging.INFO) @@ -28,38 +28,68 @@ 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('tracker') - self.config_parser.set('tracker','socket',"/var/run/tuer/presence.socket") + 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 +99,7 @@ class UWSConfig: self.writeConfigFile() else: self.checkConfigUpdates() - + def checkConfigUpdates(self): global logger if self.configfile is None: @@ -94,7 +124,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 +138,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 +156,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 +184,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 playThemeOf(user): +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 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 +251,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 +270,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 +280,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,13 +292,13 @@ 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 exitHandler(signum, frame): - logging.info("Update-Web-Status stopping") + logging.info("stopping") try: conn.close() except: @@ -233,7 +308,7 @@ def exitHandler(signum, frame): except: pass sys.exit(0) - + #signals proapbly don't work because of readline #signal.signal(signal.SIGTERM, exitHandler) signal.signal(signal.SIGINT, exitHandler) @@ -249,8 +324,10 @@ else: #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 +#RE_REQUEST = re.compile(r'Request: (\w+) (?:(Card|Phone) )?(.+)') +RE_ERROR = re.compile(r'Error: (.+)') +RE_MOVEMENT = re.compile(r'movement') + while True: try: if not os.path.exists(uwscfg.tracker_socket): @@ -263,42 +340,57 @@ while True: #sockhandle.send("listen\n") #sockhandle.send("status\n") 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() - + + #uwscfg.checkConfigUpdates() + if line == "": raise Exception("EOF on Socket, daemon seems to have quit") - + + m = RE_MOVEMENT.match(line) + if not m is None: + 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 + m = RE_PRESENCE.match(line) if not m is None: status = m.group(1) + unixts_last_presence=time.time() last_status=(status == "yes") unixts_panic_button=None - if last_status: - playThemeOf(user=last_who) + last_user=m.group(3) + 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 - + m = RE_BUTTON.match(line) if not m is None: - playThemeOf(user="PANIC") + playThemeOf(user="PANIC", fallback_default="nothing") + continue + + m = RE_ERROR.match(line) + if not m is None: + 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)) try: sockhandle.close() except: pass conn=None - sockhandle=None + sockhandle=None time.sleep(5)