Question

我想定义一个类方法，直接写入文件而不显式关闭文件。但是，如果我像这样返回对象：

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()会有什么好的解决方法？

Answer 1

您可以使用closing中的contextlib并将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

Answer 2

with语句是专门设计的，所以这是不可能的。它应该可以消除文件I / O等所需的正常try / finally块的开销。

另请参阅：http://effbot.org/zone/python-with-statement.htm

最简单的修复：不要使用，但尝试/除外。

如何在我自己的类方法中使用`with open`？

2 个答案: