Bắt SQLAlchemy phát hành CREATE SCHEMA trên create_all

Question

Tôi có một mô hình SQLAlchemy với một đối số giản đồ như vậy:

Base = declarative_base() 

class Road(Base): 
    __tablename__ = "roads" 
    __table_args__ = {'schema': 'my_schema'} 
    id = Column(Integer, primary_key=True) 

Khi tôi sử dụng Base.metadata.create_all (engine) nó một cách chính xác các vấn đề một CREATE TABLE với schema tên trên mặt trước như vậy "CREATE TABLE my_schema.roads (" nhưng Postgresql đúng phàn nàn rằng lược đồ không tồn tại.

Tôi thiếu một bước để có được SqlAlchemy phát hành CREATE SCHEMA my_schema hoặc tôi phải gọi điều này theo cách thủ công?

Answer 1

Tôi đã làm t theo cách thủ công trên tập lệnh init db của tôi như vậy:

from sqlalchemy.schema import CreateSchema 
engine.execute(CreateSchema('my_schema')) 

Nhưng điều này có vẻ ít huyền diệu hơn tôi mong đợi.

Answer 2

Tôi chạy vào cùng một vấn đề và tin tưởng một cách "sạch" của việc ban hành DDL là một cái gì đó như thế này:

from sqlalchemy import event 
from sqlalchemy.schema import CreateSchema 

event.listen(Base.metadata, 'before_create', CreateSchema('my_schema')) 

này sẽ đảm bảo rằng trước khi bất cứ điều gì chứa đựng trong các siêu dữ liệu của cơ sở của bạn được tạo, bạn có lược đồ cho nó. Tuy nhiên, điều này không kiểm tra xem lược đồ đã tồn tại hay chưa.

Bạn có thể làm CreateSchema('my_schema').execute_if(callback_=check_schema) nếu bạn có thể bị làm phiền khi viết lại số gọi check_schema ("Controlling DDL Sequences" trên should_create trong tài liệu). Hoặc, như một cách dễ dàng, chỉ cần sử dụng DDL("CREATE SCHEMA IF NOT EXISTS my_schema") thay vì (đối với Postgres):

from sqlalchemy import DDL 

event.listen(Base.metadata, 'before_create', DDL("CREATE SCHEMA IF NOT EXISTS my_schema"))