132 lines
2.9 KiB
Python
132 lines
2.9 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
# @Time : 2021/11/26 0:11
|
||
# @Author : 余少琪
|
||
import re
|
||
from faker import Faker
|
||
from tools.logControl import ERROR
|
||
import datetime
|
||
import jsonpath
|
||
|
||
|
||
class Context:
|
||
def __init__(self):
|
||
self.f = Faker(locale='zh_CN')
|
||
|
||
@property
|
||
def get_phone(self) -> int:
|
||
"""
|
||
:return: 随机生成手机号码
|
||
"""
|
||
phone = self.f.phone_number()
|
||
return phone
|
||
|
||
@property
|
||
def get_id_number(self) -> int:
|
||
"""
|
||
|
||
:return: 随机生成身份证号码
|
||
"""
|
||
|
||
id_number = self.f.ssn()
|
||
return id_number
|
||
|
||
@property
|
||
def get_female_name(self) -> str:
|
||
"""
|
||
|
||
:return: 女生姓名
|
||
"""
|
||
female_name = self.f.name_male()
|
||
return female_name
|
||
|
||
@property
|
||
def get_male_name(self) -> str:
|
||
"""
|
||
|
||
:return: 男生姓名
|
||
"""
|
||
male_name = self.f.name_female()
|
||
return male_name
|
||
|
||
@property
|
||
def get_email(self) -> str:
|
||
"""
|
||
|
||
:return: 生成邮箱
|
||
"""
|
||
email = self.f.email()
|
||
return email
|
||
|
||
@property
|
||
def merchant_self_operated_shop(self) -> int:
|
||
"""
|
||
|
||
:return: 商家端自营店铺ID
|
||
"""
|
||
self_operated_shop = 515
|
||
return self_operated_shop
|
||
|
||
@property
|
||
def get_time(self) -> datetime.datetime:
|
||
"""
|
||
计算当前时间
|
||
:return:
|
||
"""
|
||
|
||
return datetime.datetime.now()
|
||
|
||
@property
|
||
def host(self) -> str:
|
||
from tools.yamlControl import GetYamlData
|
||
from config.setting import ConfigHandler
|
||
|
||
# 从配置文件conf.yaml 文件中获取到域名，然后使用正则替换
|
||
host = GetYamlData(ConfigHandler.config_path) \
|
||
.get_yaml_data()['Host']
|
||
return host
|
||
|
||
|
||
def regular(target) -> str:
|
||
"""
|
||
使用正则替换请求数据
|
||
:return:
|
||
"""
|
||
try:
|
||
regular_pattern = r'\${{(.*?)}}'
|
||
while re.findall(regular_pattern, target):
|
||
key = re.search(regular_pattern, target).group(1)
|
||
|
||
target = re.sub(regular_pattern, str(getattr(Context(), key)), target, 1)
|
||
return target
|
||
|
||
except AttributeError:
|
||
ERROR.logger.error("未找到对应的替换的数据, 请检查数据是否正确", target)
|
||
raise
|
||
|
||
|
||
def sql_json(js_path, res):
|
||
return jsonpath.jsonpath(res, js_path)[0]
|
||
|
||
|
||
def sql_regular(value, res=None):
|
||
"""
|
||
这里处理sql中的依赖数据，通过获取接口响应的jsonpath的值进行替换
|
||
:param res: jsonpath使用的返回结果
|
||
:param value:
|
||
:return:
|
||
"""
|
||
sql_json_list = re.findall(r"\$json\((.*?)\)\$", value)
|
||
|
||
for i in sql_json_list:
|
||
pattern = re.compile(r'\$json\(' + i.replace('$', "\$").replace('[', '\[') + r'\)\$')
|
||
key = str(sql_json(i, res))
|
||
value = re.sub(pattern, key, value, count=1)
|
||
value = sql_regular(value, res)
|
||
|
||
return value
|
||
|
||
|
||
if __name__ == '__main__':
|
||
pass
|