Làm cách nào để tạo cơ sở dữ liệu mới bằng SQLAlchemy?

Sử dụng SQLAlchemy, một đối tượng Engine được tạo như sau:

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

Không thể truy cập enginenếu cơ sở dữ liệu được chỉ định trong đối số create_engine(trong trường hợp này là mydb) không tồn tại. Có thể yêu cầu SQLAlchemy tạo cơ sở dữ liệu mới nếu cơ sở dữ liệu được chỉ định không tồn tại không?

Trên postgres, ba cơ sở dữ liệu thường hiển thị theo mặc định. Nếu bạn có thể kết nối với tư cách là người dùng siêu cấp (ví dụ: postgresvai trò), thì bạn có thể kết nối với postgreshoặc template1cơ sở dữ liệu. Mặc định pg_hba.conf chỉ cho phép người dùng unix có tên postgressử dụng postgresvai trò, vì vậy điều đơn giản nhất là chỉ cần trở thành người dùng đó. Ở bất kỳ mức độ nào, hãy tạo một công cụ như bình thường với người dùng có quyền tạo cơ sở dữ liệu:

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

engine.execute()Tuy nhiên, bạn không thể sử dụng vì postgres không cho phép bạn tạo cơ sở dữ liệu bên trong các giao dịch và sqlalchemy luôn cố gắng chạy các truy vấn trong một giao dịch. Để giải quyết vấn đề này, hãy lấy kết nối cơ bản từ động cơ:

>>> conn = engine.connect()

Nhưng kết nối sẽ vẫn ở bên trong một giao dịch, vì vậy bạn phải kết thúc giao dịch đang mở bằng commit:

>>> conn.execute("commit")

Và sau đó bạn có thể tiến hành tạo cơ sở dữ liệu bằng lệnh PostgreSQL thích hợp cho nó.

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

SQLAlchemy-Utils cung cấp các kiểu dữ liệu tùy chỉnh và các chức năng tiện ích khác nhau cho SQLAlchemy. Bạn có thể cài đặt phiên bản chính thức gần đây nhất bằng cách sử dụng pip:

pip install sqlalchemy-utils

Trình trợ giúp cơ sở dữ liệu bao gồm một create_databasechức năng:

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

Có thể tránh quản lý giao dịch thủ công khi tạo cơ sở dữ liệu bằng cách cung cấp isolation_level='AUTOCOMMIT'cho create_enginechức năng:

import sqlalchemy

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

Ngoài ra, nếu bạn không chắc rằng cơ sở dữ liệu không tồn tại, có một cách để bỏ qua lỗi tạo cơ sở dữ liệu do tồn tại bằng cách loại bỏ sqlalchemy.exc.ProgrammingErrorngoại lệ:

import contextlib
import sqlalchemy.exc

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

Xin lưu ý rằng tôi không thể nhận được các đề xuất ở trên database_existsbởi vì bất cứ khi nào tôi kiểm tra xem cơ sở dữ liệu có tồn tại bằng cách sử dụng hay không, database_exists(engine.url):tôi gặp lỗi này:

InterfaceError ('(pyodbc.InterfaceError) (\' 28000 \ ', u \' [28000] [Microsoft] [SQL Server Native Client 11.0] [SQL Server] Đăng nhập không thành công cho người dùng \\ 'myUser \\'. (18456) (SQLDriverConnect); [28000] [Microsoft] [SQL Server Native Client 11.0] [SQL Server] Không thể mở cơ sở dữ liệu "MY_DATABASE" do đăng nhập yêu cầu. Đăng nhập không thành công. (4060); [28000] [Microsoft] [SQL Server Native Máy khách 11.0] [Máy ​​chủ SQL] Đăng nhập không thành công cho người dùng \\ 'myUser \\'. (18456); [28000] [Microsoft] [Máy ​​khách gốc SQL 11.0] [Máy ​​chủ SQL] Không thể mở cơ sở dữ liệu "MY_DATABASE" do đăng nhập yêu cầu . Đăng nhập không thành công. (4060) \ ')',)

Cũng contextlib/suppresskhông hoạt động và tôi không sử dụng postgresvì vậy tôi đã kết thúc việc này để bỏ qua ngoại lệ nếu cơ sở dữ liệu đã tồn tại với 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))