如何播种 Django 项目？ - 向项目中插入一堆数据进行初始化

Question

我一直在 Django 中开发，想知道是否有一种方法可以将数据播种到 Django 的数据库中。

在 ruby on rails 中，我使用 seed.rb，然后在命令行中运行“rake db:seed”。

主要原因是我想为项目初始化提供一些关于状态、类型等的数据。

有类似的东西吗？

原文由 Axil 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

与 Rails 类似，我们也可以选择为数据库做种。它是使用 管理命令 完成的。在您的一个应用程序中，使用以下文件夹结构

<project>/<app>/management/commands/seed.py

这使得 python manage.py seed 可用作管理命令。我个人遵循以下结构。

 # <project>/<app>/management/commands/seed.py
from django.core.management.base import BaseCommand
import random

# python manage.py seed --mode=refresh

""" Clear all data and creates addresses """
MODE_REFRESH = 'refresh'

""" Clear all data and do not create any object """
MODE_CLEAR = 'clear'

class Command(BaseCommand):
    help = "seed database for testing and development."

    def add_arguments(self, parser):
        parser.add_argument('--mode', type=str, help="Mode")

    def handle(self, *args, **options):
        self.stdout.write('seeding data...')
        run_seed(self, options['mode'])
        self.stdout.write('done.')

def clear_data():
    """Deletes all the table data"""
    logger.info("Delete Address instances")
    Address.objects.all().delete()

def create_address():
    """Creates an address object combining different elements from the list"""
    logger.info("Creating address")
    street_flats = ["#221 B", "#101 A", "#550I", "#420G", "#A13"]
    street_localities = ["Bakers Street", "Rajori Gardens", "Park Street", "MG Road", "Indiranagar"]
    pincodes = ["101234", "101232", "101231", "101236", "101239"]

    address = Address(
        street_flat=random.choice(street_flats),
        street_locality=random.choice(street_localities),
        pincode=random.choice(pincodes),
    )
    address.save()
    logger.info("{} address created.".format(address))
    return address

def run_seed(self, mode):
    """ Seed database based on mode

    :param mode: refresh / clear
    :return:
    """
    # Clear data from tables
    clear_data()
    if mode == MODE_CLEAR:
        return

    # Creating 15 addresses
    for i in range(15):
        create_address()

在上面的结构中，您可以添加自定义模式，并相应地播种。您还可以添加其他管理命令参数（例如 number_of_addresses 并将其传递给运行种子。命令为 python manage.py seed --mode=refresh --number_of_addresses=15 ）。

希望这可以帮助。干杯!

原文由 anurag 发布，翻译遵循 CC BY-SA 4.0 许可协议