不使用fixture的Django 1.10种子数据库

时间:2017-11-14 22:49:25

标签: python django django-fixtures

所以我查看了documentation,以及SO questiondjango-seed package,但这些似乎都不符合我的目标。< / p>

基本上,我想以编程方式从外部API中种植我的Word模型,但我能找到的所有信息似乎都依赖于首先生成一个夹具,这似乎是一个不必要的步骤。

例如,在Ruby / Rails中,您可以直接写入Games并以任何方式为数据库播种。

如果Django中有类似的功能,或者我是否需要先从API生成灯具,然后导入它?

2 个答案:

答案 0 :(得分:5)

您可以为此使用数据迁移。首先为您的应用创建一个空迁移:

$ python manage.py makemigrations yourappname --empty

在空迁移中,创建一个函数来加载数据并添加migrations.RunPython操作。以下是Django documentation on migrations的修改版本:

from __future__ import unicode_literals
from django.db import migrations

def stream_from_api():
    ...

def load_data(apps, schema_editor):
    # We can't import the Person model directly as it may be a newer
    # version than this migration expects. We use the historical version.
    Person = apps.get_model('yourappname', 'Person')

    for item in stream_from_api():
        person = Person(first=item['first'], last=item['last'], age=item['age'])
        person.save()

class Migration(migrations.Migration):
    dependencies = [('yourappname', '0009_something')]
    operations = [migrations.RunPython(load_data)]

如果您有大量简单数据,则可以从批量创建方法中受益:

from __future__ import unicode_literals
from django.db import migrations

def stream_from_api():
    ...

def load_data(apps, schema_editor):
    # We can't import the Person model directly as it may be a newer
    # version than this migration expects. We use the historical version.
    Person = apps.get_model('yourappname', 'Person')

    def stream_people():
        for item in stream_from_api():
            yield Person(first=item['first'], last=item['last'], age=item['age'])

    # Adjust (or remove) the batch size depending on your needs.
    # You won't be able to use this method if your objects depend on one-another
    Person.objects.bulk_create(stream_people(), batch_size=10000)

class Migration(migrations.Migration):
    dependencies = [('yourappname', '0009_something')]
    operations = [migrations.RunPython(load_data)]

迁移具有自动包含在事务中的额外好处,因此您可以随时停止迁移,并且不会使数据库处于不一致状态。

答案 1 :(得分:1)

您是否可以在创建数据的Games模型上编写一些类方法?据推测,此方法查询外部API,将Games()对象打包为名为games的列表,然后使用Games.objects.bulk_create(games)将其插入数据库。