0

我正在尝试为自定义 Django 管理命令编写脚本。该脚本将创建订单(一个模型),每个订单都分配有 1 到 10 个随机数量的服务器。到目前为止,我有:

class Command(BaseCommand):
    help = 'etc...'

def add_arguments(self, parser):
    parser.add_argument('dummy-orders', type=int, help='Number of dummy-orders to create')
    parser.add_argument('servers-min', type=int, help='Minimum random amount of servers')
    parser.add_argument('servers-max', type=int, help='Maximum random amount of servers (max=10)')

def handle(self, *args, **options):
    dummy_orders = options['dummy-orders']
    servers_min = options['servers-min']
    servers_max = options['servers-max']

    assert dummy_orders <= 3000
    self.check_input_integers(servers_min, servers_max, 'Servers')

    for i in range(dummy_orders):
        servers_random = randint(servers_min, servers_max)
        #test = Order.objects.create(servers_num=servers_random)
        model = Order
        #model = Order(servers_num=servers_random)
        #user = Order.objects.get_or_create()
        print(model)

基本上,每个订单也有一个随机数量的服务器与之关联。原来没有servers_num字段,我必须在模型中添加。我面临的问题是如何简单地迭代range(dummy_orders)并为其分配随机数量的服务器。有任何想法吗?由于这仅用于测试目的,我希望我也必须以某种方式删除所有这些新条目?

4

1 回答 1

0

真的很简单,我只是创建了一个像普通的模型对象,例如(model = DummyOrder(servers_num=servers_random, luns_num=luns_random))。这似乎足以满足要求。

于 2019-01-27T09:50:00.763 回答