Python递归中 return 代码陷阱

Posted 2020-10-07

最近写接口测试程序中，需要查询多层嵌套字典中某个Key的值，查到则返回，查不到返回None，查询的Key可以是任意层次的Key，如 Value本身也是多层字典，或者Key已经是叶子结点。

思想：利用Python的递归思想，逐层深入遍历，最后返回结果值

最终的成品代码参考了一下博客内容：

http://www.cnblogs.com/hahaweixiaohenqingcheng/archive/2016/11/14/6062961.html#undefined

 

尝试多次后发现参考代码已经无法再深入优化，只能照搬：

 1 #获取字典中的objkey对应的值，适用于字典嵌套
 2 #targetDict:要查找的字典;serchKey:要查找的目标key
 3 #ret:递归过程中，向外部（上层）传送 return值。被查找目标在第几层，则返回几次ret
 4 #default:查不到符合的serchKey，就返回默认值None
 5 def dict_getValue(targetDict,serchKey,default=None):
 6     for k,v in targetDict.items():
 7         if k == serchKey:
 8             return v                   
 9         elif isinstance(v,dict): 
10             ret = dict_getValue(v,serchKey,default)
11             if ret is not default: #ret与default=None不等,表示找到serchKey，则ret会作为返回值向上层返回。
12                 return ret
13     return default

View Code

测试数据，拼接在上面的代码里即可

1 if __name__ == ‘__main__‘:
2     targetDict ={"H": {"Ver": ["aaaa","bbbb"],"ACID": {‘kkk‘:"aaaaa"},"CInf": [100,2,1000]},
3                  "B": { "Login": {"Type": "LP","Name": "41SS676","Pwd": {‘aaa‘:"123456"},"ForToken": 1}}}
4     print (recursionSearch(targetDict,‘Name‘))

直接拼装上面代码即可

 

在成品之前，尝试过几种写法，都无法达到最终要求，进行了一些分析，现记录下来：

1、查找的Key只能是叶子结点，非叶子结点的无法实现查找，代码如下：

1 def recursionSearch(targetDict,serchKey):            #递归查找
2     for k,v in targetDict.items():
3         if isinstance(v,dict) :                      #值是字典元素，则递归处理
4             recursionSearch(v,serchKey)
5         elif k == serchKey:
6             pp=targetDict[k]
7             print (pp)
8             return pp

只考虑叶子结点是查询目标

结果：

print (recursionSearch(targetDict,‘kkk‘)) 看到打印出来的叶子结点的值正是我想要查找的Key=‘kkk‘的值‘aaaaaa’，

print (recursionSearch(targetDict,‘Name‘)) 换第二个分支里面的叶子结点，也能看到函数内打印结果41SS676是我想要的

应该是对的，Python输出：

41SS676
None
[Finished in 0.2s]

但为毛函数结果是None呢???

分析：

（1）.代码确实能够达到叶子结点查找到目标键名的功能，但。。。还是尝试着换换思路吧

 

2、更换if条件，不会直接到叶子结点级别才开始查找

1 def recursionSearch(targetDict,serchKey):            #递归查找
2     for k,v in targetDict.items():
3         if k == serchKey:
4             return v
5         elif isinstance(v,dict) :                      #值是字典元素，则递归处理
6             recursionSearch(v,serchKey)

任一层查找，但代码还是错误的

分析：

　　（2）.这个程序最后一行只进行了递归调用，但是没有返回递归的值，导致一旦出现递归，则必然返回断档，结果必然是None。无return的函数返回值就是None，Python规定。

　　　　　参考《Python学习手册第4版》531页 “没有renturn语句的函数”

3、那就把递归调用的返回值也return一下

1 def recursionSearch(targetDict,serchKey):            #递归查找
2     for k,v in targetDict.items():
3         if k == serchKey:
4             return v
5         elif isinstance(v,dict) :                      #值是字典元素，则递归处理
6             ret = recursionSearch(v,serchKey)
7             return ret

return递归调用的结果

 

结果：这种代码只能按照第一个元素这条线深入递归下去，无论最终找到或者找不到目标值，都会结束递归。

这种没脑子的增加return直接导致的是：

（1）.查找的Key在第一层第一个键值对的值中，且递归调用时，Key也在目标字典的第一个位置，能够返回正确值；

          如：Key=‘H‘，Key=‘ACID’，Key=‘kkk’都能返回正确值，如果Key=‘B’，Key=‘CInf’只会返回None

（2）.换句话说：for循环里只会使用第一对（k,v）

分析：

（1）.必须增加一个处理方法，让程序能够在for循环中循环下去，不能只局限在第一对（k,v）中。

　　主要就是用莫条件限制return ret是否执行，如果此return不执行，则for能继续循环下去

　　如果ret是None就继续循环，如果ret不是None就证明找到目标，应该return ret，精简之后语句：if ret is not None: return ret

正向测试：

　　print (recursionSearch(targetDict,‘H‘))                   #{‘Ver‘: [‘aaaa‘, ‘bbbb‘], ‘ACID‘: {‘kkk‘: ‘aaaaa‘}, ‘CInf‘: [100, 2, 1000]}

　　print (recursionSearch(targetDict,‘ACID‘))              #{‘kkk‘: ‘aaaaa‘}

　　print (recursionSearch(targetDict,‘kkk‘’))                #aaaaa

　　print (recursionSearch(targetDict,‘Ver‘))                 #[‘aaaa‘, ‘bbbb‘]

　　print (recursionSearch(targetDict,‘B‘))                    #{‘Login‘: {‘Type‘: ‘LP‘, ‘Name‘: ‘41SS676‘, ‘Pwd‘: {‘aaa‘: ‘123456‘}, ‘ForToken‘: 1}}

      print (recursionSearch(targetDict,‘Login‘))               #{‘Type‘: ‘LP‘, ‘Name‘: ‘41SS676‘, ‘Pwd‘: {‘aaa‘: ‘123456‘}, ‘ForToken‘: 1}

　　print (recursionSearch(targetDict,‘Pwd‘))                #{‘aaa‘: ‘123456‘}

　　print (recursionSearch(targetDict,‘aaa‘))                 #123456

以上均能返回正确目标键的值

逆向测试：

　　print (recursionSearch(targetDict,‘aaaaaa‘))  #None没有键属性是‘aaaaaa’的，只有一个键的值是‘aaaaaa’，测试函数是否是按键名查找

　　print (recursionSearch(targetDict,‘B111‘))      #None

以上均能争取返回None

 

至此，从最初级错误程序，一步一步走到正确程序。

感谢我的同事 相开征  和我不厌其烦的讨论了一天，最终弄清楚了正确程序的原理，也一步一步分析清楚错误程序错在哪里，应该如何改进。