def auth_index(): devices = db.all() auth_info = {} for device in devices: users_and_port = device.get_auth_users_and_port() auth_info.update({device.hostname: users_and_port}) return render_template('auth.html', devices=devices, auth_info=auth_info)
def test_create_delete_vlan(self): device = db.all()[0] self.assertNotIn( (7, 'test'), device.vlans()) data = dict(VLAN_NAME="test", VLAN_ID=7) r = self.client.post('/vlan/all/vlan/create', data=data, follow_redirects=False) self.assertEqual(302, r.status_code) self.assertIn( (7, 'test'), device.vlans()) data = dict(VLAN_ID=7) r = self.client.post('/vlan/all/vlan/delete', data=data, follow_redirects=False) self.assertEqual(302, r.status_code)
def vlan_remove_all(): if request.method == 'POST': devices = db.all() vlan_id = request.form.get('VLAN_ID', None) for device in devices: try: device.vlan_remove(vlan_id) except: flash("Failed to remove vlan {} on {}".format(vlan_id, device.hostname)) return redirect(url_for('.index'))
def vlan_remove_all(): if request.method == 'POST': devices = db.all() vlan_id = request.form.get('VLAN_ID', None) for device in devices: try: device.vlan_remove(vlan_id) except: flash("Failed to remove vlan {} on {}".format( vlan_id, device.hostname)) return redirect(url_for('.index'))
def test_create_delete_vlan(self): device = db.all()[0] self.assertNotIn((7, 'test'), device.vlans()) data = dict(VLAN_NAME="test", VLAN_ID=7) r = self.client.post('/vlan/all/vlan/create', data=data, follow_redirects=False) self.assertEqual(302, r.status_code) self.assertIn((7, 'test'), device.vlans()) data = dict(VLAN_ID=7) r = self.client.post('/vlan/all/vlan/delete', data=data, follow_redirects=False) self.assertEqual(302, r.status_code)
def index(): devices = db.all() return render_template('index.html', devices=devices)