3

使用 django 1.8 + Postgres 9+,我有具有自定义 PG 数据类型(如 ltree)的模型。从零创建数据库失败,因为

CREATE EXTENSION ltree;

不执行。我尝试进行空迁移,但在模型创建之后运行。在创建模型之前是否存在运行 sql 的方法?

4

2 回答 2

11

我知道这个问题很久没有答案了,也许现在你已经找到了答案。但我发帖是为了以防有人从中得到一点帮助。

对于 Postgres 中可用的扩展

如果扩展是 Postgres 可用的默认值之一,那么您可以简单地创建第一个迁移,然后加载其他迁移。

# -*- coding: utf-8 -*-
from __future__ import unicode_literals
from django.contrib.postgres.operations import HStoreExtension

from django.db import migrations


class Migration(migrations.Migration):

    run_before = [
        ('some_app_that_requires_hstore', '0001_initial'),
    ]

    operations = [
        HStoreExtension(),
    ]

注意使用run_before. 它与 完全相反dependencies。或者,您可以将此迁移作为第一个迁移,并在此之后使用dependencies.

如果此迁移由于权限问题而无法创建扩展,那么您可以简单地使用 Postgres 中的superuserandnosuperuser临时为当前用户提供运行迁移的权限,例如:

ALTER ROLE user_name superuser;
# create the extension and then remove the superuser privileges.
ALTER ROLE user_name nosuperuser;

对于 Postgres 中不可用的第三方扩展

对于第三方扩展,您可以使用 run_python 为您加载扩展,例如:

from django.db import migrations


def create_third_party_extension(apps, schema_editor):
    schema_editor.execute("CREATE EXTENSION my_custom_extension;")


def drop_third_party_extension(apps, schema_editor):
    schema_editor.execute("DROP EXTENSION IF EXISTS my_custom_extension;")


class Migration(migrations.Migration):

    dependencies = [
        ('venues', '0001_auto_20180607_1014'),
    ]

    operations = [
        migrations.RunPython(create_third_party_extension, reverse_code=drop_third_party_extension, atomic=True)
]

或者,您可以将它们作为部署脚本的一部分而不是迁移。

我希望这有帮助。

于 2018-06-20T11:25:14.673 回答
0

您还可以在模板数据库上安装扩展,当创建新数据库时(如运行测试时),新数据库将复制该模板并包含扩展。

psql -d template1 -c 'create extension if not exists hstore;'
于 2020-06-23T04:06:55.627 回答