def data_cleanup():
common = CommonUtil()
data = common.read_yaml(Element.CLEANUP_DATA)
print(data)
# 读取出来后循环执行sql
for key, values in data.items():
logging.info('==执行( {} )的数据清理开始=='.format(data[key]['dec']))
for i, sql in values.items():
if i != 'dec':
logging.info('==执行sql=={}'.format(sql))
# 判断执行影响的行数
row = Opera_DB().commit_data(sql)
if row > 0:
logging.info('清理数据条数为：{}，清理成功！'.format(row))
logging.info('==结束( {} )的数据清理=='.format(data[key]['dec']))
def post(self, request):
if not request.user.is_authenticated:
return JsonResponse({'code': '503', 'errMsg': 'caolouyaqian'})
goods_id = request.POST.get('goods_id')
print('caolouyaqian')
print(request.POST.get('goods_num'))
goods_num = int(request.POST.get('goods_num'))
cart_key = 'cart_%d' % request.user.id
conn = get_redis_connection('default')
cart_count = conn.hget(cart_key, goods_id)
if cart_count:
goods_num += int(cart_count)
conn.hset(cart_key, goods_id, goods_num)
total_count = CommonUtil.GetCartInfo(request.user.id)
return JsonResponse({
'code': 200,
'errMsg': 'caoshenhuinan',
'count': total_count
})
def get(self, request):
addr = None
try:
addr = Address.objects.filter(user_id=request.user.id,
is_default=True)
except Exception:
pass
address = ''
if len(addr) > 0:
address = addr[0]
#获取购物车数量
cart = CommonUtil.GetCartInfo(request.user.id)
#获取用户浏览记录
conn = get_redis_connection('default')
history_key = 'history_%d' % request.user.id
goods_id_list = conn.lrange(history_key, 0, -1)
goods_list = []
for goods_id in goods_id_list:
try:
goods = GoodsSKU.objects.get(id=goods_id.decode())
goods_list.append(goods)
except Goods.DoesNotExist:
print('shangpinbucunzai caolouyaqian')
return render(
request, 'userinfo.html', {
'page': 'user',
'address': address,
'cart_count': cart,
'goods_list': goods_list
})
def run():
xml_report_path = Element.REPORT_XML
html_report_path = Element.REPORT_HTML
CommonUtil().remore_filedir(html_report_path)
CommonUtil().remore_filedir(xml_report_path)
logging.info('运行测试用例')
# 定义测试集
# args = ['-s', '-q', '--alluredir', xml_report_path]
# pytest.main(args)
pytest.main()
run_allure_html(xml_report_path, html_report_path)
class GetParams:
def __init__(self):
self.common= CommonUtil()
self.cookie = OperToken().get_cookie()
self.db = Opera_DB()
self.base_url = Config().base_url
def get_params_list(self,file_name):
self.params = self.common.read_params(file_name)
_page_list = {}
for page, value in self.params.items():
parameters = value['parameters']
data_list = []
for parameter in parameters:
parameter['url'] = self.base_url + parameter['url']
parameter['header']=self.operate_header(parameter['header'])
data_list.append(parameter)
_page_list[page] = data_list
return _page_list
def operate_header(self,key):
"""
重新处理头部
:param key:
:return:
"""
if key is None:
header = {}
elif key == 'header1':
header = json.loads(Config().header1)
elif key == 'header2':
header = json.loads(Config().header2)
else:
return '头部处理错误'
header['Cookie'] = self.cookie
return header
def get_db_data(self,key,way='one'):
"""
通过关键字获取对应的sql执行后返回的数据
:param key: 关键字
:param way: 定义返回多少条数据，one一条，all是所有
:return:
"""
return self.db.select_data(self.common.get_sql(key),way)
def post(self, request):
goods_id = request.POST.get('goods_id')
conn = get_redis_connection('default')
cart_key = 'cart_%d' % request.user.id
conn.hdel(cart_key, goods_id)
total_count = CommonUtil.GetCartInfo(request.user.id)
return JsonResponse({'code': 200, 'cart_count': total_count})
def get(self, request):
default_addr = Address.objects.GetDefaultAddress(user=request.user)
address_list = Address.objects.GetAddressList(request.user.id)
cart = CommonUtil.GetCartInfo(request.user.id)
return render(
request, 'address.html', {
'page': 'address',
'address': default_addr,
'cart_count': cart,
'address_list': address_list
})
class OperToken:
def __init__(self):
self.common = CommonUtil()
def generate_cookie(self, env):
if env == 'debug':
logging.info('初始化debug用例，获取cookie')
url = Config().debug_base_url + Config().debug_loginHost
data = eval(Config().debug_loginInfo)
r = requests.post(url=url, data=data)
cookieid = requests.utils.dict_from_cookiejar(r.cookies)
cookie = "sid=" + cookieid['sid']
self.common.write_file(Element.COOKIE_FILE, cookie)
print(cookie)
logging.info('结束始化用例，已获取到cookie')
elif env == 'release':
logging.info('初始化release用例，获取cookie')
pass
else:
logging.error('环境错误')
def get_cookie(self):
return self.common.read_file(Element.COOKIE_FILE)
def session_init():
"""
数据准备
:return:
"""
env = consts.API_ENV_DEBUG
# 1、生成环境
env_key=Config().get_env(env)
CommonUtil().generate_environment(env_key)
# 2、cookie准备
OperToken().generate_cookie(env)
yield
# 后置操作
# Data_Cleanup.data_cleanup()
print(consts.TIMES_LIST)
def common_init():
return CommonUtil()
def __init__(self):
self.common= CommonUtil()
self.cookie = OperToken().get_cookie()
self.db = Opera_DB()
self.base_url = Config().base_url
def __init__(self):
self.common = CommonUtil()
