def test_db_object_from_queryset(self):
mdh = ModelDataHelper()
with self.assertRaises(NotChoicesFound):
self.sd.db_object_from_queryset(TestRelatedModel.objects.all())
mdh.fill_model(TestRelatedModel, 10)
self.assertTrue(isinstance(
self.sd.db_object_from_queryset(TestRelatedModel.objects.all()),
TestRelatedModel
))
TestRelatedModel.objects.all().delete()
def test_db_object_string(self):
mdh = ModelDataHelper()
with self.assertRaises(NotChoicesFound):
self.sd.db_object('tests.TestRelatedModel')
mdh.fill_model(models.TestRelatedModel, 1)
self.assertTrue(isinstance(
self.sd.db_object('tests.TestRelatedModel'),
models.TestRelatedModel
))
models.TestRelatedModel.objects.all().delete()
class Command(BaseCommand):
args = ''
help = 'Example data generator'
mdh = ModelDataHelper(seed=12345678901)
def handle(self, *args, **options):
for model_conf in settings.SAMPLEDATAHELPER_MODELS:
app_label, model_name = model_conf.get('model').split(".")
instances_number = model_conf.get('number')
fields_overwrite = model_conf.get('fields_overwrite', {})
model = get_model(app_label, model_name)
self.mdh.fill_model(model, instances_number, *fields_overwrite)
