forked from logzhan/RobotKernal-UESTC
235 lines
7.1 KiB
Python
235 lines
7.1 KiB
Python
#!/usr/bin/python
|
||
# -*- coding: utf-8 -*-
|
||
import os, sys, inspect
|
||
import datetime
|
||
import signal
|
||
import threading
|
||
import time
|
||
# function: get directory of current script, if script is built
|
||
# into an executable file, get directory of the excutable file
|
||
def current_file_directory():
|
||
path = os.path.realpath(sys.path[0]) # interpreter starter's path
|
||
if os.path.isfile(path): # starter is excutable file
|
||
path = os.path.dirname(path)
|
||
path = os.path.abspath(path) # return excutable file's directory
|
||
else: # starter is python script
|
||
caller_file = inspect.stack()[0][1] # function caller's filename
|
||
path = os.path.abspath(os.path.dirname(caller_file))# return function caller's file's directory
|
||
if path[-1]!=os.path.sep:path+=os.path.sep
|
||
return path
|
||
|
||
"""格式化字符串"""
|
||
def formatString(string,*argv):
|
||
string=string%argv
|
||
if string.find('$(scriptpath)')>=0:
|
||
string=string.replace('$(scriptpath)',current_file_directory())
|
||
if string.find('$(filenumber2)')>=0:
|
||
i=0
|
||
path=""
|
||
while True:
|
||
path=string.replace('$(scriptpath)',"%02d"%i)
|
||
if not os.path.lexists(path):break
|
||
i+=1
|
||
string=path
|
||
#8位日期（20140404）
|
||
if string.find('$(Date8)')>=0:
|
||
now=datetime.datetime.now()
|
||
string=string.replace('$(Date8)', now.strftime("%Y%m%d"))
|
||
#6位日期（140404）
|
||
if string.find('$(Date6)')>=0:
|
||
now=datetime.datetime.now()
|
||
string=string.replace('$(Date6)', now.strftime("%y%m%d"))
|
||
#6位时间（121212）
|
||
if string.find('$(Time6)')>=0:
|
||
now=datetime.datetime.now()
|
||
string=string.replace('$(Time6)', now.strftime("%H%M%S"))
|
||
#4位时间（1212）
|
||
if string.find('$(Time4)')>=0:
|
||
now=datetime.datetime.now()
|
||
string=string.replace('$(Time4)', now.strftime("%H%M"))
|
||
#文件编号2位（必须在最后）
|
||
if string.find('$(filenumber2)')>=0:
|
||
i=0
|
||
path=""
|
||
while True:
|
||
path=string.replace('$(filenumber2)',"%02d"%i)
|
||
if not os.path.lexists(path):break
|
||
i+=1
|
||
string=path
|
||
#文件编号3位（必须在最后）
|
||
if string.find('$(filenumber3)')>=0:
|
||
i=0
|
||
path=""
|
||
while True:
|
||
path=string.replace('$(filenumber3)',"%03d"%i)
|
||
if not os.path.lexists(path):break
|
||
i+=1
|
||
string=path
|
||
return string
|
||
|
||
"""
|
||
取得进程列表
|
||
格式：(PID,cmd)
|
||
"""
|
||
def getProcessList():
|
||
processList = []
|
||
try:
|
||
for line in os.popen("ps xa"):
|
||
fields = line.split()
|
||
# spid = fields[0]
|
||
pid = 0
|
||
try:pid = int(fields[0])
|
||
except:None
|
||
cmd = line[27:-1]
|
||
# print "PS:%d,%s"%(npid,process)
|
||
if pid != 0 and len(cmd) > 0:
|
||
processList.append((pid, cmd))
|
||
except Exception as e:
|
||
print("getProcessList except:%s" % (e))
|
||
return processList
|
||
def killCommand(cmd):
|
||
try:
|
||
processList = getProcessList()
|
||
for p in processList:
|
||
if p[1].find(cmd) != -1:
|
||
pid = p[0]
|
||
os.kill(pid, signal.SIGKILL)
|
||
except Exception as e:
|
||
print("killCommand ‘%s’ except:%s" % (cmd,e))
|
||
|
||
def check_pid(pid):
|
||
""" Check For the existence of a unix pid. """
|
||
if pid == 0:return False
|
||
try:
|
||
os.kill(pid, 0)
|
||
except OSError:
|
||
return False
|
||
else:
|
||
return True
|
||
|
||
SF=formatString
|
||
|
||
#全局异常捕获
|
||
def excepthook(excType, excValue, tb):
|
||
'''''write the unhandle exception to log'''
|
||
from log import log
|
||
import traceback
|
||
log.e('Unhandled Error: %s',''.join(traceback.format_exception(excType, excValue, tb)))
|
||
sys.exit(-1)
|
||
#sys.__excepthook__(type, value, trace)
|
||
#sys.__excepthook__(excType, excValue, tb)
|
||
|
||
_defaultGlobalExcept=sys.excepthook
|
||
|
||
def enableGlobalExcept(enable=True):
|
||
if enable:
|
||
sys.excepthook = excepthook
|
||
else:
|
||
sys.excepthook=_defaultGlobalExcept
|
||
# 默认启动全局异常处理
|
||
enableGlobalExcept()
|
||
#创建线程
|
||
def createThread(name,target,args=(),autoRun=True,daemon=True):
|
||
from log import log
|
||
def threadProc():
|
||
log.i("thread %s started!",name)
|
||
try:
|
||
target(*args)
|
||
log.i("thread %s ended!",name)
|
||
except Exception as e:
|
||
log.e("thread %s crash!err=%s",name,e)
|
||
thd=threading.Thread(name=name,target=threadProc)
|
||
thd.setDaemon(daemon)
|
||
if autoRun:thd.start()
|
||
return thd
|
||
|
||
|
||
#定时器
|
||
class Timer():
|
||
def __init__(self, timer_proc, args=(),first=0,period=0,name="Timer"):
|
||
self.name=name
|
||
self.first=first
|
||
self.period=period
|
||
self.args=args
|
||
self.timer_proc=timer_proc
|
||
self.thdWork=None
|
||
self.stopFlag=False
|
||
from log import NamedLog
|
||
self.log=NamedLog(name)
|
||
def run(self):
|
||
if self.first>0:
|
||
time.sleep(self.first)
|
||
try:
|
||
self.timer_proc(*self.args)
|
||
except Exception as e:
|
||
self.log.error("timer proc crash!err=%s",e)
|
||
while self.period>0 and not self.stopFlag:
|
||
time.sleep(self.period)
|
||
try:
|
||
self.timer_proc(*self.args)
|
||
except Exception as e:
|
||
self.log.error("timer proc crash!err=%s",e)
|
||
def start(self):
|
||
if self.isAlive():
|
||
self.log.d("already running!")
|
||
return True
|
||
self.stopFlag=False
|
||
self.thdWork=threading.Thread(name=self.name,target=self.run)
|
||
self.thdWork.setDaemon(True)
|
||
self.thdWork.start()
|
||
def stop(self,timeout=3):
|
||
if self.isAlive():
|
||
self.stopFlag=True
|
||
try:
|
||
self.thdWork.join(timeout)
|
||
except Exception as e:
|
||
self.log.e("stop timeout!")
|
||
|
||
def isAlive(self):
|
||
return self.thdWork and self.thdWork.isAlive()
|
||
#计时器
|
||
class Ticker():
|
||
def __init__(self):
|
||
self.reset()
|
||
# 片段,可以判断时长是否在一个特定毫秒段内
|
||
self.section=[]
|
||
def reset(self):
|
||
self.tick=time.time()
|
||
self.end=0
|
||
def stop(self):
|
||
self.end=time.time()
|
||
@property
|
||
def ticker(self):
|
||
if self.end==0:
|
||
return int((time.time()-self.tick)*1000)
|
||
else:
|
||
return int((self.end-self.tick)*1000)
|
||
def addSection(self,a,b):
|
||
a,b=int(a),int(b)
|
||
assert a<b
|
||
self.section.append((a,b))
|
||
def removeSection(self,a,b):
|
||
a,b=int(a),int(b)
|
||
self.section.remove((a,b))
|
||
def removeAllSectioin(self):
|
||
self.section=[]
|
||
def inSection(self):
|
||
tick=self.ticker
|
||
for (a,b) in self.section:
|
||
if tick>=a and tick<=b:
|
||
return True
|
||
return False
|
||
def __call__(self):
|
||
return self.ticker
|
||
def waitExit():
|
||
import log
|
||
log.log.i("start waiting to exit...")
|
||
try:
|
||
while(True):
|
||
time.sleep(1)
|
||
except Exception as e:
|
||
log.log.w("recv exit sign!")
|
||
|
||
def is_python3():
|
||
return sys.hexversion > 0x03000000
|