可以使用python「with」命令來選擇性地寫入文件

我有一個腳本，它接受一組數據，並根據當時所需的內容輸入幾個不同輸出的命令行選項。目前，即使沒有任何內容寫入文件，該文件也可能因爲使用「with」而被創建。下面的簡碼代表了我的觀點。如果使用-a和-b選項調用該程序，則會根據需要生成3個具有正確輸出的文件，但如果不需要a或b，則會創建「a.output」和「b.output」文件沒有任何內容。可以使用python「with」命令來選擇性地寫入文件

import argparse 

parser = argparse.ArgumentParser(description="description") 
parser.add_argument("-a", "--a", action='store_true', default = False, help = "A type analysis") 
parser.add_argument("-b", "--b", action='store_true', default = False, help = "b type analysis") 
args = parser.parse_args() 

master_dictionary = {} 
if args.a: 
    master_dictionary["A"] = A_Function() 
if args.b: 
    master_dictionary["B"] = B_Function() 
master_dictionary["default"] = default_Function() 

with open("a.output", "w") as A, open("b.output", "w") as B, open("default.output", "w") as default: 
    for entry in master_dictionary: 
     print>>entry, printing_format_function(master_dictionary[entry])

我知道我可能崩潰的打印選項中的條件函數調用後去的，但在實際的腳本事情比較複雜，這是不太理想的把打印的分析塊內。我想要一個專門修改with命令的答案。目標是僅當包含文本時才創建「a.output」和「b.output」文件。

來源

2014-11-21 ded

我不認爲你的循環會做你認爲它會。在'master_dictionary'中輸入''''會在字符串「A」和「B」上循環，而不是名稱爲'A'和'B'的文件對象。 – BrenBarn 2014-11-21 19:34:15

您不能停止創建作爲with塊的一部分。當你做with obj as A時，obj必須存在，然後with塊才能對它做任何事情。電話open('a.output', 'w')創建該文件之前with可以有任何發言權。

您可以寫自己contextmanager會自動刪除在with塊的最後文件，但不會從擺在首位創建停止和塊中的代碼將有以某種方式手動「信號」給上下文管理器來執行刪除操作（例如，通過設置一些屬性）。

在循環內部有一個單文件with塊可能會更簡單。事情是這樣的：

for output_file in files_to_create: 
    with open(output_file, 'w') as f: 
     # write to the file

哪裏files_to_create是你進入循環，通過看你的選擇和添加文件到只有被賦予了相應的選項列表之前填充列表。但是，正如我在我的評論中指出的那樣，我認爲您嘗試處理此循環的方式還存在其他問題，因此很難確切知道代碼應該是什麼樣子。

來源

2014-11-21 19:30:02 BrenBarn

感謝'with'的解釋，它符合我的想法。我喜歡要創建的東西的列表。 – ded 2014-11-21 19:39:08

如何在沒有贊成的情況下獲得接受？但是，那麼爲什麼人們回答他們並不高興的問題...... – 2014-11-21 19:42:45

可以使用python「with」命令來選擇性地寫入文件

回答

相關問題