在未命名的管道中不再需要输出时停止程序

Question

我的python程序打印了很多输出。但我经常只需要它的开头，所以我把它管道到head。一旦完成，我怎样才能将程序更改为退出？

example.py：

import time

for i in range(100):
    # Print some output
    print("x" * 100)
    # TODO: Exit here when 'head' is done
while True:
    # continue working
    time.sleep(1)

目前，当我运行python example.py | head -n10时，终端会看到10行输出，但程序仍在运行。

如果head已经收到所需的所有输出，我怎么能退出我的python脚本？