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

Question

Với SQLAlchemy, động cơ được tạo ra như thế này:

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

Truy cập engine thất bại nếu cơ sở dữ liệu là không có mặt. Có thể nói 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?

Answer 1

Trên mặt sau, ba cơ sở dữ liệu thường được hiển thị theo mặc định. Nếu bạn có thể kết nối với tư cách là siêu người dùng (ví dụ: vai trò postgres), thì bạn có thể kết nối với cơ sở dữ liệu postgres hoặc template1. Pg_hba.conf mặc định chỉ cho phép người dùng Unix có tên postgres sử dụng vai trò postgres, vì vậy điều đơn giản nhất là chỉ trở thành người dùng đó. Dù sao đi nữa, tạo ra một động cơ như bình thường với một người dùng nào có quyền truy cập để tạo ra một cơ sở dữ liệu:

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

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

>>> conn = engine.connect() 

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

>>> conn.execute("commit") 

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

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

Answer 2

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'-create_engine chứ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 chắn rằng cơ sở dữ liệu không tồn tại có một cách để bỏ qua cơ sở dữ liệu lỗi tạo do tồn tại bằng cách ngăn chặn sqlalchemy.exc.ProgrammingError ngoại lệ:

import contextlib 
import sqlalchemy.exc 

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

Answer 3

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