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()