Django 迁移 RunSQL 以数据库类型为条件

Question

我正在尝试使用 migrations.RunSQL Django 迁移来运行一些任意 SQL 代码。我想仅针对某些数据库后端运行此迁移(例如仅针对 postgres)。

我想to use something like this但我在 Migration 类中没有看到数据库连接信息。

Answer 1

以下是我解决问题的方法，因为我无法让 RunSQL 在 RunPython 中工作。值得庆幸的是，schema_editor 对象有一个 execute() method .

def forwards(apps, schema_editor):
    if not schema_editor.connection.vendor.startswith('postgres'):
        logger.info('Database vendor: {}'.format(schema_editor.connection.vendor))
        logger.info('Skipping migration without attempting to ADD CONSTRAINT')
        return

    schema_editor.execute('ALTER TABLE my_table ADD CONSTRAINT my_constraint (my_field != \'NaN\';)')


def backwards(apps, schema_editor):
    if not schema_editor.connection.vendor.startswith('postgres'):
        logger.info('Database vendor: {}'.format(schema_editor.connection.vendor))
        logger.info('Skipping migration without attempting to DROP CONSTRAINT')
        return

    schema_editor.execute('ALTER TABLE my_table DROP CONSTRAINT my_constraint;')


class Migration(migrations.Migration):

    dependencies = [
        ...
    ]

    operations = [
        migrations.RunPython(forwards, backwards, atomic=True)
    ]