def run(self): prevmsg = str() while(1): d = self._socket.recv(32) line = linesep.split(prevmsg + d) prevmsg = line.pop() for msg in line: if not msg: continue #logger.debug("receive %s"%msg) if ping.match(msg): self._socket.send("PONG :%s\r\n"%(ping.match(msg).group('ping'))) event.log_event ("ping", self._socket, ping.match(msg).group('ping')) if d == "": self._socket.close () break
def save_networkcfg(interface, address, mask, gateway): net_cfg_file = net_cfg_file_prefix + interface net_cfg_file = os.path.join(net_cfg_pos, net_cfg_file) if os.path.exists(net_cfg_file): s = open(net_cfg_file, 'r').read() if address: s = re.sub(r'(IPADDR=)([^\n]+)', r'\g<1>%s' % address, s) if mask: s = re.sub(r'(NETMASK=)([^\n]+)', r'\g<1>%s' % mask, s) if gateway: s = re.sub(r'(GATEWAY=)([^\n]+)', r'\g<1>%s' % gateway, s) # backup origin configuration if not os.path.exists(net_cfg_file+'.origin'): os.rename(net_cfg_file, net_cfg_file+".orgin") # save configuration open(net_cfg_file, 'w').write(s) else: open(net_cfg_file, 'w').write(net_cfg_items % (address, mask, gateway)) # chnage api server address for web store_ip(address) event.log_event(logging.INFO, event.event_os, event.event_os_network, 'Network setting:interface=%s address=%s mask=%s gateway=%s' % (interface, address, mask, gateway))
def reboot(): r = confirm('Confirm to reboot') if r: event.log_event(logging.INFO, event.event_os, event_os_reboot, 'Reboot') subprocess.call([grgrant_prog, '/sbin/reboot'])
def poweroff(): r = confirm('Confirm to poweroff') if r: event.log_event(logging.INFO, event.event_os, event_os_poweroff, 'poweroff') subprocess.call([grgrant_prog, '/sbin/poweroff'])
def test_log_event(self): result_json = log_event("text","type") result = json.loads(result_json) message = result['message'] trimmed_result=message[:10] self.assertEqual(trimmed_result, "event created on..."[:10])