def test_add_article():
seed_user = User.get()
article = Article.create(source=1,
publish_date=datetime.now(),
url='www.test.com/article',
author='Tester',
title='Testing test.',
body='I am testing this article.')
def query():
return UserArticle.get(UserArticle.user == seed_user, UserArticle.article == article)
with pytest.raises(UserArticle.DoesNotExist):
query()
seed_user.add_article(article)
assert query() is not None
def post(self):
iKwargs = request.form.to_dict()
VaiLogs.info(iKwargs)
# 创建一篇新文章
newarticle = ArticleModel.create(iKwargs)
# 添加文章目录分类 只能一个目录分类
article = ArticleModel.mustFindOne(str(newarticle.get('_id')))
if 'category' in iKwargs:
category = CategoryModel.find({'name': iKwargs["category"]})
# if 'cateid' in iKwargs:
# category = CategoryModel.mustFindOne(iKwargs['cateid'])
category.addArticleModel(article)
VaiLogs.info(category)
cate = {
"mk": category.get("mk", ""),
"name": category.get("name", "")
}
article.set('category', cate)
article.save()
# 添加文章tag属性 可以多个tag属性
if 'taglist' in iKwargs:
tagList = eval(iKwargs['taglist'])
for tagid in tagList:
tag = TagModel.mustFindOne(tagid)
tag.addArticleModel(article)
article.addTagModel(tag)
resp = {
'id': str(article.get('_id', '')),
'mk': str(article.get('mk', '')),
'title': article.get('title', ''),
'summary': article.get('summary', ''),
'content': article.get('content', ''),
'category': article.get('category', ''),
'tag': article.get('tag', ''),
'createDate': article.get('createDate', ''),
'updateDate': article.get('updateDate', '')
}
return resp
def api_new():
form = request.get_json()
status, data, msgs = Article.create(form)
return iv_json_response(status, data, msgs)
