def flushdb(app):
with app.app_context():
db.drop_all()
db.create_all()
c = CIDR.create('default', 'default')
db.session.add(c)
db.session.commit()
def delete_cidr(id):
c = CIDR.get(id)
if not c:
abort(404)
if c.is_default():
abort(400)
c.delete()
return redirect(url_for('admin.cidrs_show'))
def create_cidr():
if request.method == 'GET':
return render_template('create_cidr.html')
name = request.form.get('name', default='').strip()
cidr = request.form.get('cidr', default='').strip()
c = CIDR.get_by_name(name)
if c:
flash(u'Network name already exist.', 'info')
return redirect(url_for('admin.create_cidr'))
r = CIDR.create(name, cidr)
if not r:
flash(u'创建失败', 'error')
return redirect(url_for('admin.create_cidr'))
return redirect(url_for('admin.get_cidr', id=r.id))
def edit_cidr(id):
c = CIDR.get(id)
if not c:
abort(404)
if c.is_default():
abort(400)
if request.method == 'GET':
return render_template('edit_cidr.html', cidr=c)
name = request.form.get('name', default='').strip()
cidr = request.form.get('cidr', default='').strip()
result = c.edit(name=name, cidr=cidr)
if not result:
flash("edit failure", 'error')
return redirect(url_for('admin.edit_cidr', id=c.id))
flash("edit success", 'info')
return redirect(url_for('admin.get_cidr', id=c.id))
def cidrs_show():
cidrs, total = CIDR.list_cidrs(g.start, g.limit)
return render_template('list_cidrs.html', cidrs=cidrs, total=total, endpoint='admin.cidrs_show')
def get_cidr(id):
c = CIDR.get(id)
if c:
return render_template('cidr.html', cidr=c)
flash('no cidr with this id', 'error')
return redirect(url_for('admin.cidrs_show'))
