使用 python 创建 Postgres 数据库

Question

我想使用 Python 创建 Postgres 数据库。

con = psql.connect(dbname='postgres',
      user=self.user_name, host='',
      password=self.password)

cur = con.cursor()
cur.execute("CREATE DATABASE %s  ;" % self.db_name)

我收到以下错误：

InternalError: CREATE DATABASE cannot run inside a transaction block

我正在使用 psycopg2 进行连接。我不明白这是什么问题。我想做的是连接到数据库（Postgres）：

psql -postgres -U UserName

然后创建另一个数据库：

create database test;

这是我通常做的事情，我想通过创建 Python 脚本来自动执行此操作。

Answer 1

使用ISOLATION_LEVEL_AUTOCOMMIT，一个psycopg2扩展：

No transaction is started when command are issued and no commit() or rollback() is required.

import psycopg2
from psycopg2 import sql
from psycopg2.extensions import ISOLATION_LEVEL_AUTOCOMMIT # <-- ADD THIS LINE

con = psycopg2.connect(dbname='postgres',
      user=self.user_name, host='',
      password=self.password)

con.set_isolation_level(ISOLATION_LEVEL_AUTOCOMMIT) # <-- ADD THIS LINE

cur = con.cursor()

# Use the psycopg2.sql module instead of string concatenation 
# in order to avoid sql injection attacs.
cur.execute(sql.SQL("CREATE DATABASE {}").format(
        sql.Identifier(self.db_name))
    )

Answer 2

如其他答案所示，连接必须处于自动提交模式。使用 psycopg2 设置它的另一种方法是通过 autocommit 属性：

import psycopg2
from psycopg2 import sql

con = psycopg2.connect(...)
con.autocommit = True

cur = con.cursor()
# sql.SQL and sql.Identifier are needed to avoid SQL injection attacks.
cur.execute(sql.SQL('CREATE DATABASE {};').format(
    sql.Identifier(self.db_name)))

使用 python 创建 Postgres 数据库

Create a Postgres database using python

python

postgresql

psycopg2