Python少有人走過的坑

文 |?軒轅御龍
來源:Python 技術(shù)「ID: pythonall」

毫無疑問,print函數(shù)是我們?nèi)粘W畛S玫暮瘮?shù),無論是格式化輸出還是打印中間變量進(jìn)行調(diào)試,幾乎沒有print接不了的活兒。
但是上一次阿醬就差點(diǎn)被print給坑了。
坑從何來
最初是想要為自己的一個命令行小工具增加一個進(jìn)度顯示功能,于是用了threading模塊來實(shí)現(xiàn)多線程,一個線程用于執(zhí)行實(shí)際的邏輯,另一個線程用于打印當(dāng)前進(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í)行完成之后一次性輸出到控制臺中。
它長大了,也變丑了。

那我要你有何用?

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

這件事給我們的教訓(xùn)就是:千萬不要自以為是,而應(yīng)踏踏實(shí)實(shí)地解決問題,虛心對待每個細(xì)節(jié)。
實(shí)際上,之所以我們看不到實(shí)時的輸出,就是因?yàn)槲覀兏淖兞?code style="font-family: "Operator Mono", Consolas, Monaco, Menlo, monospace;font-size: 14px;padding: 2px 4px;border-radius: 4px;margin-right: 2px;margin-left: 2px;color: rgb(53, 179, 120);background-color: rgba(27, 31, 35, 0.047);word-break: break-all;">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()。
誒?還真好了?

這些可都是知識點(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)

同樣可以實(shí)現(xiàn)進(jìn)度的實(shí)時打印。
此外,還有一種方法,在調(diào)用程序時增加一個-u選項(xiàng),也可以實(shí)現(xiàn)緩沖區(qū)的實(shí)時刷新:
$?python?-u?no_flush.py

當(dāng)然這種方法就不太推薦了,畢竟不能對程序的使用者作任何預(yù)設(shè)。
總結(jié)
本文是阿醬的一次踩坑實(shí)錄,記錄了Python中一個很少有人會遇到的奇葩問題。
總的來說,要想成為一個真正的Python程序員,只是單純掌握基本語法和一些奇技淫巧是遠(yuǎn)遠(yuǎn)不夠的,還是需要對Python本身有一定的了解。
畢竟,劍客如果不熟悉自己的劍,又該如何行走江湖呢?
PS:公號內(nèi)回復(fù)「Python」即可進(jìn)入Python 新手學(xué)習(xí)交流群,一起 100 天計(jì)劃!
老規(guī)矩,兄弟們還記得么,右下角的 “在看” 點(diǎn)一下,如果感覺文章內(nèi)容不錯的話,記得分享朋友圈讓更多的人知道!


【代碼獲取方式】
