Come creare un nuovo database utilizzando SQLAlchemy?

Utilizzando SQLAlchemy, viene creato un oggetto Engine in questo modo:

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

L'accesso enginenon riesce se il database specificato nell'argomento a create_engine(in questo caso mydb) non esiste. È possibile dire a SQLAlchemy di creare un nuovo database se il database specificato non esiste?

Su postgres, normalmente sono presenti tre database per impostazione predefinita. Se sei in grado di connetterti come superutente (ad es. Il postgresruolo), puoi connetterti ai database postgreso template1. L'impostazione predefinita pg_hba.conf consente solo all'utente unix nominato postgresdi utilizzare il postgresruolo, quindi la cosa più semplice è diventare quell'utente. In ogni caso, crea un motore come al solito con un utente che abbia i permessi per creare un database:

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

engine.execute()Tuttavia, non è possibile utilizzarlo , poiché postgres non consente di creare database all'interno di transazioni e sqlalchemy cerca sempre di eseguire query in una transazione. Per aggirare questo problema, ottieni la connessione sottostante dal motore:

>>> conn = engine.connect()

Ma la connessione sarà ancora all'interno di una transazione, quindi devi terminare la transazione aperta con commit:

>>> conn.execute("commit")

E puoi quindi procedere alla creazione del database utilizzando il comando PostgreSQL appropriato per esso.

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

SQLAlchemy-Utils fornisce tipi di dati personalizzati e varie funzioni di utilità per SQLAlchemy. Puoi installare la versione ufficiale più recente usando pip:

pip install sqlalchemy-utils

Gli helper del database includono una create_databasefunzione:

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

È possibile evitare la gestione manuale delle transazioni durante la creazione del database fornendo isolation_level='AUTOCOMMIT'alla create_enginefunzione:

import sqlalchemy

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

Inoltre, se non sei sicuro che il database non esista, c'è un modo per ignorare l'errore di creazione del database dovuto all'esistenza sopprimendo l' sqlalchemy.exc.ProgrammingErroreccezione:

import contextlib
import sqlalchemy.exc

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

Si prega di notare che non sono riuscito a ottenere i suggerimenti di cui sopra database_existsperché ogni volta che controllo se il database esiste, in caso contrario database_exists(engine.url):ottengo questo errore:

InterfaceError ('(pyodbc.InterfaceError) (\' 28000 \ ', u \' [28000] [Microsoft] [SQL Server Native Client 11.0] [SQL Server] Accesso non riuscito per l'utente \\ 'myUser \\'. (18456) (SQLDriverConnect); [28000] [Microsoft] [SQL Server Native Client 11.0] [SQL Server] Impossibile aprire il database "MY_DATABASE" richiesto dall'account di accesso. Accesso non riuscito. (4060); [28000] [Microsoft] [SQL Server Native Client 11.0] [SQL Server] Accesso non riuscito per l'utente \\ 'myUser \\'. (18456); [28000] [Microsoft] [SQL Server Native Client 11.0] [SQL Server] Impossibile aprire il database "MY_DATABASE" richiesto dall'account di accesso . Accesso non riuscito. (4060) \ ')',)

Inoltre contextlib/suppressnon funzionava e non lo sto usando, postgresquindi ho finito per ignorare l'eccezione se il database esiste già con SQL Server:

import logging
import sqlalchemy

logging.basicConfig(filename='app.log', format='%(asctime)s-%(levelname)s-%(message)s', level=logging.DEBUG)
engine = create_engine('mssql+pyodbc://myUser:mypwd@localhost:1234/MY_DATABASE?driver=SQL+Server+Native+Client+11.0?trusted_connection=yes', isolation_level = "AUTOCOMMIT")

try: 
    engine.execute('CREATE DATABASE ' + a_database_name)
except Exception as db_exc:
    logging.exception("Exception creating database: " + str(db_exc))