Python: 一个使用yield实现的MyRange

cjacker

class ParamError(Exception):
    def __init__(self,args=None):
        self.args=args
        
def ParamDeco(func):
    def DetectParam(*arg):
        if len(arg) > 3:
            raise ParamError("Max 3 params,Supplies {0}".format(len(arg)))
        stop=arg[int(len(arg)/2)] or 0
        step=len(arg)==3 and arg[2] or 1
        start=len(arg) >=2 and arg[0] or 0
        return func(start,stop,step)
    return DetectParam
@ParamDeco
def myRange(start,stop,step):
    i=start
    while i<stop:
        yield i
        i=i+step
        
for i in myRange(20):
    print(i)
上例是一个使用了yield实现的myRange，通过function deco来实现参数的判定和归属。其中and/or的使用还是比较巧妙的，当然用几个判断也可以实现，代码看起来可能就要冗长一点。


本文来自ChinaUnix博客，如果查看原文请点：http://blog.chinaunix.net/u2/72995/showart_1795087.html