PostgreSQL – 在Python中创建表

本文将探讨使用Python在 PostgreSQL 数据库中创建新表的过程。

在Python中创建 PostgreSQL 表的步骤

要在 PostgreSQL 数据库中创建新表，请使用以下步骤：

1. 首先，构造 CREATE TABLE 语句。
2. 接下来，通过调用 connect()函数连接到 PostgreSQL 数据库。 connect()函数返回一个连接对象。
3. 然后，通过调用连接对象的 cursor() 方法创建一个游标对象。
4. 之后，通过调用游标对象的 execute() 方法来执行 CREATE TABLE。
5. 最后，通过调用游标和连接对象的 close() 方法关闭与 PostgreSQL 数据库服务器的通信。

创建一个Python程序 -

首先，创建一个名为 create_table.py 的新文件。

其次，在 create_table.py 文件中，定义一个名为 create_tables() 的新函数。

create_tables()函数在供应商数据库中创建四个表：vendors、parts、vendor_parts 和 part_drawings。

Python3

#!/usr / bin / python
  
import psycopg2
from config import config
  
  
def create_tables():
    """ create tables in the PostgreSQL database"""
    commands = (
        """
        CREATE TABLE vendors (
            vendor_id SERIAL PRIMARY KEY,
            vendor_name VARCHAR(255) NOT NULL
        )
        """,
        """ CREATE TABLE parts (
                part_id SERIAL PRIMARY KEY,
                part_name VARCHAR(255) NOT NULL
                )
        """,
        """
        CREATE TABLE part_drawings (
                part_id INTEGER PRIMARY KEY,
                file_extension VARCHAR(5) NOT NULL,
                drawing_data BYTEA NOT NULL,
                FOREIGN KEY (part_id)
                REFERENCES parts (part_id)
                ON UPDATE CASCADE ON DELETE CASCADE
        )
        """,
        """
        CREATE TABLE vendor_parts (
                vendor_id INTEGER NOT NULL,
                part_id INTEGER NOT NULL,
                PRIMARY KEY (vendor_id, part_id),
                FOREIGN KEY (vendor_id)
                    REFERENCES vendors (vendor_id)
                    ON UPDATE CASCADE ON DELETE CASCADE,
                FOREIGN KEY (part_id)
                    REFERENCES parts (part_id)
                    ON UPDATE CASCADE ON DELETE CASCADE
        )
        """)
    conn = None
    try:
        # read the connection parameters
        params = config()
        # connect to the PostgreSQL server
        conn = psycopg2.connect(**params)
        cur = conn.cursor()
        # create table one by one
        for command in commands:
            cur.execute(command)
        # close communication with the PostgreSQL database server
        cur.close()
        # commit the changes
        conn.commit()
    except (Exception, psycopg2.DatabaseError) as error:
        print(error)
    finally:
        if conn is not None:
            conn.close()
  
  
if __name__ == '__main__':
    create_tables()

执行Python程序——

要执行Python程序，请使用以下命令：

python create_table.py

验证表创建 -

首先，使用 psql 程序登录 PostgreSQL 数据库服务器。

其次，使用 \dt 命令显示供应商数据库中的表列表。

suppliers=# \dt

             List of relations
 Schema |     Name      | Type  |  Owner
--------+---------------+-------+----------
 public | part_drawings | table | postgres
 public | parts         | table | postgres
 public | vendor_parts  | table | postgres
 public | vendors       | table | postgres
(4 rows)

从输出中可以清楚地看到，我们在供应商数据库中成功创建了四个表。

如果您使用 pgAdmin 等其他客户端工具，则可以通过公共架构下的表列表查看表。