def test_continue_url(self):
post = Post(status=PostStatus.published, slug='test-post')
self.assertEqual(post.continue_url, None)
post.pub_date = datetime.date(2012, 4, 3)
expected = '/blog/2012/04/03/test-post#{}'.format(
self.app.config['POST_CONTINUE_LINK_FRAGMENT'])
self.assertEqual(post.continue_url, expected)
def create_post(self, user, commit=True, **kwargs):
post = Post(author=user, title=u'My First Post', slug=u'my-first-post',
body='Post body', **kwargs)
if post.status == PostStatus.published and post.pub_date is None:
post.pub_date = utc_now()
db.session.add(post)
if commit:
db.session.commit()
return post
def create_post(self, user, commit=True, **kwargs):
default_fields = {"author": user, "title": u"My First Post", "slug": u"my-first-post", "body": u"Post body"}
default_fields.update(kwargs)
post = Post(**default_fields)
if post.status == PostStatus.published and post.pub_date is None:
post.pub_date = utc_now()
db.session.add(post)
if commit:
db.session.commit()
return post
def create_post(self, user, commit=True, **kwargs):
default_fields = {
'author': user,
'title': u'My First Post',
'slug': u'my-first-post',
'body': u'Post body'
}
default_fields.update(kwargs)
post = Post(**default_fields)
if post.status == PostStatus.published and post.pub_date is None:
post.pub_date = utc_now()
db.session.add(post)
if commit:
db.session.commit()
return post
def add_post():
form = PostForm(request.form)
if form.validate_on_submit():
status = PostStatus.from_string(form.status.data)
post = Post(author=current_user, title=form.title.data,
slug=form.slug.data, body=form.body.data,
status=status, tags=form.tags.data)
if status == PostStatus.published:
post.pub_date = utc_now()
db.session.add(post)
db.session.commit()
message = u'Post "{}" successfully added.'.format(form.title.data)
flash(message, 'success')
return redirect(url_for('.view_posts'))
return render_template('blog/post_add.html', form=form)
def add_post():
form = PostForm(request.form)
if form.validate_on_submit():
status = PostStatus.from_string(form.status.data)
post = Post(author=current_user,
title=form.title.data,
slug=form.slug.data,
body=form.body.data,
status=status,
tags=form.tags.data)
if status == PostStatus.published:
post.pub_date = utc_now()
db.session.add(post)
db.session.commit()
message = u'Post "{}" successfully added.'.format(form.title.data)
flash(message, 'success')
return redirect(url_for('.view_posts'))
return render_template('blog/post_add.html', form=form)
def test_url(self):
post = Post(status=PostStatus.published, slug='test-post')
self.assertEqual(post.url, None)
post.pub_date = datetime.date(2012, 4, 3)
self.assertEqual(post.url, '/blog/2012/04/03/test-post')
