def article_update():
if 'token' not in session:
flash(u'You Need Login', 'error')
return redirect(url_for('auth.signin'))
user = User.verify_token(session['token'])
if user is None:
return redirect(url_for('auth.signin'))
form = ArticleUpdateForm()
if request.method == 'POST':
if form.validate_on_submit():
section = str(form.section.data)
section = section[section.find("'") + 1: section.find(">") - 1]
my_article = Article.find_by_id(form.id_article.data)
my_article.section_name = section
my_article.title = form.title.data
my_article.body = form.body.data
db.session.commit()
return 'update realizado'
else:
id_article = request.args.get('id')
my_article = Article.find_by_id(id_article)
if user.username != my_article.user_name:
return 'No tiene permisos'
form.id_article.data = id_article
form.user_name.data = my_article.user_name
form.title.data = my_article.title
form.body.data = my_article.body
form.section.data = my_article.section
return render_template('article/update.html', form=form)
def testarticlecreatepost(self):
user = User(
username='******',
email='*****@*****.**',
password='******',
role=1,
status=1
)
db.session.add(user)
db.session.commit()
testsection = Sections(
section_='sectionname',
description_='section created to test an article')
db.session.add(testsection)
db.session.commit()
usertest = User.query.filter(User.username == 'Testuser').first()
testarticle = Article(
title='testtitle',
body='testbody',
section_name=testsection.section_name,
section=testsection,
user_name=usertest.username,
user=usertest,
)
db.session.add(testarticle)
db.session.commit()
self.assertTrue(testarticle.find_by_id(testarticle.id) is not None)
self.assertTrue(
testarticle.find_by_author(testarticle.user_name) is not None)
self.assertTrue(
testarticle.find_by_section('testsection')is not None)
def testarticlemodelpost(self):
user = User(username="******", email="*****@*****.**", password="******", role=1, status=1)
db.session.add(user)
db.session.commit()
usertest = User.query.filter(User.username == "Testuser").first()
testarticle = Article(
title="testtitle",
body="testbody",
created="01-01-2015",
section_name="testname",
section="testsection",
user_name=usertest.username,
user=usertest,
)
db.session.add(section_todelete)
db.session.commit()
self.assertTrue(testarticle.all() != None)
self.assertTrue(testarticle.find_by_id() != None)
self.assertTrue(testarticle.find_by_author(usertest.username) != None)
self.assertTrue(testarticle.find_by_section("testsection") != None)
self.assertTrue(testarticle.slug() != None)
self.assertTrue(testarticle.create_in_words() != None)
def testarticlecreatepost(self):
user = User(
username='******',
email='*****@*****.**',
password='******',
role=1,
status=1
)
db.session.add(user)
db.session.commit()
usertest = User.query.filter(User.username == 'Testuser').first()
new_article = Article(
title='testtitle',
body='testbody',
user_name=usertest.username,
user=usertest,
)
db.session.add(new_article)
db.session.commit()
self.assertTrue(
new_article.find_by_id(new_article.id) is not None)
self.assertTrue(
new_article.find_by_author(new_article.user_name) is not None)
