如何在 PostgreSQL 中使用循环填充数据库表？-Python教程-PHP中文网

如何在 PostgreSQL 中使用循环填充数据库表？

花韻仙語

发布： 2025-08-31 23:41:01

原创

921人浏览过

如何在 postgresql 中使用循环填充数据库表？

本文旨在介绍如何在 PostgreSQL 数据库中使用 for 循环有效地填充数据表，并避免常见的错误。文章将详细讲解循环计数器的正确使用方法，以及如何使用参数化查询来防止 SQL 注入攻击，从而确保数据安全和代码的健壮性。

使用循环向 PostgreSQL 表中插入数据

在 PostgreSQL 中，使用 for 循环可以方便地将数据批量插入到数据库表中。以下示例展示了如何使用 Python 和 psycopg2 库将一组艺术家姓名插入到名为 Artist 的表中。

关键点：循环计数器

一个常见的错误是在循环内部重置计数器变量。这会导致每次循环迭代都使用相同的值，从而违反主键约束或触发 ON CONFLICT 子句。正确的做法是在循环外部初始化计数器，并在每次迭代中递增它。

示例代码：

import psycopg2

# 数据库连接信息
conn = psycopg2.connect(
    host="your_host",
    database="your_database",
    user="your_user",
    password="your_password"
)

artist_name = ['Madonna', 'Slayer', 'Disturbed', 'Michael Jackson', 'Katty Parry']

try:
    with conn.cursor() as cur:
        id_num = 0  # 在循环外部初始化计数器
        for artists in artist_name:
            id_num += 1
            cur.execute(f"""INSERT INTO Artist (Id, Name) 
                       VALUES ('{id_num}', '{artists}') 
                       ON CONFLICT DO NOTHING""");
        conn.commit()  # 提交事务
        print("数据插入成功！")

except Exception as e:
    print(f"发生错误：{e}")
    conn.rollback() # 回滚事务

finally:
    if conn:
        cur.close()
        conn.close()

登录后复制

注意事项：

ImgCleaner

一键去除图片内的任意文字，人物和对象

220

查看详情

确保数据库连接信息正确。
conn.commit() 用于提交事务，使更改永久生效。
使用 try...except...finally 块来处理潜在的异常，并确保在发生错误时回滚事务。

防止 SQL 注入：使用参数化查询

直接使用字符串插值（例如 f-strings）构建 SQL 查询容易受到 SQL 注入攻击。即使数据源看起来是可信的，也最好使用参数化查询来避免潜在的安全风险。

示例代码：使用参数化查询

import psycopg2

# 数据库连接信息
conn = psycopg2.connect(
    host="your_host",
    database="your_database",
    user="your_user",
    password="your_password"
)

artist_name = ['Madonna', 'Slayer', 'Disturbed', 'Michael Jackson', 'Katty Parry']

try:
    with conn.cursor() as cur:
        id_num = 0
        for artists in artist_name:
            id_num += 1
            cur.execute(
                """
                INSERT INTO Artist (Id, Name) 
                VALUES (%s, %s) 
                ON CONFLICT DO NOTHING
                """,
                (id_num, artists)  # 传递参数元组
            )
        conn.commit()
        print("数据插入成功！")

except Exception as e:
    print(f"发生错误：{e}")
    conn.rollback()

finally:
    if conn:
        cur.close()
        conn.close()

登录后复制

解释：