请帮我生成pg 插入users 表的Python脚本

当然可以，这里是一个简单的 Python 脚本，用于向 PostgreSQL 数据库的 `users` 表插入数据。首先，你需要安装 `psycopg2` 库来处理 PostgreSQL 的连接。如果你还没有安装，可以使用 `pip install psycopg2`。

import psycopg2

# 连接到 PostgreSQL 数据库
def connect_to_db(host, database, user, password):
    conn = psycopg2.connect(
        host=host,
        database=database,
        user=user,
        password=password
    )
    return conn

# 定义用户数据字典
def create_user_data(name, email, age):
    return {
        'name': name,
        'email': email,
        'age': age
    }

# 函数来插入用户到 users 表
def insert_user(conn, data):
    cursor = conn.cursor()
    
    # 假设 users 表有如下的结构：id serial PRIMARY KEY, name varchar(100), email varchar(254), age int
    sql_query = """
        INSERT INTO users (name, email, age)
        VALUES (%s, %s, %s);
    """

    try:
        cursor.execute(sql_query, data)
        conn.commit()  # 提交事务
        print(f"User {data['name']} inserted successfully.")
    except Exception as e:
        print(f"Error occurred: {e}")
        conn.rollback()  # 如果发生错误，回滚事务

# 使用函数
if __name__ == "__main__":
    # 配置数据库连接信息
    host = "your_host"
    database = "your_database"
    user = "your_username"
    password = "your_password"

    conn = connect_to_db(host, database, user, password)

    # 创建用户数据
    user_data = create_user_data("John Doe", "john.doe@example.com", 30)

    # 插入用户
    insert_user(conn, user_data)

# 关闭连接
conn.close()