2021-12-28 11:36:02 +00:00
|
|
|
import flask
|
|
|
|
from flask.views import View
|
2022-01-12 20:02:47 +00:00
|
|
|
from flask import Blueprint, url_for, request
|
2021-12-28 12:55:56 +00:00
|
|
|
from flask_login import login_required, current_user
|
2021-12-28 11:36:02 +00:00
|
|
|
|
2021-12-29 12:44:28 +00:00
|
|
|
from ereuse_devicehub.resources.lot.models import Lot
|
2021-12-30 12:53:28 +00:00
|
|
|
from ereuse_devicehub.resources.device.models import Device
|
2022-01-19 12:40:40 +00:00
|
|
|
from ereuse_devicehub.inventory.forms import LotDeviceForm, LotForm, UploadSnapshotForm, NewDeviceForm
|
2021-12-28 11:36:02 +00:00
|
|
|
|
2021-12-28 12:55:56 +00:00
|
|
|
devices = Blueprint('inventory.devices', __name__, url_prefix='/inventory')
|
2021-12-28 11:36:02 +00:00
|
|
|
|
|
|
|
|
|
|
|
class DeviceListView(View):
|
2021-12-28 12:55:56 +00:00
|
|
|
decorators = [login_required]
|
2021-12-28 11:36:02 +00:00
|
|
|
template_name = 'inventory/device_list.html'
|
|
|
|
|
2022-01-25 09:30:46 +00:00
|
|
|
def dispatch_request(self, lot_id=None):
|
2021-12-29 09:13:34 +00:00
|
|
|
# TODO @cayop adding filter
|
2022-01-25 09:30:46 +00:00
|
|
|
# https://github.com/eReuse/devicehub-teal/blob/testing/ereuse_devicehub/resources/device/views.py#L56
|
2021-12-28 12:55:56 +00:00
|
|
|
filter_types = ['Desktop', 'Laptop', 'Server']
|
2021-12-30 16:23:34 +00:00
|
|
|
lots = Lot.query.filter(Lot.owner_id == current_user.id)
|
2021-12-30 20:40:44 +00:00
|
|
|
lot = None
|
2022-01-25 09:30:46 +00:00
|
|
|
if lot_id:
|
|
|
|
lot = lots.filter(Lot.id == lot_id).one()
|
2021-12-30 20:40:44 +00:00
|
|
|
devices = [dev for dev in lot.devices if dev.type in filter_types]
|
2022-01-12 19:40:57 +00:00
|
|
|
devices = sorted(devices, key=lambda x: x.updated, reverse=True)
|
2021-12-30 20:40:44 +00:00
|
|
|
else:
|
|
|
|
devices = Device.query.filter(
|
|
|
|
Device.owner_id == current_user.id).filter(
|
2022-01-18 10:24:27 +00:00
|
|
|
Device.type.in_(filter_types)).filter(Device.lots == None).order_by(
|
|
|
|
Device.updated.desc())
|
2021-12-30 16:23:34 +00:00
|
|
|
|
|
|
|
context = {'devices': devices,
|
|
|
|
'lots': lots,
|
2022-01-03 10:32:12 +00:00
|
|
|
'form_lot_device': LotDeviceForm(),
|
2021-12-30 16:23:34 +00:00
|
|
|
'lot': lot}
|
|
|
|
return flask.render_template(self.template_name, **context)
|
|
|
|
|
|
|
|
|
2022-01-13 15:08:55 +00:00
|
|
|
class DeviceDetailsView(View):
|
|
|
|
decorators = [login_required]
|
|
|
|
template_name = 'inventory/device_details.html'
|
|
|
|
|
|
|
|
def dispatch_request(self, id):
|
|
|
|
lots = Lot.query.filter(Lot.owner_id == current_user.id)
|
|
|
|
device = Device.query.filter(
|
|
|
|
Device.owner_id == current_user.id).filter(Device.devicehub_id == id).one()
|
|
|
|
|
|
|
|
context = {'device': device,
|
|
|
|
'lots': lots}
|
|
|
|
return flask.render_template(self.template_name, **context)
|
|
|
|
|
|
|
|
|
2021-12-30 12:53:28 +00:00
|
|
|
class LotDeviceAddView(View):
|
|
|
|
methods = ['POST']
|
|
|
|
decorators = [login_required]
|
|
|
|
template_name = 'inventory/device_list.html'
|
|
|
|
|
|
|
|
def dispatch_request(self):
|
2022-01-03 10:32:12 +00:00
|
|
|
form = LotDeviceForm()
|
2021-12-30 12:53:28 +00:00
|
|
|
if form.validate_on_submit():
|
2021-12-30 20:35:54 +00:00
|
|
|
form.save()
|
|
|
|
|
2022-01-26 10:48:37 +00:00
|
|
|
next_url = request.referrer or url_for('inventory.devices.devicelist')
|
|
|
|
return flask.redirect(next_url)
|
2022-01-03 10:32:12 +00:00
|
|
|
|
|
|
|
|
|
|
|
class LotDeviceDeleteView(View):
|
|
|
|
methods = ['POST']
|
|
|
|
decorators = [login_required]
|
|
|
|
template_name = 'inventory/device_list.html'
|
|
|
|
|
|
|
|
def dispatch_request(self):
|
|
|
|
form = LotDeviceForm()
|
|
|
|
if form.validate_on_submit():
|
|
|
|
form.remove()
|
|
|
|
|
2022-01-26 10:48:37 +00:00
|
|
|
next_url = request.referrer or url_for('inventory.devices.devicelist')
|
|
|
|
return flask.redirect(next_url)
|
2021-12-30 20:35:54 +00:00
|
|
|
|
|
|
|
|
2022-01-12 16:09:20 +00:00
|
|
|
class LotCreateView(View):
|
2021-12-30 20:35:54 +00:00
|
|
|
methods = ['GET', 'POST']
|
|
|
|
decorators = [login_required]
|
|
|
|
template_name = 'inventory/lot.html'
|
|
|
|
title = "Add a new lot"
|
|
|
|
|
2022-01-12 16:09:20 +00:00
|
|
|
def dispatch_request(self):
|
|
|
|
form = LotForm()
|
|
|
|
if form.validate_on_submit():
|
|
|
|
form.save()
|
2022-01-25 09:30:46 +00:00
|
|
|
next_url = url_for('inventory.devices.lotdevicelist', lot_id=form.id)
|
2022-01-12 16:09:20 +00:00
|
|
|
return flask.redirect(next_url)
|
|
|
|
|
2022-01-12 18:44:10 +00:00
|
|
|
lots = Lot.query.filter(Lot.owner_id == current_user.id)
|
|
|
|
return flask.render_template(self.template_name, form=form, title=self.title, lots=lots)
|
2022-01-12 16:09:20 +00:00
|
|
|
|
|
|
|
|
|
|
|
class LotUpdateView(View):
|
|
|
|
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():
|
2022-01-05 10:10:32 +00:00
|
|
|
form.save()
|
2022-01-25 09:30:46 +00:00
|
|
|
next_url = url_for('inventory.devices.lotdevicelist', lot_id=id)
|
2021-12-30 12:53:28 +00:00
|
|
|
return flask.redirect(next_url)
|
|
|
|
|
2022-01-12 18:44:10 +00:00
|
|
|
lots = Lot.query.filter(Lot.owner_id == current_user.id)
|
|
|
|
return flask.render_template(self.template_name, form=form, title=self.title, lots=lots)
|
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)
|
|
|
|
form.remove()
|
|
|
|
next_url = url_for('inventory.devices.devicelist')
|
|
|
|
return flask.redirect(next_url)
|
|
|
|
|
|
|
|
|
2022-01-17 13:17:52 +00:00
|
|
|
class UploadSnapshotView(View):
|
|
|
|
methods = ['GET', 'POST']
|
|
|
|
decorators = [login_required]
|
|
|
|
template_name = 'inventory/upload_snapshot.html'
|
|
|
|
|
|
|
|
def dispatch_request(self):
|
2022-01-18 10:24:27 +00:00
|
|
|
lots = Lot.query.filter(Lot.owner_id == current_user.id).all()
|
2022-01-17 13:17:52 +00:00
|
|
|
form = UploadSnapshotForm()
|
|
|
|
if form.validate_on_submit():
|
|
|
|
form.save()
|
|
|
|
|
|
|
|
return flask.render_template(self.template_name, form=form, lots=lots)
|
|
|
|
|
|
|
|
|
2022-01-19 12:40:40 +00:00
|
|
|
class CreateDeviceView(View):
|
|
|
|
methods = ['GET', 'POST']
|
|
|
|
decorators = [login_required]
|
|
|
|
template_name = 'inventory/create_device.html'
|
|
|
|
|
|
|
|
def dispatch_request(self):
|
|
|
|
lots = Lot.query.filter(Lot.owner_id == current_user.id).all()
|
|
|
|
form = NewDeviceForm()
|
|
|
|
if form.validate_on_submit():
|
|
|
|
form.save()
|
|
|
|
next_url = url_for('inventory.devices.devicelist')
|
|
|
|
return flask.redirect(next_url)
|
|
|
|
|
|
|
|
return flask.render_template(self.template_name, form=form, lots=lots)
|
|
|
|
|
|
|
|
|
2021-12-29 09:13:34 +00:00
|
|
|
devices.add_url_rule('/device/', view_func=DeviceListView.as_view('devicelist'))
|
2022-01-13 15:08:55 +00:00
|
|
|
devices.add_url_rule('/device/<string:id>/', view_func=DeviceDetailsView.as_view('device_details'))
|
2022-01-25 09:30:46 +00:00
|
|
|
devices.add_url_rule('/lot/<string:lot_id>/device/', view_func=DeviceListView.as_view('lotdevicelist'))
|
2021-12-30 20:35:54 +00:00
|
|
|
devices.add_url_rule('/lot/devices/add/', view_func=LotDeviceAddView.as_view('lot_devices_add'))
|
2022-01-03 10:32:12 +00:00
|
|
|
devices.add_url_rule('/lot/devices/del/', view_func=LotDeviceDeleteView.as_view('lot_devices_del'))
|
2022-01-12 16:09:20 +00:00
|
|
|
devices.add_url_rule('/lot/add/', view_func=LotCreateView.as_view('lot_add'))
|
2022-01-03 11:28:42 +00:00
|
|
|
devices.add_url_rule('/lot/<string:id>/del/', view_func=LotDeleteView.as_view('lot_del'))
|
2022-01-12 16:09:20 +00:00
|
|
|
devices.add_url_rule('/lot/<string:id>/', view_func=LotUpdateView.as_view('lot_edit'))
|
2022-01-17 13:17:52 +00:00
|
|
|
devices.add_url_rule('/upload-snapshot/', view_func=UploadSnapshotView.as_view('upload_snapshot'))
|
2022-01-19 12:40:40 +00:00
|
|
|
devices.add_url_rule('/device/add/', view_func=CreateDeviceView.as_view('device_add'))
|