蒙珣的博客

import os
from faker import Faker

# 判断文件是否存在，存在则删除
def file_exists():
    file_path = '/root/file'
    if os.path.exists(file_path):
        os.remove(file_path)

# 封装句子生成
def create_sentence(i):
    f = Faker('zh_CN')
    s = f.sentence(nb_words=i, variable_nb_words=False)
    return s

# 将句子与标签拼接，生成文本
def create_document(sentence_list):
    create_paragraph = '<audio src="询问.wav">{}</audio>{}<audio src="金额.wav">{}</audio>{}<audio src="请您在.wav" >{}</audio>{}<audio src="逾期提醒.wav" >{}</audio>\n'.format(*sentence_list)

    with open('/root/file','at',encoding='utf-8') as f:
        f.write(create_paragraph)

# 主程序
def main():
    file_exists()

    concurrent_time = int(input("请输入并发次数："))
    loop_time = int(input("请输入循环次数："))

    word_list = [21,3,15,1,18,5,33]

    for i in range(loop_time*concurrent_time):
        sentence_list = []
        for i in word_list:
            data = create_sentence(i)
            sentence_list.append(data)
        create_document(sentence_list)

main()

# 导入pymysql模块
import pymysql

# 连接database
conn = pymysql.connect(host='ip', user='root', port=3306, password='admin', db='iod',charset='utf8mb4')
# 得到一个可以执行SQL语句的光标对象
cursor = conn.cursor()
# 插入sql
sql = """INSERT INTO table(column1,column2...) VALUES (value1,value2,...);
"""

# 执行SQL语句，一次生成100W条太费时间，每1000条循环一次
for k in range(1000):
    [cursor.execute(sql) for i in range(1000)]
    print("Done "+str(k))
# 提交事务
conn.commit()
cursor.close()
conn.close()