def test_get_model_field(self): article = create_test_article(text="abc", medium=create_test_medium(name="The Guardian")) self.assertEqual(article.medium.name, "The Guardian") self.assertEqual(get_model_field(article, "medium__name"), "The Guardian") self.assertEqual(get_model_field(article, "medium"), article.medium) self.assertEqual(get_model_field(article, "text"), "abc")
def test_get_model_field(self): article = create_test_article( text="abc", medium=create_test_medium(name="The Guardian")) self.assertEqual(article.medium.name, "The Guardian") self.assertEqual(get_model_field(article, "medium__name"), "The Guardian") self.assertEqual(get_model_field(article, "medium"), article.medium) self.assertEqual(get_model_field(article, "text"), "abc")
def get_article_dict_from_model(article): for field_name in ARTICLE_FIELDS: value = get_model_field(article, field_name) if field_name in ARTICLE_CLEAN_FIELDS: value = _clean(value) yield ARTICLE_FIELD_MAP.get(field_name, field_name), value