上次通过selenium模拟浏览器注册webtours实现批量注册用户,缺点很明显:
生成速度太慢,100个用户花了几分钟时间。
为了提高效率,今天实现另一个方案,直接生成文件。
用户注册(仅填写用户名和密码)后生成的文件内容如下图所示:
写了两种方法,用timeit()随便测了一下,效率差别不大:
import timeit
def method_1():
name = ['\n', ';', '\n', '\n', '\n', ';', '\n']
for i in range(1, 2):
f = open(r'D:\output\user_date\lib\ben' + str(i), 'a')
f.write(str(i))
for j in name:
f.writelines(j)
f.close()
def method_2():
name = '\n;\n\n\n;\n'
for i in range(1, 2):
f = open(r'D:\output\user_date\lib\end' + str(i), 'w')
f.write(str(i) + name)
f.close()
print timeit.timeit(stmt = method_1, number = 10000)
print timeit.timeit(stmt = method_2, number = 10000)
一万次耗时分别如下: