faker

利用faker模拟生成数据，并写入到数据库。

先安装第三方库

pip install pymysql faker

代码如下：

import pymysql
from faker import Faker

# 初始化faker
faker = Faker('zh_CN')

# 数据库连接配置
db_config = {
    'host': 'ip',
    'user': 'user',
    'password': 'pwd',
    'database': 'db',
    'charset': 'utf8mb4',
    'cursorclass': pymysql.cursors.DictCursor
}

# 连接数据库
connection = pymysql.connect(**db_config)

# 创建表如果表不存在
def create_table():
    with connection.cursor() as cursor:
        sql = """
        CREATE TABLE IF NOT EXISTS users (
            id INT AUTO_INCREMENT PRIMARY KEY,
            name VARCHAR(50) NOT NULL,
            email VARCHAR(100) NOT NULL,
            phone VARCHAR(20),  -- 手机号字段
            address VARCHAR(255),  -- 地址字段
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
        )
        """
        cursor.execute(sql)
    connection.commit()

# 生成并插入模拟数据
def mock_data(num):
    with connection.cursor() as cursor:
        for _ in range(num):
            name = faker.name()
            email = faker.email()
            phone = faker.phone_number()
            address = faker.address()
            sql = "insert into users (name,email,phone,address) values (%s,%s,%s,%s)"
            cursor.execute(sql,(name,email,phone,address))
    connection.commit()

# 主函数
def main():
    try:
        create_table()
        mock_data(num=10)
        print("数据插入成功！")
    finally:
        connection.close()

if __name__ == "__main__":
    main()