大家好，我是米洛，求三连！
回顾
上一节我们引入了AceEditor帮助我们在线执行/调试SQL语句，这一届我们讲点儿断言相关的内容。
数据比对
在接口测试中，我们常常需要对接口的返回参数进行校验。如果采用数据驱动的方式，涉及到多组入参/出参的比对的情况下，怎么对预期json和实际json进行一个全方位的对比呢？
今天，他来了。
其实这个话题很早以前源自于虫师selenium群，大师兄对精哥的指点，而我和饭佬，属于偷师成功的典范。
不过我在这个基础上也加了一点自己的东西，最终效果是一样的。
核心点
这个算法的核心点就在于递归，一层一层去递归，最终达到拿到所有差异的过程。
具体一点
具体一点，假设我们现在有2个json对象，分别是:
# 预期结果 a = """ { "name": "lixiaoyao", "age": 19, "wife": ["linyueru", "zhaolinger"], "job": { "yuhang": "混混", "suzhou": "林家堡姑爷", "suoyaota": "仙剑派弟子" } } """ # 实际结果 b = """ { "name": "lixiaoyao", "age": 23, "wife": ["anu", "zhaolinger"], "job": { "yuhang": "混混", "suzhou": "林家堡姑爷", "suoyaota": "仙剑派子弟" } } """
不仔细看还挺难发现这里面的差异，因为json内容不算很少，所以肉眼比较难看出来。
那么我们何不利用代码，去帮我们智能对比呢？
步步为营
确认参数
首先我们需要知道我们要比对什么，其实就是2个string，但他们是JSON格式的。
所以我们可以确定好2个基本参数: def compare(a: str, b: str)
- a 预期结果
- b 实际结果
但是我们需要深层次比对，所以我们需要额外传入一些数据: - ans
用来存放比对的信息，比如 123 != 124 - path
这个用来存放当前的路径，比如上述例子的job->suoyaota这个地方的值就不一样，一个是仙剑派弟子，另一个是仙剑派子弟，所以我们不但要记录值，还要记录他的路径。
思考
在Python中，数据结构较为简单，我们看看JSON序列化JSONDecoder类
就能大概知道了:
image
可以看到，基本上json的数据类型能够对应的，我们可以再简化一下:
能继续深入对比 和 不能继续深入对比这2种。
什么意思呢？
比如a和b的name都是lixiaoyao，lixiaoyao是个字符串，当它不为json字符串的时候，是一个不能继续深入对比的数据。
所以此时我们的递归到这一层就应该终止，直接比对a和b的name字段，如果不一样，根据path，把diff结果添加到ans中。
那什么又是可深入比较对象呢？我认为有3种:
- List
Python的数组里面可以继续遍历，里面还有可能继续有json数据，所以可继续对比。
- Dict
这个不用多说了，大家都知道这个是最容易疯狂嵌套的。
- JSON字符串
注意他其实是字符串的一种，只不过他能被反序列化为可继续遍历的对象。
编写转换为Python对象的方法
def _to_json(string): try: float(string) return string except: try: if isinstance(string, str): return json.loads(string) return string except: return string
首先我们拿到的数据是我们期望它是一个字符串，我们最先判断它是不是数值类型，如果是，直接返回这个字符串。
为什么呢？因为这个字符串如果是数值类型，那么他已经确定不可继续遍历了，我们把它原路返回。
但因为他也可能不是字符串而是Python对象比如dict或者其他数据，所以我们接着判断他是不是字符串，如果能被反序列化又不是数值的话，那说明他就是JSON字符串，如果通通不是，那我们把数据原路返回。
这一步只是为了筛选出字符串内容为JSON的数据，如果不是则直接返回之前的数据。
编写_compare核心方法
def _compare(self, a, b, ans, path): a = self._to_json(a) b = self._to_json(b) if type(a) != type(b): ans.append(f"{self._weight(path)} 类型不一致, 分别为{type(a)} {type(b)}") return if isinstance(a, dict): keys = [] for key in a.keys(): pt = path + "/" + key if key in b.keys(): self._compare(a[key], b[key], ans, pt) keys.append(key) else: ans.append(f"{self._weight(pt)} 在后者中不存在") for key in b.keys(): if key not in keys: pt = path + "/" + key ans.append(f"{self._weight(pt)} 在后者中多出") elif isinstance(a, list): i = j = 0 while i < len(a): pt = path + "/" + str(i) if j >= len(b): ans.append(f"{self._weight(pt)} 在后者中不存在") i += 1 j += 1 continue self._compare(a[i], b[j], ans, pt) i += 1 j += 1 while j < len(b): pt = path + "/" + str(j) ans.append(f"{self._weight(pt)} 在前者中不存在") j += 1 else: if a != b: ans.append( f"{self._weight(path)} 数据不一致: {JsonService._color(a)} " f"!= {self._color(b, 1)}" if path != "" else f"数据不一致: {self._color(a)} != {JsonService._color(b, 1)}")
先用_to_json转为Python对象，获得a和b。接着判断他们的类型是否一致，如果不一致则没必要继续比较了，比如一个是list，另一个是dict，根本没有比较的意义，直接ans.append错误信息即可，记得带上path参数。
self._weight是为了在html日志中更好地展示效果，加了一些style样式，可以先忽略。
如果类型也一致了，我们继续来看a是什么类型。
- 如果是字典
我们的比较是以a（预期结果）为单位的，所以一切以a为标准。
那么我们遍历a和b的keys，分别找出a字典里面有，b字典没有的key，和b字典里面有，而a字典里面没有的key。注意，这里代码可以简化，字典的keys是支持集合操作的，交由大家思考优化。
中间去遍历了a和b都有key，然后继续调用了self._compare方法，并把path改为了path+"/"+key，这样的话路径就为字典的深一层的路径了，继续递归调用。 - 如果是list
与dict其实类似，定义了2个指针，依次走完2个数组，当a数组已经走完了，b里面还有值，就把b里面剩下的值（属于多出的信息）都添加到错误信息之中。
其中也获取了新的path，只不过数组是用的索引，而dict用的是key作为路径。
接着递归。。。 - 如果不是这2种
注意这里是递归结束的条件，那我们直接比较。数据不一致，则把不一致的数据写到ans数组中。
大体思路就是这样，给大家看看color和weight。
image
测验刚才的结果
为了方便显示diff，我取消了文字高亮
可以看到age不一样，老婆不一样，锁妖塔的职业也不一样。所以，你学费了吗？
提高点
优化字典之间的key
最终源码+测试代码
import json class JsonCompare: def compare(self, exp, act): ans = [] self._compare(exp, act, ans, '') return ans def _compare(self, a, b, ans, path): a = self._to_json(a) b = self._to_json(b) if type(a) != type(b): ans.append(f"{path} 类型不一致, 分别为{type(a)} {type(b)}") return if isinstance(a, dict): keys = [] for key in a.keys(): pt = path + "/" + key if key in b.keys(): self._compare(a[key], b[key], ans, pt) keys.append(key) else: ans.append(f"{pt} 在后者中不存在") for key in b.keys(): if key not in keys: pt = path + "/" + key ans.append(f"{pt} 在后者中多出") elif isinstance(a, list): i = j = 0 while i < len(a): pt = path + "/" + str(i) if j >= len(b): ans.append(f"{pt} 在后者中不存在") i += 1 j += 1 continue self._compare(a[i], b[j], ans, pt) i += 1 j += 1 while j < len(b): pt = path + "/" + str(j) ans.append(f"{pt} 在前者中不存在") j += 1 else: if a != b: ans.append( f"{path} 数据不一致: {a} " f"!= {b}" if path != "" else f"数据不一致: {a} != {b}") def _color(self, text, _type=0): if _type == 0: # 说明是绿色 return """<span style="color: #13CE66">{}</span>""".format(text) return """<span style="color: #FF4949">{}</span>""".format(text) def _weight(self, text): return """<span style="font-weight: 700">{}</span>""".format(text) def _to_json(self, string): try: float(string) return string except: try: if isinstance(string, str): return json.loads(string) return string except: return string if __name__ == "__main__": # 预期结果 a = """ { "name": "lixiaoyao", "age": 19, "wife": ["linyueru", "zhaolinger"], "job": { "yuhang": "混混", "suzhou": "林家堡姑爷", "suoyaota": "仙剑派弟子" } } """ # 实际结果 b = """ { "name": "lixiaoyao", "age": 23, "wife": ["anu", "zhaolinger"], "job": { "yuhang": "混混", "suzhou": "林家堡姑爷", "suoyaota": "仙剑派子弟" } } """ obj = JsonCompare() ans = obj.compare(a, b) print(ans)
