V1.2

app/routes/r_inventory.py

@@ -1,19 +1,18 @@
-from datetime import datetime
-from flask import Blueprint, request, render_template, flash, redirect, url_for, session
+from flask import Blueprint, request, render_template, flash, redirect, url_for, abort
 from flask_login import login_required, current_user
-from ipaddress import IPv4Network, IPv4Address
-
-from app._config import app_version
 
 from app.extensions import *
 from app.models import *
-from app.forms import NewInventoryItem
+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))
@@ -21,22 +20,210 @@ def add_item():
             db.session.commit()
             flash("Item ADDED", 'success')
     
-    return render_template('panel/inventory_add.html', title='ADD ITEM', form=form)
+    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):
-    db_items = Inventory.query.filter_by(host_type=item_type).all()
-    return render_template('panel/inventory_view.html', Title='View Items', item_list=db_items)
+    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('base.dashboard'))
+        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'),
+    ]
+    
+    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
+        # 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
+        
+        # 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.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)