This repository has been archived on 2024-05-31. You can view files and clone it, but cannot push or open issues or pull requests.
devicehub-teal/ereuse_devicehub/inventory/views.py

608 lines
19 KiB
Python
Raw Normal View History

2022-02-24 13:15:58 +00:00
import csv
import logging
from distutils.util import strtobool
2022-02-22 10:11:53 +00:00
from io import StringIO
2022-02-24 13:15:58 +00:00
import flask
import flask_weasyprint
from flask import Blueprint, g, make_response, request, url_for
2021-12-28 11:36:02 +00:00
from flask.views import View
2022-02-03 09:50:36 +00:00
from flask_login import current_user, login_required
2022-03-08 09:05:32 +00:00
from werkzeug.exceptions import NotFound
2021-12-28 11:36:02 +00:00
2022-04-29 11:10:44 +00:00
from ereuse_devicehub import messages
2022-03-08 09:05:32 +00:00
from ereuse_devicehub.db import db
2022-02-24 13:15:58 +00:00
from ereuse_devicehub.inventory.forms import (
2022-02-25 11:27:14 +00:00
AllocateForm,
DataWipeForm,
2022-05-30 14:33:01 +00:00
EditTransferForm,
2022-03-07 12:07:26 +00:00
FilterForm,
2022-02-25 11:27:14 +00:00
LotForm,
NewActionForm,
NewDeviceForm,
TagDeviceForm,
2022-02-25 11:34:02 +00:00
TradeDocumentForm,
TradeForm,
2022-05-27 14:32:22 +00:00
TransferForm,
2022-02-25 11:57:13 +00:00
UploadSnapshotForm,
2022-02-25 11:56:01 +00:00
)
2022-03-31 14:12:35 +00:00
from ereuse_devicehub.labels.forms import PrintLabelsForm
2022-02-24 13:15:58 +00:00
from ereuse_devicehub.resources.action.models import Trade
2022-02-25 11:57:13 +00:00
from ereuse_devicehub.resources.device.models import Computer, DataStorage, Device
from ereuse_devicehub.resources.documents.device_row import ActionRow, DeviceRow
2022-02-24 13:15:58 +00:00
from ereuse_devicehub.resources.hash_reports import insert_hash
2021-12-29 12:44:28 +00:00
from ereuse_devicehub.resources.lot.models import Lot
2022-01-11 12:13:57 +00:00
from ereuse_devicehub.resources.tag.model import Tag
2022-04-29 11:10:44 +00:00
from ereuse_devicehub.views import GenericMixView
2021-12-28 11:36:02 +00:00
devices = Blueprint('inventory', __name__, url_prefix='/inventory')
2021-12-28 11:36:02 +00:00
logger = logging.getLogger(__name__)
2021-12-28 11:36:02 +00:00
class DeviceListMix(GenericMixView):
template_name = 'inventory/device_list.html'
def get_context(self, lot_id, only_unassigned=True):
2022-04-29 11:10:44 +00:00
super().get_context()
lots = self.context['lots']
form_filter = FilterForm(lots, lot_id, only_unassigned=only_unassigned)
2022-05-09 11:43:02 +00:00
devices = form_filter.search()
2021-12-30 20:40:44 +00:00
lot = None
2022-05-30 14:33:01 +00:00
form_transfer = ''
2022-01-25 11:53:36 +00:00
2022-01-25 09:30:46 +00:00
if lot_id:
lot = lots.filter(Lot.id == lot_id).one()
2022-05-30 14:33:01 +00:00
if not lot.is_temporary and lot.transfer:
2022-05-30 15:35:27 +00:00
form_transfer = EditTransferForm(lot_id=lot.id)
2022-05-12 12:23:01 +00:00
form_new_action = NewActionForm(lot=lot_id)
2022-04-29 11:10:44 +00:00
self.context.update(
{
2022-05-09 11:43:02 +00:00
'devices': devices,
2022-04-29 11:10:44 +00:00
'form_tag_device': TagDeviceForm(),
'form_new_action': form_new_action,
2022-05-12 12:23:01 +00:00
'form_new_allocate': AllocateForm(lot=lot_id),
'form_new_datawipe': DataWipeForm(lot=lot_id),
2022-05-30 14:33:01 +00:00
'form_transfer': form_transfer,
2022-04-29 11:10:44 +00:00
'form_filter': form_filter,
'form_print_labels': PrintLabelsForm(),
'lot': lot,
'tags': self.get_user_tags(),
'list_devices': self.get_selected_devices(form_new_action),
'unassigned_devices': only_unassigned,
2022-04-29 11:10:44 +00:00
}
)
2022-01-10 14:53:11 +00:00
return self.context
def get_user_tags(self):
return (
Tag.query.filter(Tag.owner_id == current_user.id)
.filter(Tag.device_id.is_(None))
.order_by(Tag.id.asc())
)
def get_selected_devices(self, action_form):
"""Retrieve selected devices (when action form is submited)"""
action_devices = action_form.devices.data
if action_devices:
return [int(x) for x in action_devices.split(",")]
return []
class DeviceListView(DeviceListMix):
def dispatch_request(self, lot_id=None):
only_unassigned = request.args.get(
'only_unassigned', default=True, type=strtobool
)
self.get_context(lot_id, only_unassigned)
return flask.render_template(self.template_name, **self.context)
2021-12-30 16:23:34 +00:00
class DeviceDetailView(GenericMixView):
2022-01-13 15:08:55 +00:00
decorators = [login_required]
template_name = 'inventory/device_detail.html'
2022-01-13 15:08:55 +00:00
def dispatch_request(self, id):
2022-04-29 11:10:44 +00:00
self.get_context()
2022-02-24 13:15:58 +00:00
device = (
Device.query.filter(Device.owner_id == current_user.id)
.filter(Device.devicehub_id == id)
.one()
)
2022-01-13 15:08:55 +00:00
2022-04-29 11:10:44 +00:00
self.context.update(
{
'device': device,
'page_title': 'Device {}'.format(device.devicehub_id),
}
)
return flask.render_template(self.template_name, **self.context)
2022-01-13 15:08:55 +00:00
class LotCreateView(GenericMixView):
2021-12-30 20:35:54 +00:00
methods = ['GET', 'POST']
decorators = [login_required]
template_name = 'inventory/lot.html'
title = "Add a new lot"
def dispatch_request(self):
form = LotForm()
if form.validate_on_submit():
form.save()
next_url = url_for('inventory.lotdevicelist', lot_id=form.id)
return flask.redirect(next_url)
2022-04-29 11:10:44 +00:00
self.get_context()
self.context.update(
{
'form': form,
'title': self.title,
}
)
return flask.render_template(self.template_name, **self.context)
2022-05-03 15:00:02 +00:00
class LotUpdateView(GenericMixView):
methods = ['GET', 'POST']
decorators = [login_required]
template_name = 'inventory/lot.html'
title = "Edit a new lot"
def dispatch_request(self, id):
2021-12-30 20:35:54 +00:00
form = LotForm(id=id)
if form.validate_on_submit():
form.save()
next_url = url_for('inventory.lotdevicelist', lot_id=id)
2021-12-30 12:53:28 +00:00
return flask.redirect(next_url)
2022-04-29 11:10:44 +00:00
self.get_context()
self.context.update(
{
'form': form,
'title': self.title,
}
)
return flask.render_template(self.template_name, **self.context)
2021-12-30 20:35:54 +00:00
2021-12-30 12:53:28 +00:00
2022-01-03 11:28:42 +00:00
class LotDeleteView(View):
methods = ['GET']
decorators = [login_required]
template_name = 'inventory/device_list.html'
def dispatch_request(self, id):
form = LotForm(id=id)
if form.instance.trade:
msg = "Sorry, the lot cannot be deleted because have a trade action "
messages.error(msg)
next_url = url_for('inventory.lotdevicelist', lot_id=id)
return flask.redirect(next_url)
2022-01-03 11:28:42 +00:00
form.remove()
next_url = url_for('inventory.devicelist')
2022-01-03 11:28:42 +00:00
return flask.redirect(next_url)
class UploadSnapshotView(GenericMixView):
2022-01-17 13:17:52 +00:00
methods = ['GET', 'POST']
decorators = [login_required]
template_name = 'inventory/upload_snapshot.html'
def dispatch_request(self, lot_id=None):
2022-04-29 11:10:44 +00:00
self.get_context()
2022-01-17 13:17:52 +00:00
form = UploadSnapshotForm()
2022-04-29 11:10:44 +00:00
self.context.update(
{
'page_title': 'Upload Snapshot',
'form': form,
'lot_id': lot_id,
}
)
2022-01-17 13:17:52 +00:00
if form.validate_on_submit():
snapshot = form.save(commit=False)
if lot_id:
2022-04-29 11:10:44 +00:00
lots = self.context['lots']
lot = lots.filter(Lot.id == lot_id).one()
lot.devices.add(snapshot.device)
db.session.add(lot)
db.session.commit()
2022-01-17 13:17:52 +00:00
2022-04-29 11:10:44 +00:00
return flask.render_template(self.template_name, **self.context)
2022-01-17 13:17:52 +00:00
class DeviceCreateView(GenericMixView):
2022-01-19 12:40:40 +00:00
methods = ['GET', 'POST']
decorators = [login_required]
template_name = 'inventory/device_create.html'
2022-01-19 12:40:40 +00:00
def dispatch_request(self, lot_id=None):
2022-04-29 11:10:44 +00:00
self.get_context()
2022-01-19 12:40:40 +00:00
form = NewDeviceForm()
2022-04-29 11:10:44 +00:00
self.context.update(
{
'page_title': 'New Device',
'form': form,
'lot_id': lot_id,
}
)
2022-01-19 12:40:40 +00:00
if form.validate_on_submit():
snapshot = form.save(commit=False)
next_url = url_for('inventory.devicelist')
if lot_id:
next_url = url_for('inventory.lotdevicelist', lot_id=lot_id)
2022-04-29 11:10:44 +00:00
lots = self.context['lots']
lot = lots.filter(Lot.id == lot_id).one()
lot.devices.add(snapshot.device)
db.session.add(lot)
db.session.commit()
2022-03-08 16:49:56 +00:00
messages.success('Device "{}" created successfully!'.format(form.type.data))
2022-01-19 12:40:40 +00:00
return flask.redirect(next_url)
2022-04-29 11:10:44 +00:00
return flask.render_template(self.template_name, **self.context)
2022-01-19 12:40:40 +00:00
class TagLinkDeviceView(View):
2022-01-25 11:53:36 +00:00
methods = ['POST']
decorators = [login_required]
# template_name = 'inventory/device_list.html'
2022-01-25 11:53:36 +00:00
def dispatch_request(self):
form = TagDeviceForm()
if form.validate_on_submit():
form.save()
return flask.redirect(request.referrer)
2022-04-29 11:10:44 +00:00
class TagUnlinkDeviceView(GenericMixView):
2022-01-25 13:39:15 +00:00
methods = ['POST', 'GET']
2022-01-25 11:53:36 +00:00
decorators = [login_required]
template_name = 'inventory/tag_unlink_device.html'
2022-01-25 11:53:36 +00:00
2022-01-25 13:39:15 +00:00
def dispatch_request(self, id):
2022-04-29 11:10:44 +00:00
self.get_context()
2022-01-25 13:39:15 +00:00
form = TagDeviceForm(delete=True, device=id)
2022-01-25 11:53:36 +00:00
if form.validate_on_submit():
form.remove()
next_url = url_for('inventory.devicelist')
2022-01-26 11:29:03 +00:00
return flask.redirect(next_url)
2022-01-25 11:53:36 +00:00
2022-04-29 11:10:44 +00:00
self.context.update(
{
'form': form,
'referrer': request.referrer,
}
2022-02-24 13:15:58 +00:00
)
2022-01-19 12:40:40 +00:00
2022-04-29 11:10:44 +00:00
return flask.render_template(self.template_name, **self.context)
2022-01-19 12:40:40 +00:00
class NewActionView(View):
methods = ['POST']
decorators = [login_required]
form_class = NewActionForm
def dispatch_request(self):
self.form = self.form_class()
2022-04-21 12:02:16 +00:00
next_url = self.get_next_url()
if self.form.validate_on_submit():
2022-02-24 13:15:58 +00:00
self.form.save()
messages.success(
'Action "{}" created successfully!'.format(self.form.type.data)
)
next_url = self.get_next_url()
return flask.redirect(next_url)
2022-04-21 12:02:16 +00:00
messages.error('Action {} error!'.format(self.form.type.data))
return flask.redirect(next_url)
def get_next_url(self):
lot_id = self.form.lot.data
if lot_id:
return url_for('inventory.lotdevicelist', lot_id=lot_id)
return url_for('inventory.devicelist')
2022-01-03 11:28:42 +00:00
class NewAllocateView(NewActionView, DeviceListMix):
methods = ['POST']
form_class = AllocateForm
2022-02-07 13:01:56 +00:00
def dispatch_request(self):
self.form = self.form_class()
if self.form.validate_on_submit():
2022-02-24 13:15:58 +00:00
self.form.save()
messages.success(
'Action "{}" created successfully!'.format(self.form.type.data)
)
next_url = self.get_next_url()
return flask.redirect(next_url)
2022-04-21 12:02:16 +00:00
messages.error('Action {} error!'.format(self.form.type.data))
2022-04-29 15:53:59 +00:00
for k, v in self.form.errors.items():
value = ';'.join(v)
2022-05-12 08:33:04 +00:00
key = self.form[k].label.text
messages.error('Action Error {key}: {value}!'.format(key=key, value=value))
2022-04-21 12:02:16 +00:00
next_url = self.get_next_url()
return flask.redirect(next_url)
2022-02-07 13:01:56 +00:00
class NewDataWipeView(NewActionView, DeviceListMix):
methods = ['POST']
form_class = DataWipeForm
2022-02-07 13:01:56 +00:00
def dispatch_request(self):
self.form = self.form_class()
if self.form.validate_on_submit():
2022-02-24 13:15:58 +00:00
self.form.save()
messages.success(
'Action "{}" created successfully!'.format(self.form.type.data)
)
next_url = self.get_next_url()
return flask.redirect(next_url)
2022-02-07 13:01:56 +00:00
2022-04-21 12:02:16 +00:00
messages.error('Action {} error!'.format(self.form.type.data))
next_url = self.get_next_url()
return flask.redirect(next_url)
2022-02-07 13:01:56 +00:00
2022-02-09 13:06:28 +00:00
class NewTradeView(NewActionView, DeviceListMix):
methods = ['POST']
form_class = TradeForm
def dispatch_request(self):
self.form = self.form_class()
if self.form.validate_on_submit():
2022-02-24 13:15:58 +00:00
self.form.save()
messages.success(
'Action "{}" created successfully!'.format(self.form.type.data)
)
2022-02-09 13:06:28 +00:00
next_url = self.get_next_url()
return flask.redirect(next_url)
2022-04-21 12:02:16 +00:00
messages.error('Action {} error!'.format(self.form.type.data))
next_url = self.get_next_url()
return flask.redirect(next_url)
2022-02-09 13:06:28 +00:00
class NewTradeDocumentView(View):
methods = ['POST', 'GET']
decorators = [login_required]
template_name = 'inventory/trade_document.html'
form_class = TradeDocumentForm
title = "Add new document"
def dispatch_request(self, lot_id):
self.form = self.form_class(lot=lot_id)
2022-04-29 11:10:44 +00:00
self.get_context()
if self.form.validate_on_submit():
self.form.save()
messages.success('Document created successfully!')
next_url = url_for('inventory.lotdevicelist', lot_id=lot_id)
return flask.redirect(next_url)
2022-04-29 11:10:44 +00:00
self.context.update({'form': self.form, 'title': self.title})
return flask.render_template(self.template_name, **self.context)
2022-05-27 14:32:22 +00:00
class NewTransferView(GenericMixView):
methods = ['POST', 'GET']
template_name = 'inventory/new_transfer.html'
form_class = TransferForm
title = "Add new transfer"
def dispatch_request(self, lot_id, type_id):
self.form = self.form_class(lot_id=lot_id, type=type_id)
self.get_context()
if self.form.validate_on_submit():
self.form.save()
new_lot_id = lot_id
if self.form.newlot.id:
2022-05-30 15:35:27 +00:00
new_lot_id = "{}".format(self.form.newlot.id)
2022-05-30 14:33:01 +00:00
Lot.query.filter(Lot.id == new_lot_id).one()
2022-05-27 14:32:22 +00:00
messages.success('Transfer created successfully!')
next_url = url_for('inventory.lotdevicelist', lot_id=new_lot_id)
return flask.redirect(next_url)
self.context.update({'form': self.form, 'title': self.title})
return flask.render_template(self.template_name, **self.context)
2022-05-30 14:33:01 +00:00
class EditTransferView(GenericMixView):
methods = ['POST']
form_class = EditTransferForm
def dispatch_request(self, lot_id):
self.get_context()
form = self.form_class(request.form, lot_id=lot_id)
next_url = url_for('inventory.lotdevicelist', lot_id=lot_id)
if form.validate_on_submit():
form.save()
messages.success('Transfer updated successfully!')
return flask.redirect(next_url)
messages.error('Transfer updated error!')
return flask.redirect(next_url)
2022-02-22 10:11:53 +00:00
class ExportsView(View):
methods = ['GET']
decorators = [login_required]
2022-02-24 13:15:58 +00:00
def dispatch_request(self, export_id):
2022-02-22 10:11:53 +00:00
export_ids = {
2022-02-24 13:15:58 +00:00
'metrics': self.metrics,
'devices': self.devices_list,
'certificates': self.erasure,
2022-02-22 10:11:53 +00:00
}
if export_id not in export_ids:
return NotFound()
return export_ids[export_id]()
2022-02-24 13:15:58 +00:00
def find_devices(self):
args = request.args.get('ids')
ids = args.split(',') if args else []
query = Device.query.filter(Device.owner == g.user)
return query.filter(Device.devicehub_id.in_(ids))
def response_csv(self, data, name):
2022-02-22 10:11:53 +00:00
bfile = data.getvalue().encode('utf-8')
2022-02-24 13:15:58 +00:00
# insert proof
insert_hash(bfile)
2022-02-22 10:11:53 +00:00
output = make_response(bfile)
2022-02-24 13:15:58 +00:00
output.headers['Content-Disposition'] = 'attachment; filename={}'.format(name)
2022-02-22 10:11:53 +00:00
output.headers['Content-type'] = 'text/csv'
return output
2022-02-24 13:15:58 +00:00
def devices_list(self):
"""Get device query and put information in csv format."""
data = StringIO()
cw = csv.writer(data, delimiter=';', lineterminator="\n", quotechar='"')
first = True
for device in self.find_devices():
d = DeviceRow(device, {})
if first:
cw.writerow(d.keys())
first = False
cw.writerow(d.values())
return self.response_csv(data, "export.csv")
def metrics(self):
"""Get device query and put information in csv format."""
data = StringIO()
cw = csv.writer(data, delimiter=';', lineterminator="\n", quotechar='"')
first = True
devs_id = []
# Get the allocate info
for device in self.find_devices():
devs_id.append(device.id)
for allocate in device.get_metrics():
d = ActionRow(allocate)
if first:
cw.writerow(d.keys())
first = False
cw.writerow(d.values())
# Get the trade info
query_trade = Trade.query.filter(
Trade.devices.any(Device.id.in_(devs_id))
).all()
lot_id = request.args.get('lot')
if lot_id and not query_trade:
lot = Lot.query.filter_by(id=lot_id).one()
if hasattr(lot, "trade") and lot.trade:
if g.user in [lot.trade.user_from, lot.trade.user_to]:
query_trade = [lot.trade]
for trade in query_trade:
data_rows = trade.get_metrics()
for row in data_rows:
d = ActionRow(row)
if first:
cw.writerow(d.keys())
first = False
cw.writerow(d.values())
return self.response_csv(data, "actions_export.csv")
def erasure(self):
template = self.build_erasure_certificate()
res = flask_weasyprint.render_pdf(
flask_weasyprint.HTML(string=template),
download_filename='erasure-certificate.pdf',
)
insert_hash(res.data)
return res
def build_erasure_certificate(self):
erasures = []
for device in self.find_devices():
if isinstance(device, Computer):
for privacy in device.privacy:
erasures.append(privacy)
elif isinstance(device, DataStorage):
if device.privacy:
erasures.append(device.privacy)
params = {
'title': 'Erasure Certificate',
'erasures': tuple(erasures),
'url_pdf': '',
}
return flask.render_template('inventory/erasure.html', **params)
2022-02-22 10:11:53 +00:00
devices.add_url_rule('/action/add/', view_func=NewActionView.as_view('action_add'))
2022-02-09 13:06:28 +00:00
devices.add_url_rule('/action/trade/add/', view_func=NewTradeView.as_view('trade_add'))
2022-02-24 13:15:58 +00:00
devices.add_url_rule(
'/action/allocate/add/', view_func=NewAllocateView.as_view('allocate_add')
)
devices.add_url_rule(
'/action/datawipe/add/', view_func=NewDataWipeView.as_view('datawipe_add')
)
devices.add_url_rule(
'/lot/<string:lot_id>/trade-document/add/',
view_func=NewTradeDocumentView.as_view('trade_document_add'),
)
2021-12-29 09:13:34 +00:00
devices.add_url_rule('/device/', view_func=DeviceListView.as_view('devicelist'))
2022-02-24 13:15:58 +00:00
devices.add_url_rule(
'/device/<string:id>/', view_func=DeviceDetailView.as_view('device_details')
)
devices.add_url_rule(
'/lot/<string:lot_id>/device/', view_func=DeviceListView.as_view('lotdevicelist')
)
devices.add_url_rule('/lot/add/', view_func=LotCreateView.as_view('lot_add'))
2022-02-24 13:15:58 +00:00
devices.add_url_rule(
'/lot/<string:id>/del/', view_func=LotDeleteView.as_view('lot_del')
)
devices.add_url_rule('/lot/<string:id>/', view_func=LotUpdateView.as_view('lot_edit'))
2022-02-24 13:15:58 +00:00
devices.add_url_rule(
'/upload-snapshot/', view_func=UploadSnapshotView.as_view('upload_snapshot')
)
devices.add_url_rule(
2022-03-08 16:49:56 +00:00
'/lot/<string:lot_id>/upload-snapshot/',
view_func=UploadSnapshotView.as_view('lot_upload_snapshot'),
)
devices.add_url_rule('/device/add/', view_func=DeviceCreateView.as_view('device_add'))
devices.add_url_rule(
2022-03-08 16:49:56 +00:00
'/lot/<string:lot_id>/device/add/',
view_func=DeviceCreateView.as_view('lot_device_add'),
)
2022-02-24 13:15:58 +00:00
devices.add_url_rule(
'/tag/devices/add/', view_func=TagLinkDeviceView.as_view('tag_devices_add')
)
devices.add_url_rule(
'/tag/devices/<int:id>/del/',
view_func=TagUnlinkDeviceView.as_view('tag_devices_del'),
)
devices.add_url_rule(
'/export/<string:export_id>/', view_func=ExportsView.as_view('export')
)
2022-05-27 14:32:22 +00:00
devices.add_url_rule(
'/lot/<string:lot_id>/transfer/<string:type_id>/',
view_func=NewTransferView.as_view('new_transfer'),
)
2022-05-30 14:33:01 +00:00
devices.add_url_rule(
'/lot/<string:lot_id>/transfer/',
view_func=EditTransferView.as_view('edit_transfer'),
)