學習Python：導入存在

Question

我正在學習Python，我是一個超級初學者！ 我剛完成一項練習並想創建它的變體。 我的問題是，當使用IF/ELSE聲明時，我該如何避免使用「out_file = open（to_file，'w'）out_file.write（indata）」兩次使用 。 在此先感謝！

from sys import argv 
from os.path import exists 

script, from_file, to_file = argv 

print "Copying from %s to %s" % (from_file, to_file) 

in_file = open(from_file) 
indata = in_file.read() 

print "The input file is %d bytes long" % len(indata) 

if exists(to_file): 
    print "File already exists, override?" 
    raw_input() 
else: 
    out_file = open(to_file,'w') 
    out_file.write(indata) 

out_file = open(to_file,'w') 
out_file.write(indata) 

print"Done." 

out_file.close() 
in_file.close() 

Answer 1

目前您的腳本並沒有真正考慮用戶的輸入是否覆蓋，而且您不管輸入如何都會覆蓋。它看起來像你要考慮他們的意見，所以我建議是這樣的：

proceed = False 
if exists(to_file): 
    print "File already exists, override?" 
    ans = raw_input("y/n: ") 
    if ans == "y": 
     proceed = True 
else: 
    proceed = True 

if proceed: 
    out_file = open(to_file,'w') 
    out_file.write(indata) 

此外，你可能會想要做一些錯誤處理，如果from_file不存在的，因爲在那種情況下調用到open()將增加IOError - 見https://docs.python.org/2/library/functions.html#open