mirror of
https://github.com/TakeusV1/flask_ipam.git
synced 2026-10-13 17:15:33 +02:00
241 lines
9.0 KiB
Python
241 lines
9.0 KiB
Python
from flask import Blueprint, request, render_template, flash, redirect, url_for, abort
|
|
from flask_login import login_required, current_user
|
|
|
|
from app.extensions import *
|
|
from app.models import *
|
|
from app.forms import NewInventoryItem, ChangeInventoryItem
|
|
|
|
inventory = Blueprint('inventory', __name__)
|
|
|
|
@inventory.route('/add', methods=['GET','POST'])
|
|
@login_required
|
|
def add_item():
|
|
if current_user.is_reado:
|
|
return abort(403)
|
|
|
|
form = NewInventoryItem()
|
|
if form.validate_on_submit() and request.method == 'POST':
|
|
add_net = Inventory(host_type=int(form.item_type.data),host_name=str(form.item_name.data),host_desc=str(form.item_desc.data))
|
|
db.session.add(add_net)
|
|
db.session.commit()
|
|
flash("Item ADDED", 'success')
|
|
|
|
return render_template('panel/inventory_add.html', title='ADD ITEM', form=form, navcolor='success')
|
|
|
|
@inventory.route('/list/<int:item_type>', methods=['GET','POST'])
|
|
@login_required
|
|
def list_items(item_type):
|
|
if item_type == 5:
|
|
db_items = Inventory.query.all()
|
|
else:
|
|
db_items = Inventory.query.filter_by(host_type=item_type).all()
|
|
|
|
if item_type == 0:
|
|
title = 'Others'
|
|
elif item_type == 1:
|
|
title = 'Computers'
|
|
elif item_type == 2:
|
|
title = 'Servers'
|
|
elif item_type == 3:
|
|
title = 'VM\'s'
|
|
elif item_type == 4:
|
|
title = 'Network'
|
|
else:
|
|
title = 'All'
|
|
|
|
return render_template('panel/inventory_view.html', title=title, item_list=db_items, current_page=item_type, navcolor='success')
|
|
|
|
@inventory.route('/action/<int:item_id>/<int:action>', methods=['GET','POST'])
|
|
@login_required
|
|
def action_item(item_id,action):
|
|
if current_user.is_reado:
|
|
return abort(403)
|
|
|
|
if action == 1:
|
|
db_item = Inventory.query.filter_by(id=item_id).first()
|
|
db_allocs = Allocation.query.filter_by(device_id=db_item.id).all()
|
|
# Delete Link Allocation(s)
|
|
if db_allocs != []:
|
|
for alloc in db_allocs:
|
|
alloc.device_id = None
|
|
|
|
db.session.delete(db_item)
|
|
db.session.commit()
|
|
|
|
return redirect(url_for('inventory.list_items',item_type=request.args.get('page', type=int)))
|
|
|
|
## MODAL ROUTES
|
|
|
|
@inventory.get('/<int:item_id>/view_modal')
|
|
@login_required
|
|
def item_view_modal(item_id):
|
|
db_item = Inventory.query.filter_by(id=item_id).first()
|
|
modal = {
|
|
'title': f'Informations of <b>{db_item.host_name}</b> ({db_item.id})',
|
|
'modal_type': 'infos',
|
|
'color': 'info',
|
|
'action': 'delete',
|
|
'action_text': 'Delete',
|
|
'action_url': url_for('inventory.action_item',item_id=item_id,action=1)
|
|
}
|
|
|
|
if db_item.host_type == 0:
|
|
host_type = 'Others'
|
|
elif db_item.host_type == 1:
|
|
host_type = 'Computers'
|
|
elif db_item.host_type == 2:
|
|
host_type = 'Servers'
|
|
elif db_item.host_type == 3:
|
|
host_type = 'VM\'s'
|
|
elif db_item.host_type == 4:
|
|
host_type = 'Network'
|
|
else:
|
|
host_type = 'Error'
|
|
|
|
informations = [
|
|
('Type',host_type,'col-6'),
|
|
('Name',db_item.host_name,'col-6'),
|
|
('Description',db_item.host_desc,'col-6'),
|
|
('Model',db_item.host_model,'col-6'),
|
|
('Manufacturer',db_item.host_manufacturer,'col-6'),
|
|
('SerialNumber',db_item.host_serialnumber,'col-6'),
|
|
('OS',db_item.host_os,'col-6'),
|
|
('Location',db_item.host_location,'col-6'),
|
|
('Owner',db_item.host_owner,'col-6'),
|
|
('Contact',db_item.host_contact,'col-6'),
|
|
('Allocations',db_item.host_allocations,'col-12'),
|
|
('Notes',db_item.host_notes,'col-12'),
|
|
]
|
|
|
|
return render_template('panel/_modal.html', modal=modal, item_id=item_id, informations=informations, old_page=request.args.get('page', type=int), pagination_active=True)
|
|
|
|
@inventory.get('/<int:item_id>/del_modal')
|
|
@login_required
|
|
def item_del_modal(item_id):
|
|
db_item = Inventory.query.filter_by(id=item_id).first()
|
|
modal = {
|
|
'title': 'Delete Item',
|
|
'body': f'Are you sure you want to delete <b>{db_item.host_name}</b> ?',
|
|
'color': 'danger',
|
|
'action': 'delete',
|
|
'action_text': 'Delete',
|
|
'action_url': url_for('inventory.action_item',item_id=item_id,action=1)
|
|
}
|
|
return render_template('panel/_modal.html', modal=modal, item_id=item_id, old_page=request.args.get('page', type=int), pagination_active=True)
|
|
|
|
@inventory.route('/<int:item_id>/edit_modal', methods=['GET','POST'])
|
|
@login_required
|
|
def item_edit_modal(item_id):
|
|
if current_user.is_reado:
|
|
return abort(403)
|
|
|
|
db_item = Inventory.query.filter_by(id=item_id).first()
|
|
|
|
form = ChangeInventoryItem()
|
|
if form.validate_on_submit() and request.method == 'POST':
|
|
|
|
db_item.host_type = form.item_type.data
|
|
db_item.host_name = form.item_name.data
|
|
# Description
|
|
if form.item_desc.data != '' and form.item_desc.data != 'None':
|
|
db_item.host_desc = form.item_desc.data
|
|
else:
|
|
db_item.host_desc = None
|
|
# Model
|
|
if form.item_model.data != '' and form.item_model.data != 'None':
|
|
db_item.host_model = form.item_model.data
|
|
else:
|
|
db_item.host_model = None
|
|
# Manufacturer
|
|
if form.item_manufacturer.data != '' and form.item_manufacturer.data != 'None':
|
|
db_item.host_manufacturer = form.item_manufacturer.data
|
|
else:
|
|
db_item.host_serialnumber = None
|
|
# Serial Number
|
|
if form.item_serial_number.data != '' and form.item_serial_number.data != 'None':
|
|
db_item.host_serialnumber = form.item_serial_number.data
|
|
else:
|
|
db_item.host_serialnumber = None
|
|
# OS
|
|
if form.item_os.data != '' and form.item_os.data != 'None':
|
|
db_item.host_os = form.item_os.data
|
|
else:
|
|
db_item.host_os = None
|
|
# Location
|
|
if form.item_location.data != '' and form.item_location.data != 'None':
|
|
db_item.host_location = form.item_location.data
|
|
else:
|
|
db_item.host_location = None
|
|
# Owner
|
|
if form.item_owner.data != '' and form.item_owner.data != 'None':
|
|
db_item.host_owner = form.item_owner.data
|
|
else:
|
|
db_item.host_owner = None
|
|
# Contact
|
|
if form.item_contact.data != '' and form.item_contact.data != 'None':
|
|
db_item.host_contact = form.item_contact.data
|
|
else:
|
|
db_item.host_contact = None
|
|
# Notes
|
|
if form.item_notes.data != '' and form.item_notes.data != 'None':
|
|
db_item.host_notes = form.item_notes.data
|
|
else:
|
|
db_item.host_notes = None
|
|
|
|
# Allocations
|
|
db_existing_allocs = Allocation.query.filter_by(device_id=db_item.id).all()
|
|
# Unlink old Allocation(s)
|
|
if db_existing_allocs != []:
|
|
for alloc in db_existing_allocs:
|
|
alloc.device_id = None
|
|
|
|
if form.item_ip_allocations.data != '' and form.item_ip_allocations.data != 'None':
|
|
|
|
allocs = form.item_ip_allocations.data.split(',')
|
|
put_allocs_db = ""
|
|
for alloc in allocs:
|
|
db_alloc = Allocation.query.filter_by(ipv4=alloc).first()
|
|
if db_alloc != None:
|
|
put_allocs_db += db_alloc.ipv4+','
|
|
db_alloc.device_id = db_item.id
|
|
elif allocs[-1] == '':
|
|
pass
|
|
else:
|
|
flash('Unknown IPv4', 'warning')
|
|
return redirect(url_for('inventory.list_items',item_type=request.args.get('page', type=int)))
|
|
|
|
db_item.host_allocations = put_allocs_db
|
|
|
|
else:
|
|
db_item.host_allocations = None
|
|
|
|
db.session.commit()
|
|
flash("Item Updated", 'success')
|
|
return redirect(url_for('inventory.list_items',item_type=request.args.get('page', type=int)))
|
|
|
|
# Setting default values
|
|
form.item_name.default = db_item.host_name
|
|
form.item_desc.default = db_item.host_desc
|
|
form.item_type.default = db_item.host_type
|
|
form.item_model.default =db_item.host_model
|
|
form.item_manufacturer.default = db_item.host_manufacturer
|
|
form.item_serial_number.default = db_item.host_serialnumber
|
|
form.item_os.default = db_item.host_os
|
|
form.item_location.default = db_item.host_location
|
|
form.item_owner.default = db_item.host_owner
|
|
form.item_contact.default = db_item.host_contact
|
|
form.item_ip_allocations.default = db_item.host_allocations
|
|
form.item_notes.default = db_item.host_notes
|
|
form.process()
|
|
|
|
modal = {
|
|
'title': 'Update Item',
|
|
'body': f'Change informations for <b>{db_item.host_name}</b>.',
|
|
'color': 'primary',
|
|
'modal_type': 'form',
|
|
'action_text': 'Save',
|
|
'form_url': url_for('inventory.item_edit_modal',item_id=item_id),
|
|
'inventory':True,
|
|
}
|
|
|
|
return render_template('panel/_modal.html', modal=modal, form=form, old_page=request.args.get('page', type=int), pagination_active=True) |