import types
import ConfigParser
import traceback
+import random
logger = logging.getLogger()
logger.setLevel(logging.INFO)
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.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','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')
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
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,args=[]):
+def runRemoteCommand(remote_host,remote_shell,user,args=[]):
global sshp,uwscfg
sshp = None
try:
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))
+ 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))
subprocess.call(["kill","-9",str(sshp.pid)])
time.sleep(5)
-def runShellCommand(cmd,ptimeout,stdinput,args=[]):
+def runShellCommand(cmd,ptimeout,stdinput,user,args=[]):
global uwscfg
- cmd = cmd.replace("%ARG%"," ".join(args))
+ cmd = cmd.replace("%ARG%"," ".join(args)).replace("%USER%", user)
if ptimeout is None or float(ptimeout) > 45:
ptimeout = 45
- popenTimeout2(cmd,stdinput,float(ptimeout))
+ else:
+ ptimeout = int(float(ptimeout))
+ popenTimeout2(cmd,stdinput,ptimeout=ptimeout)
-def executeAction(action_name, args=[]):
+def executeAction(action_name, user, args=[]):
if action_name is None:
logging.error("executeAction: action_name is None")
return False
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))
#"registered" actions
if action_type == "remotecmd":
- return runRemoteCommand(uwscfg.getValue(action_name+"_remote_host"), uwscfg.getValue(action_name+"_remote_shell"), 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"), 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,args)
+ return executeAction(action_type,user=user,args=args)
-def playThemeOf(user):
+def playThemeOf(user,fallback_default):
global uwscfg
uwscfg.checkConfigUpdates()
+ if user is None:
+ user = ""
config=uwscfg.getValue("mapping_"+str(user))
if config is None:
- config=uwscfg.getValue("mapping_default")
+ config=uwscfg.getValue("mapping_"+str(fallback_default))
logging.debug("playThemeOf: action for user %s: %s" % (user,config))
- executeAction(config)
+ executeAction(config,user,[])
def popenTimeout1(cmd, pinput, returncode_ok=[0], ptimeout = 20.0, pcheckint = 0.25):
logging.debug("popenTimeout1: starting: " + cmd)
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))
return False
def exitHandler(signum, frame):
- logging.info("Update-Web-Status stopping")
+ logging.info("stopping")
try:
conn.close()
except:
#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):
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))