Python少有人走過的坑

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

反饋

文章來(lái)源于公眾號(hào)：Python技術(shù)，作者派森醬

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

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

坑從何來(lái)

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

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

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

不過好在，print也提供了接口來(lái)改變打印的末尾字符，通過指定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í)時(shí)打印了。

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

它長(zhǎng)大了，也變丑了。

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

那我要你有何用？

啥問題呢？

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

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

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

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

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

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

誒？還真好了？

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

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

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

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

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

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

對(duì)于flush參數(shù)，當(dāng)其值為False（默認(rèn)）時(shí)，是否緩沖依賴file；而當(dāng)其值為True時(shí)，則會(huì)強(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í)時(shí)打印