class mockDataGenerator():
def __init__(self, language):
self.fake = Factory().create(language)
def random_phone_number(self):
return self.fake.phone_number()
def random_name(self):
return self.fake.name()
def random_address(self):
return self.fake.addres()
def random_email(self):
return self.fake.email()
def random_ipv4(self):
return self.fake.ipv4()
def random_str(self, min_chars=0, max_chars=8):
return self.fake.pystr(min_chars=min_chars, max_chars=max_chars)
def random_text(self):
return self.fake.text()
def random_timestamp(self):
return self.fake.unix_time(end_datetime=None)
def random_int(self):
return self.fake.pyint()
def random_bool(self):
return self.fake.pybool()
def random_description(self):
return self.fake.paragraph(nb_sentences=3,
variable_nb_sentences=True,
ext_word_list=None)
def random_title(self):
return self.fake.sentence(nb_words=6,
variable_nb_words=True,
ext_word_list=None)
def random_type(self):
i = random.randint(0, 10)
if i == 0:
type_ = '前端'
else:
type_ = '后端'
return type_
def get_some_things(self, faker_type: str = "" or list) -> str or int or float:
"""
根据faker type生成响应数据，并返回结果
:param faker_type: mobile, farm, name, shop, int, float
:return:
"""
data = ""
fake = Factory().create('zh_CN')
# 传入为list时,随机返回list中的一个值
if isinstance(faker_type, list):
return random.choice(faker_type)
elif isinstance(faker_type, str):
# 若传入时下列关键字,返回对应的值
record = faker_type.lower()
if record == 'mobile':
data = fake.phone_number()
self.L.info("随机生成手机号:%s" % data)
elif record == 'farm':
data = fake.company_prefix() + "的" + fake.name() + "在中国" + fake.city() + "的农场"
self.L.info("随机生成农场名:%s" % data)
elif record == 'name':
data = fake.name()
self.L.info("随机生成用户民:%s" % data)
elif record == 'shop':
data = fake.company()
self.L.info("随机生成店铺名:%s" % data)
elif record == 'integer':
data = random.randint(1, 100)
self.L.info("随机生成整数:%s" % data)
elif record == 'decimal':
data = random.uniform(1, 50)
self.L.info("随机生成小数:%s" % data)
elif record == 'text':
data = (fake.text().replace("\n", " "))[:20]
self.L.info("随机20位字符串:%s" % data)
elif record == 'address':
data = fake.address()
self.L.info("随机生成地址:%s" % data)
else:
data = faker_type
return data
# if __name__ == '__main__':
# f = KeyFaker()
# f.get_some_things('farm')
# 导入faker模块
from faker import Factory
fake = Factory().create(locale="zh_CN")
# print(dir(fake))
for i in range(1, 11):
print("第{0}次随机产生的数据:".format(i))
print("用户姓名:{0}".format(fake.name()))
print("用户名:{0}".format(fake.user_name()))
print("密码:{0}".format(fake.password()))
print("电子信箱:{0}".format(fake.email()))
print("电话号码:{0}".format(fake.phone_number()))
print("住址:{0}".format(fake.address()))
print("所在城市:{0}".format(fake.city()))
print("段落:{0}".format(fake.paragraph()))
print("文本:{0}".format(fake.text()))
print("IPv4:{0}".format(fake.ipv4()))
print("网址:{0}".format(fake.url()))
print("指定长度的字符串:{0}".format(fake.pystr(min_chars=6, max_chars=12)))
class FackData(object):
def __init__(self, locale):
self.faker = Factory().create(locale)
self.faker_funcs = dir(self.faker)
self.lock = compat.Lock()
self.auto_inc = Dict()
self.current_num = 0
######## mysql 数值类型 #############
def fake_tinyint(self, digits=None, unsigned=False):
return self.faker.random_int(
0, 255) if unsigned else self.faker.random_int(-128, 127)
def fake_smallint(self, digits=None, unsigned=False):
return self.faker.random_int(
0, 65535) if unsigned else self.faker.random_int(-32768, 32767)
def fake_mediumint(self, digits=None, unsigned=False):
return self.faker.random_int(
0, 16777215) if unsigned else self.faker.random_int(
-8388608, 8388607)
def fake_int(self, min=None, max=None, unsigned=False):
if min or max:
return self.faker.random_int(min, max)
return self.faker.random_int(
0, 4294967295) if unsigned else self.faker.random_int(
-2147483648, 2147483647)
def fake_integer(self, *args):
return self.fake_int(*args)
def fake_bigint(self, *args):
return self.faker.random_int(0, 18446744073709551615) if len(args) > 0 \
else self.faker.random_int(-9223372036854775808, 9223372036854775807)
def fake_float(self, *args):
return self.faker.pyfloat()
def fake_double(self, *args):
return self.fake_float()
def fake_decimal(self,
digits,
right_digits,
flag=None,
min_value=None,
max_value=None):
"""
mysql中DECIMAL(6,2);
最多可以存储6位数字，小数位数为2位; 因此范围是从-9999.99到9999.99
而pyfloat left_digits, right_digits 表示小数点左右数字位数
:param args:
:return:
"""
if flag is None:
flag = random.randint(0, 1)
number = self.faker.pyfloat(left_digits=(digits - right_digits),
right_digits=right_digits,
positive=True,
min_value=min_value,
max_value=max_value)
return number if flag == 1 else -number
############ mysql 日期和时间类型 ###################
def fake_date(self,
start_date='-30y',
end_date='today',
format='%Y-%m-%d'):
"""
以今天为基点，start_day, end_day两个参数，往前后推的天数
end_day默认今天
format为输出格式
:param args:
:return:
"""
thedate = self.faker.date_between(start_date, end_date)
return thedate.strftime(format)
def fake_datetime_between(self, sdt, edt, foramt='%Y-%m-%d %H:%M:%S'):
sdatetime = datetime.datetime.strptime(sdt, '%Y-%m-%d %H:%M:%S')
stimestamp = time.mktime(sdatetime.timetuple())
edatetime = datetime.datetime.strptime(edt, '%Y-%m-%d %H:%M:%S')
etimestamp = time.mktime(edatetime.timetuple())
timestamp = random.randint(stimestamp, etimestamp)
ltime = time.localtime(timestamp)
return time.strftime(foramt, ltime)
def fake_date_between(self, start_date, end_date, format='%Y-%m-%d'):
# 去掉时分秒，不然后续计算天差值会出错
start_date_time = '{0} 00:00:00'.format(start_date)
end_date_time = '{0} 23:59:59'.format(end_date)
random_datetime = self.fake_datetime_between(start_date_time,
end_date_time)
random_date = datetime.datetime.strptime(random_datetime.split()[0],
'%Y-%m-%d').date()
return datetime.datetime.strftime(random_date, format)
def fake_time(self, *args):
return self.faker.time()
def fake_year(self, *args):
return self.faker.year()
def fake_datetime(self, now=0, format='%Y-%m-%d %H:%M:%S'):
dt = datetime.datetime.now() if now else self.faker.date_time()
return dt.strftime(format)
def fake_timestamp(self, now=0):
timestamp = int(time.time()) if now else self.faker.unix_time()
return timestamp
########### mysql 字符串类型##############
def fake_char(self, *args):
return self.faker.pystr(min_chars=1, max_chars=255)
def fake_varchar(self, max_chars=255):
return self.faker.pystr(min_chars=1, max_chars=max_chars)
def fake_tinyblob(self, *args):
# TODO 待实现
return None
def fake_tinytext(self, *args):
max_nb_chars = args[0] if len(args) else 255
return self.faker.text(max_nb_chars=max_nb_chars)
def fake_text(self, *args):
max_nb_chars = args[0] if len(args) else 65535
return self.faker.text(max_nb_chars=max_nb_chars)
def fake_mediumtext(self, *args):
# TODO 待实现
return None
def fake_longtext(self, *args):
# TODO 待实现
return None
############ hive 基本数据类型 #############
def fake_number(self, digits=None, fix_len=0, positive=0):
"""
digits=None, fix_len=0, positive=0
:param digits:
:param fix_len:
:param positive:
:return:
"""
fixlen = (fix_len == 1)
val = self.faker.random_number(digits=digits, fix_len=fixlen)
if positive > 0:
val = val if val >= 0 else -val
if positive < 0:
val = val if val <= 0 else -val
return val
def fake_string(self, *args):
return self.faker.pystr(*args)
####### 定制函数 ##########
def fake_age(self, *args):
if not args:
args = [0, 100]
return self.faker.random_int(*args)
def fake_inc(self, mark, start=0, step=1):
"""
用于实现整型变量自增
:param args:
:return:
"""
with self.lock:
if mark not in self.auto_inc:
self.auto_inc[mark] = int(start)
ret = self.auto_inc[mark]
self.auto_inc[mark] += int(step)
return ret
def fake_enum(self, *args):
"""
实现枚举类型，随机返回一个列表中值
:param args: 枚举数组
:return:
"""
return random.choice(list(args))
def fake_order_enum(self, *args):
"""
用于循环顺序产生枚举值。常用于多列关联产生值
:param args: 数组值
:return:
"""
datas = list(args)
num = len(datas)
idx = (self.current_num % num) - 1
return datas[idx]
def fake_op(self, *args):
"""
实现多字段四项运算
:param args:
:return:
"""
return None
######## 执行主函数 #########
def do_fake(self, keyword, args, current_num):
"""
首先查看是否在faker类的成员函数内，如果在则调用；
否者调用FakeData类中自定义的成员函数
:param keyword:
:param args:
:return:
"""
self.current_num = current_num
method = getattr(self, 'fake_' + keyword, None)
if callable(method):
return method(*args)
if keyword in self.faker_funcs:
method = getattr(self.faker, keyword, None)
if callable(method):
return method(*args)
return None
