Python 模拟数据生成模块 Faker

原帖地址： http://blog.csdn.net/huilan_same/article/details/77367293
有时候接口或UI上传入的数据需要符合指定的格式，我们在参数化的过程中又不愿意在excel中一遍遍去构造这样的数据，这时我们可以加入生成器来为我们产生符合某些固定格式的数据。

这里我推荐一个挺有意思的库，Faker，能够为你产生各种假数据

别忘了在ReadMe.md中添上你要用的库。

在utils中创建一个generator.py，用来生成数据

"""一些生成器方法，生成随机数，手机号，以及连续数字等"""
import random
from faker import Factory

fake = Factory().create('zh_CN')

def random_phone_number():
"""随机手机号"""
return fake.phone_number()

def random_name():
"""随机姓名"""
return fake.name()

def random_address():
"""随机地址"""
return fake.address()

def random_email():
"""随机email"""
return fake.email()

def random_ipv4():
"""随机IPV4地址"""
return fake.ipv4()

def random_str(min_chars=0, max_chars=8):
"""长度在最大值与最小值之间的随机字符串"""
return fake.pystr(min_chars=min_chars, max_chars=max_chars)

def factory_generate_ids(starting_id=1, increment=1):
""" 返回一个生成器函数，调用这个函数产生生成器，从starting_id开始，步长为increment。 """
def generate_started_ids():
val = starting_id
local_increment = increment
while True:
yield val
val += local_increment
return generate_started_ids

def factory_choice_generator(values):
""" 返回一个生成器函数，调用这个函数产生生成器，从给定的list中随机取一项。 """
def choice_generator():
my_list = list(values)
rand = random.Random()
while True:
yield random.choice(my_list)
return choice_generator

if __name__ == '__main__':
print(random_phone_number())
print(random_name())
print(random_address())
print(random_email())
print(random_ipv4())
print(random_str(min_chars=6, max_chars=8))
id_gen = factory_generate_ids(starting_id=0, increment=2)()
for i in range(5):
print(next(id_gen))

choices = ['John', 'Sam', 'Lily', 'Rose']
choice_gen = factory_choice_generator(choices)()
for i in range(5):
print(next(choice_gen))

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
[/code]

你还可以添加各种各样的生成器，比如指定长度中文、英文、特殊字符的字符串，指定格式的json串等等，可以省去很多构造测试数据的烦恼。

所有的代码我都放到了GITHUB上传送，可以自己下载去学习，有什么好的建议或者问题，可以留言或者加我的QQ群:455478219讨论。