使用夹具或脚本预填充数据库?

时间:2016-05-06 16:04:47

标签: python django database django-testing django-migrations

我不是专家,但我认为使用Class来定义选择并使用这些选择预填充数据库是一个好主意。我认为更容易改变选择等等

所以在models.py我有:

class City(models.Model):
    name = models.CharField(max_length=32)
    distance = models.SmallIntegerField(blank=True, null=True)
    #etc

class OtherClass(models.Model):
    name = models.CharField(max_length=32)
    #etc

class UserProfile(models.Model):
    name = models.CharField(max_length=32)
    city = models.ForeignKey(City)
    otherfield = models.ForeignKey(OtherClass)
    #etc

UserProfile是用户编译的内容,CityOtherClass是程序员提出选项的地方。

迁移后,我必须创建一些CityOtherClass个对象:它们将是选项(是的,它们必须修复)。

我刚刚了解fixtures。到现在为止我使用的是script

import os
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'sitopossedimenti.settings')

import django
django.setup()

from core.models import *

def populate():
    namecity1 = add_source('city1', None)
    namecity2 = add_source('city2', None)
    @etc

    nameotherclass1 = add_otherclass('name1', #etc)

    #etc some thousands more

def add_source(name, distance):
    s = model.Source.objects.get_or_create(name=name, distance=distance)[0]
    s.save()
    return s

def add_otherclass:
    #etc

if __name__ == '__main__':
    print ("Starting myapp population script...")
    populate()

现在剧本有效(关于),我害怕改变......但你怎么看?夹具更好吗?为什么?有差异吗?

1 个答案:

答案 0 :(得分:2)

俗话说,如果有效,就不要修理它。夹具是更常用的方法,但使用自己的方法没有任何危害。如果你正在编写一个新的测试用例,你可能想要使用fixtures,但如果我是你,我会让它成为。

如果您想要一种实现结果的全自动方式,请考虑migration.RunPython。链接文档包含一个完整示例,显示正在加载的数据。显然,./manage.py migrate会发生这种情况,而无需额外的步骤。

使用migrations.RunPython的优点是,如果您要与同事共享您的应用程序或安装在其他服务器上,所需的数据将自动加载到生产服务器中,测试也可以完全访问它在测试数据库中。