Come creare un nuovo database usando SQLAlchemy?

Question

Con SQLAlchemy, il motore si crea in questo modo:

from sqlalchemy import create_engine 
engine = create_engine("postgresql://localhost/mydb") 

Accesso engine fallisce se il database non è presente. È possibile indicare a SQLAlchemy di creare un nuovo database se il database specificato non esiste?

Answer 1

Su postgres, tre database sono normalmente presenti per impostazione predefinita. Se si è in grado di connettersi come superutente (ad esempio, il ruolo postgres), è possibile connettersi ai database postgres o template1. L'impostazione predefinita pg_hba.conf consente solo all'utente unix denominato postgres di utilizzare il ruolo postgres, quindi la cosa più semplice è diventare quell'utente. In ogni caso, creare un motore come al solito con un utente che dispone delle autorizzazioni per creare un database:

>>> engine = sqlalchemy.create_engine("postgres://postgres@/postgres") 

Non è possibile utilizzare engine.execute() però, perché Postgres non consente di creare database all'interno di transazioni, e SQLAlchemy cerca sempre per eseguire query in una transazione. Per aggirare il problema, ottenere il collegamento sottostante dal motore:

>>> conn = engine.connect() 

Ma la connessione sarà ancora all'interno di una transazione, in modo da avere per terminare la transazione aperta con un commit:

>>> conn.execute("commit") 

E è quindi possibile procedere alla creazione del database utilizzando il comando PostgreSQL appropriato.

>>> conn.execute("create database test") 
>>> conn.close() 

Answer 2

E 'possibile evitare la gestione delle transazioni manuale durante la creazione del database, fornendo isolation_level='AUTOCOMMIT'-create_engine funzione:

import sqlalchemy 

with sqlalchemy.create_engine(
    'postgresql:///postgres', 
    isolation_level='AUTOCOMMIT' 
).connect() as connection: 
    connection.execute('CREATE DATABASE my_database') 

anche se non si è certi che database non esiste c'è un modo per ignorare banca dati errore di creazione grazie all'esistenza sopprimendo sqlalchemy.exc.ProgrammingError eccezione:

import contextlib 
import sqlalchemy.exc 

with contextlib.suppress(sqlalchemy.exc.ProgrammingError): 
    # creating database as above 

Answer 3

Vedere http://sqlalchemy-utils.readthedocs.org/en/latest/database_helpers.html

from sqlalchemy import create_engine 
from sqlalchemy_utils import database_exists, create_database 

engine = create_engine("postgres://localhost/mydb") 
if not database_exists(engine.url): 
    create_database(engine.url) 

print(database_exists(engine.url))