怎么在python中將sys.stdout作為默認(rèn)函數(shù)參數(shù)

本篇文章為大家展示了怎么在python中將sys.stdout作為默認(rèn)函數(shù)參數(shù)，內(nèi)容簡明扼要并且容易理解，絕對能使你眼前一亮，通過這篇文章的詳細(xì)介紹希望你能有所收獲。

虛擬函數(shù)：

import sys

def writeline(text, stream=sys.stdout):
 stream.write(text + '
')

with open('/path/to/file', 'w') as f:
 # writes to /path/to/file
 writeline('foo', f)

# writes to standard output
writeline('bar')

鑒于Python在定義時(shí)評估了函數(shù)的默認(rèn)參數(shù),將sys.stdout設(shè)置為默認(rèn)參數(shù)是安全的,還是會產(chǎn)生意想不到的副作用？

最佳答案

我想到的一個(gè)問題是,您有時(shí)有時(shí)想自己將sys.stdout重定向到文件(或管道,設(shè)備等).

例如,您的主程序可能如下所示：

if __name__ == '__main__':
 if len(sys.argv) > 1:
  sys.stdout = open(sys.argv[1],'w')
 try:
  # ... run the program
 finally:
  if len(sys.argv) > 1:
   sys.stdout.close()

如果您希望程序在提及一個(gè)文件時(shí)將其記錄到一個(gè)文件中(例如python3 file.py logfile.log),這可能會很有用.現(xiàn)在,由于您設(shè)置了sys.stdout,因此您的writeline方法將不會注意到該修改.

因此,我認(rèn)為寫這樣更安全：

def writeline(text, stream = None):
 if stream is None:
  stream = sys.stdout
 stream.write(text + '
')

通常,最好將不可變對象設(shè)置為默認(rèn)參數(shù)(例如None,False,(1)等).只有在極少數(shù)情況下,Python中才會使用不可變的變量(或可能更改引用的變量).

但是,如果您確定不會將sys.stdout重定向到文件,管道等,則是安全的.

補(bǔ)充拓展：Python 實(shí)現(xiàn)將sys.stdout還原為默認(rèn)值

我想將輸出寫入文件，因此我做了

sys.stdout = open(outfile, 'w+')

但是后來我想在寫入文件后打印回控制臺

sys.stdout.close()
sys.stdout = None

我得到了

AttributeError: 'NoneType' object has no attribute 'write'

顯然默認(rèn)輸出流不能None，所以我對Python說：

sys.stdout = use_the_default_one()

解決方案

重新分配給sys.__stdout__。

來自文檔

在程序開始時(shí)包含stdin，stderr和stdout的原始值。它們在完成過程中使用，并且無論是否已重定向sys.std *對象，將其打印到實(shí)際的標(biāo)準(zhǔn)流都很有用。

或者你可以做

print(output, file=myoutputfile)

上述內(nèi)容就是怎么在python中將sys.stdout作為默認(rèn)函數(shù)參數(shù)，你們學(xué)到知識或技能了嗎？如果還想學(xué)到更多技能或者豐富自己的知識儲備，歡迎關(guān)注億速云行業(yè)資訊頻道。