七 脚本优化:重复代码的提取成函数;与项目接口相关的都封装到一个类中;添加手工验证脚本,增加输入值的判断逻辑
将所有与该项目接口相关的封装成类
class ProjectApi: #3个数据源接口
def api_1(self,code):
pass
def api_2(self,code):
pass
def api_3(self,code):
pass #数据源拼接函数
def result_of_3api(self,code):
pass #获取开发接口数据
def api_dev(self,codelist,cookie):
pass
#拼接结果与开发结果对比
def compare_dev_vs_actual(self,codelist,cookie):
pass #拼接结果与数据库对比
def compare_dev_vs_db(self):
pass
这样更方便调用,只要创建了项目接口对象,即可查看到该对象有哪些方法可调用,也方便调试。
apiobject = ProjectApi() #创建对象 #与开发接口对比
apiobject.compare_dev_vs_actual(codelist,cookie)
#与数据库对比
apiobject.compare_dev_vs_db() #协助手工测试
while True:
code= input('输入代码:')
if code.isdigit() and len(code) == 6 and code[0:2] in ('', '', ''): # 代码合法性判断
print('请求中......')
result= apiobject.result_of_3api(code)
print(result)
else:
print('代码输入有误')