SQLAlchemy с несколькими базами данных и ошибка из-за создания дубликата индекса - PullRequest
0 голосов
/ 21 марта 2019

У нас есть база данных postgres с двумя таблицами Models и Drives, и мы создаем запросы, используя sqlalchemy для анализа данных в этих таблицах.

Models Таблица имеет следующую схему:

CREATE TABLE models (
    id SERIAL PRIMARY KEY,
    vendor_name character varying(32) NOT NULL,
    model character varying(32) NOT NULL,
    drive_capacity bigint NOT NULL
);

-- Indices -------------------------------------------------------

CREATE UNIQUE INDEX models_pkey ON models(id int4_ops);
CREATE INDEX idx_models_vendor_name ON models(vendor_name text_ops);
CREATE INDEX idx_models_model ON models(model text_ops); 

и представлен классом Models в tables.py:

class Models(Base, DeferredReflection):
    __tablename__ = "models"
    id = Column("id", Integer, primary_key=True)

Drives таблицы имеют следующую схему:

CREATE TABLE drives (
    id SERIAL PRIMARY KEY,
    serial_number character varying(32) NOT NULL UNIQUE,
    model integer NOT NULL REFERENCES models(id),
    role character varying(16) NOT NULL
);

-- Indices -------------------------------------------------------

CREATE UNIQUE INDEX drives_pkey ON drives(id int4_ops);
CREATE UNIQUE INDEX drives_serial_number_key ON drives(serial_number text_ops);
CREATE INDEX drives_role_idx ON drives(role text_ops);
CREATE INDEX index_drives_model_fk ON drives(model int4_ops);

и представлены с помощьюDrives класс в tables.py:

class Drives(Base, DeferredReflection):
    __tablename__ = "drives"
    id = Column("id", Integer, primary_key=True)
    model = Column("model", Integer, ForeignKey(Models.id), nullable=False)

Как вы можете заметить, обе таблицы имеют дополнительные индексы, созданные в определенных столбцах.В нашей инфраструктуре модульных тестов мы создаем базу данных sqlite в памяти и создаем таблицы и индексы с той же схемой, что и база данных postgresql.Затем мы вставляем поддельные строки в экземпляр sqlite, запускаем наши реальные запросы поверх базы данных sqlite и проверяем правильность реальных запросов.Код выглядит следующим образом:

sqlite_engine = create_engine('sqlite:///', echo=True)
from tables import Base
Base.metadata.create_all(bind=sqlite_engine)
# add fake rows to sqlite
# run real query on sqlite engine
# compare against expected output to confirm correctness of real queries.

Однако на шаге create_all выше я получаю сообщение об ошибке.В то время как я могу создать таблицу моделей и ее индексы, я получаю ошибку индекса для таблицы drives:

cursor = <sqlite3.Cursor object at 0x7f466b877570>
statement = 'CREATE INDEX index_drives_model_fk ON drives (model)'
parameters = ()
context = <sqlalchemy.dialects.sqlite.base.SQLiteExecutionContext object at 0x7f466b2f4160>

    def do_execute(self, cursor, statement, parameters, context=None):
>       cursor.execute(statement, parameters)
E       sqlalchemy.exc.OperationalError: (sqlite3.OperationalError) index index_drives_model_fk already exists [SQL: 'CREATE INDEX index_drives_model_fk ON drives (model)'] (Background on this error at: http://sqlalche.me/e/e3q8)

С echo=True я заметил, что SQLAlchemy выдает следующие операторы:

CREATE TABLE models (
    id INTEGER NOT NULL,
    vendor_name VARCHAR(32) NOT NULL,
    model VARCHAR(32) NOT NULL,
    drive_capacity BIGINT NOT NULL,
    CONSTRAINT models_pkey PRIMARY KEY (id)
)
log.py                     110 INFO     COMMIT
log.py                     110 INFO     CREATE INDEX idx_models_model ON models (model)
log.py                     110 INFO     COMMIT
log.py                     110 INFO     CREATE INDEX idx_models_vendor_name ON models (vendor_name)
log.py                     110 INFO     COMMIT
log.py                     110 INFO
CREATE TABLE drives (
    id INTEGER NOT NULL,
    model INTEGER NOT NULL,
    serial_number VARCHAR(32) NOT NULL,
    role VARCHAR(16) NOT NULL,
    CONSTRAINT drives_pkey PRIMARY KEY (id),
    FOREIGN KEY(model) REFERENCES models (id),
    CONSTRAINT drives_serial_number_key UNIQUE (serial_number),
    CONSTRAINT drives_serial_number_key UNIQUE (serial_number)
)
log.py                     110 INFO     COMMIT
log.py                     110 INFO     CREATE INDEX index_drives_model_fk ON drives (model)
log.py                     110 INFO     COMMIT
log.py                     110 INFO     CREATE INDEX drives_role_idx ON drives (role)
log.py                     110 INFO     COMMIT
log.py                     110 INFO     CREATE INDEX index_drives_model_fk ON drives (model) <<<<<<---- Why create index_drives_model_fk twice?
log.py                     110 INFO     ROLLBACK

Если вы заметили последние 6 строк фрагмента выше, вы увидите, что index_drives_model_fk создается дважды, и, следовательно, возникает ошибка.Мой вопрос: что заставляет sqlalchemy создавать два индекса в таблице drives, но, похоже, таблица models (и связанные с ней индексы) созданы правильно?Я также могу подтвердить, если я удалю drives_role_idx и index_drives_model_fk из таблицы postgres drive, тогда все будет работать правильно, как и ожидалось.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...