def test_meta(self):
meta = get_model_meta(HorribleModel)
self.assertEqual(
meta,
{'db_table': '"my_fave"', 'verbose_name': '"Dr. Strangelove,"+"""or how I learned to stop worrying\nand love the bomb"""'},
)
def prep_for_freeze(model, last_models=None):
if last_models:
fields = last_models[model_key(model)]
else:
fields = modelsparser.get_model_fields(model, m2m=True)
# Remove useless attributes (like 'choices')
for name, field in fields.items():
fields[name] = remove_useless_attributes(field)
# See if there's a Meta
if last_models:
meta = last_models[model_key(model)].get("Meta", {})
else:
meta = modelsparser.get_model_meta(model)
if meta:
fields['Meta'] = remove_useless_meta(meta)
return fields
def prep_for_freeze(model, last_models=None):
if last_models:
fields = last_models[model_key(model)]
else:
fields = modelsparser.get_model_fields(model, m2m=True)
# Remove useless attributes (like 'choices')
for name, field in fields.items():
fields[name] = remove_useless_attributes(field)
# See if there's a Meta
if last_models:
meta = last_models[model_key(model)].get("Meta", {})
else:
meta = modelsparser.get_model_meta(model)
if meta:
fields['Meta'] = remove_useless_meta(meta)
return fields
def prep_for_stub(model, last_models=None):
if last_models:
fields = last_models[model_key(model)]
else:
fields = modelsparser.get_model_fields(model)
# Now, take only the PK (and a 'we're a stub' field) and freeze 'em
pk = model._meta.pk.name
fields = {
pk: ormise_triple(model._meta.pk, remove_useless_attributes(fields[pk])),
"_stub": True,
}
# Meta is important too.
if last_models:
meta = last_models[model_key(model)].get("Meta", {})
else:
meta = modelsparser.get_model_meta(model)
if meta:
fields['Meta'] = remove_useless_meta(meta)
return fields
def prep_for_stub(model, last_models=None):
if last_models:
fields = last_models[model_key(model)]
else:
fields = modelsparser.get_model_fields(model)
# Now, take only the PK (and a 'we're a stub' field) and freeze 'em
pk = model._meta.pk.name
fields = {
pk: remove_useless_attributes(fields[pk]),
"_stub": True,
}
# Meta is important too.
if last_models:
meta = last_models[model_key(model)].get("Meta", {})
else:
meta = modelsparser.get_model_meta(model)
if meta:
fields['Meta'] = remove_useless_meta(meta)
return fields
