Flask-SQLAlchemy 小写索引 - 跳过功能，SQLAlchemy 反射不支持

Question

首先。抱歉，如果已回答此问题，但我无法在任何地方找到答案。

我需要在 Flask-SQLAlchemy 对象上定义一个小写索引。

我遇到的问题是我需要一个模型用户名和电子邮件字段以小写形式存储，以便我可以检查它们 User.query.filter(func.lower(username) == func.lower(username)).first()

到目前为止，我一直通过将这些字段插入为小写来处理这个问题，但在这个特定的例子中，我需要用户名来保留它定义的大小写。

我认为我在正确的轨道上，但遇到了一个我以前从未见过的问题。

class User(UserMixin, db.Model):
    __tablename__ = 'users'

    id = db.Column(db.Integer, primary_key=True)
    # Indexes for username and email are defined below User
    # They are functional lowercase indexes
    username = db.Column(db.String(32))
    email = db.Column(db.String(255))
    password_hash = db.Column(db.String(255))

    ...

db.Index('ix_users_username', func.lower(User.username), unique=True)
db.Index('ix_users_email', func.lower(User.email), unique=True)

我看不出这有什么问题。我猜func.lower()需要将索引定义为与我可以通过其他搜索判断的内容分开定义。

现在问题来了，当我运行迁移时，我收到以下错误：

INFO  [alembic.runtime.migration] Context impl PostgresqlImpl.
INFO  [alembic.runtime.migration] Will assume transactional DDL.
INFO  [alembic.autogenerate.compare] Detected added table 'users'
.../lib/python3.4/site-packages/alembic/util/messaging.py:69: UserWarning: autogenerate skipping functional index ix_users_email; not supported by SQLAlchemy reflection
  warnings.warn(msg)
.../lib/python3.4/site-packages/alembic/util/messaging.py:69: UserWarning: autogenerate skipping functional index ix_users_username; not supported by SQLAlchemy reflection

我不是 100% 确定为什么这不受反射支持。我期待在我的迁移中它会像以前一样添加它们，但它被包裹在较低的位置。

我不反对手动编写迁移（因为我假设它是可能的，而不是 100% 确定如何），但有人能指出为什么这不能立即使用吗？

提前致谢乔

更新

我通过在迁移中添加以下行来解决这个问题。

op.create_index('ix_users_username', 'users', [sa.text('lower(username)')])
op.create_index('ix_users_email', 'users', [sa.text('lower(email)')])

Answer 1

您需要UNIQUE INDEX使用 alembic 的execute方法手动添加。将类似的内容添加到您的迁移脚本中。

from alembic import op
# ...

def upgrade():
    # ...
    op.execute(
        """ CREATE UNIQUE INDEX users_normalized_username
            ON users
            (lower(username))
        """
    )

def downgrade():
    # ...
    op.execute("DROP INDEX users_normalized_username")

您还可以添加一个，ColumnProperty以便您可以访问username.

from sqlalchemy import Column, String, func, orm

class User(Base):
    __tablename__ = 'users'

    username = Column(String(32))
    normalized_username = orm.column_property(func.lower(username))

    # ...

这不会在您的数据库中创建额外的列，但您将能够对其进行查询和比较。

session.query(User).filter(User.normalized_username == func.lower(username)).first()