2012-10-01 41 views
3

我想要定義一個類方法直接寫入文件而不顯式關閉文件。但是,如果我回到喜歡的對象,因此:如何在我自己的類中使用`open`方法?

class sqlBuilder(object): 
    ... 

    def save_sql_stat(self, file_n, mode = 'w'): 
     try: 
      with open(file_n, mode) as sql_out: 
       return sql_out 

     except IOError, IOe: 
      print str(IOe) 

我不能夠做到:

t = sqlBuilder(table) 
out = t.save_sql_stat(sql_file) 
out.write(...) 

,因爲我要得到一個ValueError。如果不撥打out.close(),這將是一個很好的解決方法?

+3

這可能有所幫助:http://effbot.org/zone/python-with-statement.htm – mshsayem

+0

感謝您的鏈接。事實上,它揭示了這個問題... – LarsVegas

回答

5

您可以使用從contextlibclosing外移動with聲明...

from contextlib import closing 

def save_sql_stat(self, file_n, mode='w'): 
    try: 
     return closing(open(file_n, mode)) 
    except IOError as e: 
     print e.message 

sql = SqlBuilder() 
with sql.save_sql_stat('testing.sql') as sql_out: 
    pass # whatever 
+3

這就是我一直在尋找的東西。謝謝。 – LarsVegas

0

with語句是專門設計,所以這是不可能的。它應該消除諸如文件I/O之類的東西所需的正常try/finally塊的開銷。

參見:http://effbot.org/zone/python-with-statement.htm

最簡單的解決:不使用,但嘗試/除外。

相關問題