import flask from flask.views import View from flask import Blueprint, url_for, request from flask_login import login_required, current_user from ereuse_devicehub.resources.lot.models import Lot from ereuse_devicehub.resources.device.models import Device from ereuse_devicehub.inventory.forms import LotDeviceForm, LotForm devices = Blueprint('inventory.devices', __name__, url_prefix='/inventory') class DeviceListView(View): decorators = [login_required] template_name = 'inventory/device_list.html' def dispatch_request(self, id=None): # TODO @cayop adding filter filter_types = ['Desktop', 'Laptop', 'Server'] lots = Lot.query.filter(Lot.owner_id == current_user.id) lot = None if id: lot = lots.filter(Lot.id == id).one() # import pdb; pdb.set_trace() devices = [dev for dev in lot.devices if dev.type in filter_types] devices = sorted(devices, key=lambda x: x.updated, reverse=True) else: devices = Device.query.filter( Device.owner_id == current_user.id).filter( Device.type.in_(filter_types)).filter(Device.lots == None).order_by('id', 'updated') context = {'devices': devices, 'lots': lots, 'form_lot_device': LotDeviceForm(), 'lot': lot} return flask.render_template(self.template_name, **context) class LotDeviceAddView(View): methods = ['POST'] decorators = [login_required] template_name = 'inventory/device_list.html' def dispatch_request(self): form = LotDeviceForm() if form.validate_on_submit(): form.save() return flask.redirect(request.referrer) 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() # TODO @cayop It's possible this redirect not work in production return flask.redirect(request.referrer) class LotCreateView(View): 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.devices.lotdevicelist', id=form.id) return flask.redirect(next_url) lots = Lot.query.filter(Lot.owner_id == current_user.id) return flask.render_template(self.template_name, form=form, title=self.title, lots=lots) class LotUpdateView(View): methods = ['GET', 'POST'] decorators = [login_required] template_name = 'inventory/lot.html' title = "Edit a new lot" def dispatch_request(self, id): form = LotForm(id=id) if form.validate_on_submit(): form.save() next_url = url_for('inventory.devices.lotdevicelist', id=id) return flask.redirect(next_url) lots = Lot.query.filter(Lot.owner_id == current_user.id) return flask.render_template(self.template_name, form=form, title=self.title, lots=lots) 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) devices.add_url_rule('/device/', view_func=DeviceListView.as_view('devicelist')) devices.add_url_rule('/lot//device/', view_func=DeviceListView.as_view('lotdevicelist')) devices.add_url_rule('/lot/devices/add/', view_func=LotDeviceAddView.as_view('lot_devices_add')) devices.add_url_rule('/lot/devices/del/', view_func=LotDeviceDeleteView.as_view('lot_devices_del')) devices.add_url_rule('/lot/add/', view_func=LotCreateView.as_view('lot_add')) devices.add_url_rule('/lot//del/', view_func=LotDeleteView.as_view('lot_del')) devices.add_url_rule('/lot//', view_func=LotUpdateView.as_view('lot_edit'))