Python PostgreSQL - 创建表

PostgreSQL 是一种关系型数据库管理系统，提供了多种高级特性如复杂查询、事务、外键、触发器等，而 Python 则是一个优秀的脚本语言，可以方便地操作 PostgreSQL 数据库。在本文中，我们将介绍如何使用 Python 中的 psycopg2 包来创建 PostgreSQL 数据库中的表。

安装 psycopg2 包

在使用 psycopg2 前，需要先安装该包。可以使用以下命令来通过 pip 安装：

pip install psycopg2

创建数据库连接

首先，我们需要使用 psycopg2.connect() 函数来创建与 PostgreSQL 数据库的连接。在创建连接时，需要指定数据库的名称、主机地址、用户名和密码等必要的信息。

import psycopg2

try:
    connection = psycopg2.connect(
        user="postgres",
        password="password",
        host="localhost",
        port="5432",
        database="mydatabase"
    )

    cursor = connection.cursor()

except (Exception, psycopg2.Error) as error:
    print("Error while connecting to PostgreSQL:", error)

以上代码创建了一个与 PostgreSQL 数据库 "mydatabase" 的连接，并创建了一个指针通过该连接来执行 SQL 查询。

创建表

创建表的过程需要执行 SQL 语句，可以使用 psycopg2 提供的 cursor.execute() 函数来执行 SQL 语句。

create_table_query = '''
    CREATE TABLE employees
    (ID INT PRIMARY KEY     NOT NULL,
    NAME           TEXT    NOT NULL,
    AGE            INT     NOT NULL,
    ADDRESS        CHAR(50),
    SALARY         REAL);
'''

try:
    cursor.execute(create_table_query)
    connection.commit()
    print("Table created successfully in PostgreSQL")

except (Exception, psycopg2.Error) as error:
    print("Error while creating PostgreSQL table:", error)

finally:
    cursor.close()
    connection.close()

以上代码创建了一个名为 "employees" 的表，并定义了 ID、NAME、AGE、ADDRESS 和 SALARY 等字段的数据类型和属性。

总结

本文介绍了如何使用 Python 中的 psycopg2 包来创建 PostgreSQL 数据库中的表。首先，我们需要安装该包，并通过 connect() 函数创建与数据库的连接。接着，我们可以使用 cursor.execute() 函数执行 SQL 语句来创建表。最后，记得关闭连接与指针以释放资源。