Flask-sqlalchemy创建多列独立列的多列独立列

Question

我在烧瓶项目中有一个模型：

class Location(db.Model):
    __tablename__ = 'location'

    id = db.Column(db.Integer, primary_key=True)
    longitude = db.Column(db.Float, nullable=False)
    latitude = db.Column(db.Float, nullable=False)
    address_1 = db.Column(db.String(50))
    address_2 = db.Column(db.String(8))
    country_id = db.Column(db.Integer, db.ForeignKey('country.id', ondelete='SET NULL'))
    city_id = db.Column(db.Integer, db.ForeignKey('city.id', ondelete='SET NULL'))

我需要通过对除ID以外的所有列应用唯一约束来确保位置的唯一性。我已经尝试了这个：

__table_args__ = (
        db.UniqueConstraint(
            longitude,
            latitude,
            address_1,
            address_2,
            country_id,
            city_id,
            name="unique_loc"),
    )

...但是，由于可取消的addresp_1'，'adversion_2'（在更多详细信息中）。我已经进行了一些调查，发现可能解决方案：

CREATE UNIQUE INDEX unique_loc ON location
(longitude, latitude, COALESCE(address_1, ''), COALESCE(address_2, ''), country_id, city_id);

是否可以通过ORM或RAW查询编写此约束？我使用的数据库 - Postgres 13。任何帮助将不胜感激。

Answer 1

您可以使用index（带有unique = true选项）和func.coalesce：

   from sqlalchemy import func
   ...   

   __table_args__ = (
        db.Index(
            "unique_loc",
            longitude,
            latitude,
            func.coalesce(address_1, ''),
            func.coalesce(address_2, ''),
            country_id,
            city_id,
            unique=True
        ),
    )

Answer 2

因此，我发现了如何使用RAW查询创建索引。您所需要的只是用flask db迁移生成新的迁移，并修改def升级（）和'def defalgrade“：

def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    ...
    op.execute(f"CREATE UNIQUE INDEX unique_loc ON location "
               f"(longitude, latitude, COALESCE(address_1, '-1'), "
               f"COALESCE(address_2, '-1'), country_id, city_id);")
    ...
    # ### end Alembic commands ###

def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    ...
    op.drop_index('unique_loc', table_name='location')
    ...
    # ### end Alembic commands ###