python – 如何在我自己的类方法中使用`with 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()会有什么好的解决方法？

解决方法:

您可以使用从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

标签：file-handling,python,methods
来源： https://codeday.me/bug/20190901/1785380.html