【吴师傅的Python面包店】Day 3

"""python面包师傅在出售提前准备好面包的同时，还要同时准备新面包。"""

import time
import threading

TOTAL = 0
END_FLAG = False

def preproduce():
    """提前准备好的面包。"""
    global TOTAL
    TOTAL += 10

def producer():
    """生产面包。"""
    num = 0
    global TOTAL
    global END_FLAG
    while True:
        time.sleep(1)
        TOTAL += 1
        num += 1
        if num >= 10:
            # 师傅做了10个面包就要下班了。
            END_FLAG = True    # 关门了，顾客消费也要停止
            print('Close.')
            break

def consumer():
    """消费者，买到面包总要说点什么吧。"""
    global TOTAL
    global END_FLAG
    while not END_FLAG:
        if TOTAL > 0:
            print("I am so happy.")
            TOTAL -= 1
        else:
            time.sleep(2)
            print("I am waiting!")

def run():
    preproduce()
    p = threading.Thread(target=producer)
    c = threading.Thread(target=consumer)
    p.setDaemon(True)
    p.start()
    c.start()
    p.join()
    c.join()

if __name__ == "__main__":
    run()