Как создать новую базу данных с помощью SQLAlchemy?

Question

С SQLAlchemy, двигатель создается следующим образом:

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

Доступ engine завершается неудачно, если база данных нет. Можно ли сказать SQLAlchemy о создании новой базы данных, если указанная база данных не существует?

Answer 1

В postgres по умолчанию обычно присутствуют три базы данных. Если вы можете подключиться как суперпользователь (например, роль postgres), вы можете подключиться к базам данных postgres или template1. По умолчанию pg_hba.conf разрешает только пользователю unix с именем postgres использовать роль postgres, поэтому проще всего просто стать этим пользователем. Во всяком случае, создать двигатель, как обычно, с пользователем, который имеет право доступа для создания базы данных:

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

Вы не можете использовать engine.execute() однако, поскольку Postgres не позволяет создавать базы данных внутри транзакций и SQLAlchemy всегда пытается для запуска запросов в транзакции. Чтобы обойти эту проблему, получите основную связь от двигателя:

>>> conn = engine.connect() 

Но связь все равно будет внутри транзакции, поэтому вы должны закончить открытую сделку с commit:

>>> conn.execute("commit") 

И вы можете приступить к созданию базы данных, используя для этого правильную команду PostgreSQL.

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

Answer 2

Это позволяет избежать ручного управления транзакциями при создании базы данных, предоставляя isolation_level='AUTOCOMMIT' для create_engine функции:

import sqlalchemy 

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

Кроме того, если вы не уверены в том, что база данных не существует, есть способ игнорировать базы данных ошибка при создании из-за существования подавляя sqlalchemy.exc.ProgrammingError исключение:

import contextlib 
import sqlalchemy.exc 

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

Answer 3

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