def fake_links():
twitter = Link(name='Twitter', url='#')
facebook = Link(name='Facebook', url='#')
linkedin = Link(name='LinkedIn', url='#')
google = Link(name='Google+', url='#')
db.session.add_all([twitter, facebook, linkedin, google])
db.session.commit()
def __create_link(self, name, url, order):
link = Link(name=name, url=url, order=order)
link.save()
print "Created link : {} | {}".format(name, url)
return link
def fake_links(count=50):
for i in range(count):
link = Link(title=fake.sentence(),
url=fake.url(),
category=LinkCategory.query.get(
random.randint(1, LinkCategory.query.count())))
db.session.add(link)
db.session.commit()
def setUp(self):
super(AdminTestCase, self).setUp()
self.login()
category = Category(name='Default')
post = Post(title='Hello', category=category, body='Blah...')
comment = Comment(body='A comment', post=post, from_admin=True)
link = Link(name='GitHub', url='https://github.com/greyli')
db.session.add_all([category, post, comment, link])
db.session.commit()
def new_link():
form = LinkForm()
if form.validate_on_submit():
name = form.name.data
url = form.url.data
link = Link(name=name, url=url)
db.session.add(link)
db.session.commit()
flash('Link created.', 'success')
return redirect(url_for('.manage_link'))
return render_template('admin/new_link.html', form=form)
def new_link():
form = LinkForm()
if form.validate_on_submit():
name = form.name.data
url = form.url.data
if Link.query.filter_by(name=name).count() == 1:
flash("链接已存在!", "err")
link = Link(name=name, url=url)
db.session.add(link)
db.session.commit()
flash('添加链接成功!', 'ok')
oplog = Oplog(
user='******',
ip=request.remote_addr,
reason="新增链接-%s" % form.name.data,
)
db.session.add(oplog)
db.session.commit()
return redirect(url_for("admin.new_link"))
return render_template('admin/new_link.html', form=form)
