如何用python实现一个多线程定时器

来源：互联网发布：linux halt命令编辑：程序博客网时间：2024/06/05 09:33

因为自已要写一个和时间有关的方法，每过几秒钟之后要运行一个函数，但在主线程里写一个死循环来作定时器总是觉得不好。正好今天学习了一下python的多线程，可以拿来练手。写了下边的python定时器类，使用这个类你可以在你的代码中加入一个定时器。

代码如下(pytimer.py):

#!/usr/bin/env python

# -*- coding: utf-8 -*-

import threading

import time

from Queue import Queue

class _timerThread(threading.Thread):

def __init__(self, t_name,queue,cond):

threading.Thread.__init__(self, name=t_name)

self.threadtimes = []

self.threadFunc = {}

self.lasttimes = {}

self.queue = queue

def setNewTimer(self,newobj):

if newobj.func == None:

self.threadtimes.remove(newobj.secendtime)

self.threadFunc[str(newobj.secendtime)] = None

self.lasttimes[str(newobj.secendtime)] = None

else:

if newobj.secendtime in self.threadtimes:

self.threadFunc[str(newobj.secendtime)] = newobj.func

self.lasttimes[str(newobj.secendtime)] = int(time.time())

else:

self.threadtimes.append(newobj.secendtime)

self.threadFunc[str(newobj.secendtime)] = newobj.func

self.lasttimes[str(newobj.secendtime)] = int(time.time())

def run(self):

while(True):

if not self.queue.empty():

objtmp = self.queue.get()

self.setNewTimer(objtmp)

timetmp = int(time.time())

for tx in self.threadtimes:

if timetmp - self.lasttimes[str(tx)] >= tx:

self.lasttimes[str(tx)] = timetmp

self.threadFunc[str(tx)](timetmp)

self.condition.release()

class _timerObj():

def __init__(self,secendt,funct):

self.secendtime = secendt

self.func = funct

class pytimer():

def __init__(self):

self.queue = Queue()

self.cond = threading.Condition()

self.t_thread = _timerThread(str(int(time.time())),self.queue, self.cond)

self._timers = []

self._initTimer()

def _initTimer(self):

self.t_thread.setDaemon(True)

self.t_thread.start()

def getTimers(self):

return self._timers

def setTimer(self,secendTime,Func):

objtmp = _timerObj(secendTime,Func)

self._timers.append(secendTime)

self.queue.put(objtmp)

def removeTimer(self,secendTime):

objtmp = _timerObj(secendTime,None)

self._timers.remove(secendTime)

self.queue.put(objtmp)

功能测试(test.py):

#!/usr/bin/env python

# -*- coding: utf-8 -*-

import pytimer

def timerCallBack(timex):

print timex

def main():

timerx = pytimer.pytimer()

timerx.setTimer(2, timerCallBack)

while(True):

pass

if __name__ == '__main__':

main()

运行测试程序后输出结果：

1446334639

1446334641

1446334643

1446334645

1446334647

1446334649

1446334651

1446334653

...

运行结果与预期想要的结果相同。每两秒调用了一次定时器返回函数。

源码获取地址：https://github.com/fengmm521/pytimer/tree/master

阅读全文

0 0