Як створити нову базу даних за допомогою SQLAlchemy?

Використовуючи SQLAlchemy, об’єкт Engine створюється так:

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

Не engineвдається отримати доступ, якщо база даних, зазначена в аргументі create_engine(у цьому випадку mydb), не існує. Чи можна сказати SQLAlchemy створити нову базу даних, якщо зазначена база даних не існує?

На 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()

SQLAlchemy-Utils надає власні типи даних та різні функції утиліти для SQLAlchemy. Ви можете встановити останню офіційну версію за допомогою pip:

pip install sqlalchemy-utils

В помічники бази даних включають в себеcreate_database функції:

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

Це дозволяє уникнути ручного управління транзакціями при створенні бази даних, надаючи 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

Зауважте, що я не міг отримати вищезазначені пропозиції, database_existsтому що коли я перевіряю, чи існує база даних, використовуючи, якщо ні, database_exists(engine.url):я отримую цю помилку:

InterfaceError ('(pyodbc.InterfaceError) (\' 28000 \ ', u \' [28000] [Microsoft] [Native Client SQL Server 11.0] [SQL Server] Не вдалося ввійти для користувача \\ 'myUser \\'. (18456) (SQLDriverConnect); [28000] [Microsoft] [SQL Server Native Client 11.0] [SQL Server] Неможливо відкрити базу даних "MY_DATABASE", яку вимагає вхід. Логін не вдалося. (4060); [28000] [Microsoft] [SQL Server Native Клієнт 11.0] [SQL Server] Не вдалось ввійти для користувача \\ 'myUser \\'. (18456); [28000] [Microsoft] [SQL Server Native Client 11.0] [SQL Server] Неможливо відкрити базу даних "MY_DATABASE", яку вимагає вхід . Вхід не вдалося. (4060) \ ')',)

Також contextlib/suppressне працював, і я не використовую, postgresтому я закінчив це робити, щоб ігнорувати виняток, якщо база даних вже існує з 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))