python简单的任务队列

不写注释是个好习惯

import time
from datetime import datetime
from queue import Queue
from threading import Thread

doing = []


class TaskQueue(Queue):
    def __init__(self, num_workers=1):
        Queue.__init__(self)
        self.num_workers = num_workers
        self.start_workers()

    def add_task(self, **kwargs):
        kwargs = kwargs or {}
        self.put((kwargs))

    def start_workers(self):
        for i in range(self.num_workers):
            t = Thread(target=self.worker)
            t.daemon = True
            t.start()

    def worker(self):
        while True:
            # tupl = self.get()
            dict = self.get()
            # 添加执行中
            doing.append(dict)
            # 执行
            runTask(dict)
            self.task_done()
            # 移除执行中
            doing.remove(dict)


def runTask(dict):
    """
    执行
    :param dict:
    :return:
    """
    print(dict, "-step1", datetime.now())

    time.sleep(1)
    print(dict, "-step2", datetime.now())

    time.sleep(1)
    print(dict, "-finish", datetime.now())

    return


if __name__ == "__main__":
    q = TaskQueue(num_workers=1)
    for i in range(10):
        q.add_task(id=i,name="测试-"+str(i))
    q.join()
    print("完成")