Python PostgreSQL-创建数据库

在 Python 中使用 PostgreSQL，可以通过 psycopg2 模块实现连接并创建数据库。

连接到 PostgreSQL

要连接到 PostgreSQL，我们需要使用 psycopg2 模块建立一个连接。其中包括 PostgreSQL 服务器的地址、用户名、密码和要连接的数据库。

import psycopg2

try:
    conn = psycopg2.connect(
        dbname="testdb",
        user="postgres",
        password="password",
        host="localhost",
        port="5432"
    )
except:
    print("Unable to connect to the database")

注意替换上述代码中的 dbname、user、password、host 和 port 为你自己的值。

创建数据库

一旦连接到 PostgreSQL 服务器，就可以创建一个新的数据库。要创建数据库，我们可以使用 psycopg2 模块中的 cursor() 方法。

import psycopg2

try:
    conn = psycopg2.connect(
        dbname="testdb",
        user="postgres",
        password="password",
        host="localhost",
        port="5432"
    )
except:
    print("Unable to connect to the database")

cursor = conn.cursor()
cursor.execute("CREATE DATABASE mydatabase")

如果执行成功，将无输出。要验证数据库是否成功创建，可以使用以下命令：

cursor.execute("SELECT datname FROM pg_database WHERE datistemplate = false")
print(cursor.fetchall())

这将输出所有非模板数据库的名称。现在应该可以看到新创建的数据库。

关闭连接

完成后，我们应该关闭与 PostgreSQL 的连接。

cursor.close()
conn.close()

完整代码

import psycopg2

try:
    conn = psycopg2.connect(
        dbname="testdb",
        user="postgres",
        password="password",
        host="localhost",
        port="5432"
    )
except:
    print("Unable to connect to the database")

cursor = conn.cursor()
cursor.execute("CREATE DATABASE mydatabase")
cursor.execute("SELECT datname FROM pg_database WHERE datistemplate = false")
print(cursor.fetchall())
cursor.close()
conn.close()

结论

现在我们已经成功连接到 PostgreSQL，创建了一个新的数据库。虽然本例只是基础的演示，但它提供了一个为 Python 开发人员创建和管理 PostgreSQL 数据库的良好起点。