web-dev-qa-db-ja.com

pythonを使用してPostgresデータベースを作成します

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スクリプトを作成してこれを自動化したい。

34
kiran6

ISOLATION_LEVEL_AUTOCOMMIT 、psycopg2拡張機能を使用します。

コマンドが発行されたときにトランザクションは開始されず、commit()またはrollback()は必要ありません。

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

他の回答に示されているように、接続は自動コミットモードでなければなりません。 psycopg2を使用して設定する別の方法は、 autocommit 属性を使用することです。

import psycopg2

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

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