X-Git-Url: https://git.realraum.at/?a=blobdiff_plain;f=play-sound-status.py;h=976bed40c542ea4aff6114c2df37f34c4d0eace9;hb=8f8ffe1d0ca5e05b9942f7cacf28dd4a348053c2;hp=63197389aa72385c0eeed223bcd62724b76b4ddb;hpb=2adf4b79741d22ed128da6ede3a6323577e264eb;p=svn42.git diff --git a/play-sound-status.py b/play-sound-status.py index 6319738..976bed4 100755 --- a/play-sound-status.py +++ b/play-sound-status.py @@ -15,6 +15,7 @@ import subprocess import types import ConfigParser import traceback +import random logger = logging.getLogger() logger.setLevel(logging.INFO) @@ -28,34 +29,57 @@ 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('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','DEFAULT',"halflife2") + self.config_parser.set('mapping','PANIC',"monkeyscream") + 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') @@ -108,8 +132,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 +150,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,31 +178,61 @@ 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) + +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") - logging.debug("playThemeOf, user=%s, config=%s, type=%s, delay=%s" % (user,config,type,delay)) - if not delay is None: - time.sleep(float(delay)) - if type == "remotecmd": - playRemoteSound(config) - else: - logging.debug("playThemeOf: Error, unknown type") + 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) @@ -180,7 +244,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)) @@ -209,7 +273,8 @@ 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) @@ -226,7 +291,7 @@ def popenTimeout2(cmd, pinput, returncode_ok=[0], ptimeout=21): return False def exitHandler(signum, frame): - logging.info("Update-Web-Status stopping") + logging.info("stopping") try: conn.close() except: @@ -252,7 +317,9 @@ 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) )?(.+)') +#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): @@ -270,7 +337,7 @@ 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") @@ -281,13 +348,18 @@ while True: last_status=(status == "yes") unixts_panic_button=None if last_status: - playThemeOf(user=m.group(3)) + playThemeOf(user=m.group(3), fallback_default="DEFAULT") continue m = RE_BUTTON.match(line) if not m is None: - playThemeOf(user="panic") - continue + playThemeOf(user="PANIC", fallback_default="nothing") + continue + + m = RE_ERROR.match(line) + if not m is None: + playThemeOf(user="ERROR", fallback_default="nothing") + continue except Exception, ex: logging.error("main: "+str(ex))