Python少有人走過的坑

猿友 2020-09-30 11:48:44 瀏覽數(shù) (2352)

反饋

文章來源于公眾號：Python技術(shù)，作者派森醬

毫無疑問，print函數(shù)是我們?nèi)粘Ｗ畛Ｓ玫暮瘮?shù)，無論是格式化輸出還是打印中間變量進(jìn)行調(diào)試，幾乎沒有print接不了的活兒。

但是上一次阿醬就差點(diǎn)被print給坑了。

坑從何來

最初是想要為自己的一個命令行小工具增加一個進(jìn)度顯示功能，于是用了threading模塊來實(shí)現(xiàn)多線程，一個線程用于執(zhí)行實(shí)際的邏輯，另一個線程用于打印當(dāng)前進(jìn)度。

命令行小工具增加一個進(jìn)度顯示功能

根據(jù)我們多年使用命令行的經(jīng)驗(yàn)，一般打印進(jìn)度都是在行內(nèi)打印，而 Python 的print則會默認(rèn)在結(jié)尾打印一個換行符，這就十分不美了。

不過好在，print也提供了接口來改變打印的末尾字符，通過指定print的end參數(shù)，即可改變print的打印結(jié)果。

所以我就哼哧哼哧地開干了，把打印進(jìn)度的print("#")調(diào)用改為print("#", end="")。

類似這樣：

import time
import threading


def print_sharp():
    while True:
        time.sleep(0.5)
        print("#", end="")




t1 = threading.Thread(target=print_sharp)
t1.setDaemon(True)
t1.start()


time.sleep(5)

哪成想，這么一改卻出了大問題：進(jìn)度沒法實(shí)時打印了。

也就是說，本來應(yīng)該在程序執(zhí)行期間，挨個打印出來的#號不再是聽話的、可愛的#號了，而是在整個程序執(zhí)行完成之后一次性輸出到控制臺中。

它長大了，也變丑了。

進(jìn)度沒法實(shí)時打印

那我要你有何用？

啥問題呢？

一開始阿醬以為是多線程出了問題，傻乎乎地到處找資料來“佐證”自己的各種猜測——事后想來實(shí)在太傻了，以至于現(xiàn)在說起還是會哈哈哈

這件事給我們的教訓(xùn)就是：千萬不要自以為是，而應(yīng)踏踏實(shí)實(shí)地解決問題，虛心對待每個細(xì)節(jié)。

實(shí)際上，之所以我們看不到實(shí)時的輸出，就是因?yàn)槲覀兏淖兞?code>print的結(jié)尾字符。

為了盡量減少I/O操作， Python 存在一個這樣的機(jī)制：盡量將輸出字符緩存起來，當(dāng)遇到字符串結(jié)束、換行符或強(qiáng)制刷新緩沖區(qū)時，才會一次性將緩沖區(qū)的內(nèi)容輸出到相應(yīng)的流中。

——而我們改掉的地方，就是把print默認(rèn)的換行符去掉了，所以原本每一個print都會觸發(fā)一次緩沖區(qū)刷新，變成了現(xiàn)在一直觸發(fā)不了緩沖區(qū)刷新，直到程序結(jié)束觸發(fā)一次。

好嘛，知道了啥問題，我們又吭哧吭哧找資料，聽說sys.stdout.flush可以強(qiáng)制觸發(fā)標(biāo)準(zhǔn)輸出緩沖區(qū)的刷新，于是在print后面，緊跟著又加上了sys.stdout.flush()。

誒？還真好了？

標(biāo)準(zhǔn)輸出緩沖區(qū)的刷新

這些可都是知識點(diǎn)，快記下來記下來，要考的

讓我們查看print的官方文檔，其原型為：

print(*objects, sep=' ', end='\n', file=sys.stdout, flush=False)

根據(jù)其下的描述， Python 中print的輸出是否進(jìn)行緩沖，取決于兩個參數(shù)：file和flush。

file的類型有的需要緩沖，比如sys.stdout；而有的則不需要緩沖，比如sys.stderr。

對于flush參數(shù)，當(dāng)其值為False（默認(rèn)）時，是否緩沖依賴file；而當(dāng)其值為True時，則會強(qiáng)制刷新緩沖區(qū)。

我們把示例調(diào)用中的print調(diào)用修改一下：

import sys
import time
import threading




def print_sharp():
    while True:
        time.sleep(0.5)
        print("#", end="", flush=True)




t1 = threading.Thread(target=print_sharp)
t1.setDaemon(True)
t1.start()


time.sleep(5)

進(jìn)度的實(shí)時打印