¿Cómo crear una nueva base de datos usando SQLAlchemy?

Question

Con SQLAlchemy, el motor se crea de esta manera:

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

Acceso engine falla si la base de datos no está presente. ¿Es posible decirle a SQLAlchemy que cree una nueva base de datos si la base de datos especificada no existe?

Answer 1

En postgres, tres bases de datos están normalmente presentes por defecto. Si puede conectarse como superusuario (por ejemplo, la función postgres), puede conectarse a las bases de datos postgres o template1. El valor predeterminado pg_hba.conf permite que solo el usuario de Unix llamado postgres use el rol postgres, por lo que lo más simple es convertirse en ese usuario. En cualquier caso, crear un motor como de costumbre con un usuario que tenga los permisos para crear una base de datos:

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

No puede utilizar engine.execute() sin embargo, debido a Postgres no le permiten crear bases de datos dentro de las transacciones, y sqlalchemy siempre trata para ejecutar consultas en una transacción. Para evitar esto, conseguir la conexión subyacente del motor:

>>> conn = engine.connect() 

Pero la conexión todavía estará dentro de una transacción, así que hay que poner fin a la transacción abierta con un commit:

>>> conn.execute("commit") 

Y a continuación, puede proceder a crear la base de datos utilizando el comando PostgreSQL adecuado para ello.

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

Answer 2

Es posible evitar la gestión de transacciones manuales mientras que la creación de bases de datos, proporcionando isolation_level='AUTOCOMMIT' a create_engine función:

import sqlalchemy 

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

Además, si usted no está seguro de que la base de datos no existe hay una manera de hacer caso omiso de la base de datos error de creación debido a la existencia mediante la supresión de sqlalchemy.exc.ProgrammingError excepción:

import contextlib 
import sqlalchemy.exc 

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

Answer 3

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