python with statement 进阶理解

本文探讨了使用Python的with语句简化文件操作的方法。通过四种不同的实现方式——函数封装、生成器、类与with结合及直接使用with语句,展示了如何更优雅地处理文件打开、读取和关闭等常见任务。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

由于之前有一个项目老是要打开文件,然后用pickle.load(file),再处理。。。最后要关闭文件,所以觉得有点繁琐,代码也不简洁。所以向python with statement寻求解决方法。

 

在网上看到一篇文章:http://effbot.org/zone/python-with-statement.htm是介绍with 的,参考着例子进行了理解。

 

如果经常有这么一些代码段的话,可以用一下几种方法改进:

代码段:

set thing up

try:

    do something

except :

    handle exception

finally:

    tear thing down

 

案例1:

假如现在要实现这么一个功能,就是打开文件,从文件里面读取数据,然后打印到终端,之后关闭文件。

那么从逻辑上来说,可以抽取“打印到终端”为数据处理部分,应该可以独立开来作为一个函数。其他像打开、关闭文件应该是一起的。

文件名为:for_test.txt

 

方法1:

用函数,把公共的部分抽取出来。

#!/usr/bin/env python

from __future__ import with_statement 

filename = 'for_test.txt'

def output(content):
    print content

#functio solution
def controlled_execution(func):
    #prepare thing
    f = None
    try:
        #set thing up
        f = open(filename, 'r')
        content = f.read()
        if not callable(func):
            return
        #deal with thing 
        func(content)

    except IOError, e:
        print 'Error %s' % str(e)

    finally:
        if f: 
            #tear thing down
            f.close()

def test():
    controlled_execution(output)

test()
 

方法2:

用yield实现一个只产生一项的generator。通过for - in 来循环。

代码片段如下:

#yield solution
def controlled_execution():
    f = None
    try:
        f = open(filename, 'r')
        thing = f.read()
        #for thing in f:
        yield thing
    except IOError,e:
        print 'Error %s' % str(e)
    finally:
        if f: 
            f.close()

def test2():
    for content in controlled_execution():
        output(content)

 

方法3:

用类的方式加上with实现。

代码片段如下:

#class solution
class controlled_execution(object):
    def __init__(self):
        self.f = None
        
    def __enter__(self):
        try:
            f = open(filename, 'r')
            content = f.read()
            return content
        except IOError ,e:
            print 'Error %s' % str(e)
            #return None

    def __exit__(self, type, value, traceback):
        if self.f:
            print 'type:%s, value:%s, traceback:%s' % \
                    (str(type), str(value), str(traceback))
            self.f.close()

def test3():
    with controlled_execution() as thing:
        if thing:
            output(thing)

 

方法4:

用with实现。不过没有exception handle 的功能。

def test4():
    with open(filename, 'r') as f:
        output(f.read())

    print f.read()

 最后一句print是用来测试f是否已经被关闭了。

 

 

 

 

    最后总结一下,写这篇文章的目的主要是受了一句话的刺激:“使用语言的好特性,不要使用那些糟糕的特性”!python真是有很多很优雅的好特性,路漫漫其修远兮,吾将上下而求索。。。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值