import time
import threading
import queue

# 一個線程，間隔一定的時間，把一個遞增的數字寫入隊列
# 生產者
class Producer(threading.Thread):

    def __init__(self, work_queue):
        super().__init__() # 必須調用
        self.work_queue = work_queue
       
    def run(self):
        num = 1
        while True:
            self.work_queue.put(num)
            num = num+1
            time.sleep(1) # 暫停1秒

# 一個線程，從隊列取出數字，并顯示到終端
class Printer(threading.Thread):

    def __init__(self, work_queue):
        super().__init__() # 必須調用
        self.work_queue = work_queue

    def run(self):
        while True:
            num = self.work_queue.get() # 當隊列為空時，會阻塞，直到有數據
            print(num)

def main():
    work_queue = queue.Queue()

    producer = Producer(work_queue)
    producer.daemon = True # 當主線程退出時子線程也退出
    producer.start()

    printer = Printer(work_queue)
    printer.daemon = True # 當主線程退出時子線程也退出
    printer.start()

    work_queue.join() # 主線程會停在這里，直到所有數字被get()，并且task_done()，因為沒有調用task_done()，所在這里會一直阻塞，直到用戶按^C

if __name__ == '__main__':
    main()