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)