Создание базы данных Postgres с использованием python

Я хочу создать базу данных Postgres с помощью Python.

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 script.

Ответы

Ответ 1

Используйте ISOLATION_LEVEL_AUTOCOMMIT, расширения psycopg2:

При выполнении команды транзакция не запускается, а commit() или Требуется откат().

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))
    )

Ответ 2

Как показано в другом ответе, соединение должно быть в автоматическом режиме. Другой способ настройки с помощью psycopg2 заключается в атрибуте autocommit:

import psycopg2

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

cur = con.cursor()
cur.execute('CREATE DATABASE {};'.format(self.db_name))