devicehub-teal/ereuse_devicehub/inventory/views.py

import flask
from io import StringIO
from flask import Blueprint, request, url_for, g, make_response
from flask.views import View
from flask_login import current_user, login_required
from werkzeug.exceptions import NotFound

from ereuse_devicehub import messages
from ereuse_devicehub.inventory.forms import (AllocateForm, LotDeviceForm,
                                              LotForm, NewActionForm,
                                              NewDeviceForm, TagDeviceForm,
                                              TagForm, TagUnnamedForm,
                                              UploadSnapshotForm, DataWipeForm,
                                              TradeForm, TradeDocumentForm)
from ereuse_devicehub.resources.device.models import Device
from ereuse_devicehub.resources.lot.models import Lot
from ereuse_devicehub.resources.tag.model import Tag

# TODO(@slamora): rename base 'inventory.devices' --> 'inventory'
devices = Blueprint('inventory.devices', __name__, url_prefix='/inventory')


class DeviceListMix(View):
    decorators = [login_required]
    template_name = 'inventory/device_list.html'

    def get_context(self, lot_id):
        # TODO @cayop adding filter
        # https://github.com/eReuse/devicehub-teal/blob/testing/ereuse_devicehub/resources/device/views.py#L56
        filter_types = ['Desktop', 'Laptop', 'Server']
        lots = Lot.query.filter(Lot.owner_id == current_user.id)
        lot = None
        tags = Tag.query.filter(Tag.owner_id == current_user.id).filter(
            Tag.device_id == None).order_by(Tag.created.desc())

        if lot_id:
            lot = lots.filter(Lot.id == lot_id).one()
            devices = [dev for dev in lot.devices if dev.type in filter_types]
            devices = sorted(devices, key=lambda x: x.updated, reverse=True)
            form_new_action = NewActionForm(lot=lot.id)
            form_new_allocate = AllocateForm(lot=lot.id)
            form_new_datawipe = DataWipeForm(lot=lot.id)
            form_new_trade = TradeForm(
                lot=lot.id,
                user_to=g.user.email,
                user_from=g.user.email,
            )
        else:
            devices = Device.query.filter(
                Device.owner_id == current_user.id).filter(
                Device.type.in_(filter_types)).filter(Device.lots == None).order_by(
                    Device.updated.desc())
            form_new_action = NewActionForm()
            form_new_allocate = AllocateForm()
            form_new_datawipe = DataWipeForm()
            form_new_trade = ''
        action_devices = form_new_action.devices.data
        list_devices = []
        if action_devices:
            list_devices.extend([int(x) for x in action_devices.split(",")])

        self.context = {
            'devices': devices,
            'lots': lots,
            'form_lot_device': LotDeviceForm(),
            'form_tag_device': TagDeviceForm(),
            'form_new_action': form_new_action,
            'form_new_allocate': form_new_allocate,
            'form_new_datawipe': form_new_datawipe,
            'form_new_trade': form_new_trade,
            'lot': lot,
            'tags': tags,
            'list_devices': list_devices
        }

        return self.context


class DeviceListView(DeviceListMix):

    def dispatch_request(self, lot_id=None):
        self.get_context(lot_id)
        return flask.render_template(self.template_name, **self.context)


class DeviceDetailView(View):
    decorators = [login_required]
    template_name = 'inventory/device_detail.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,
            'page_title': 'Device {}'.format(device.devicehub_id),
        }
        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()

            next_url = request.referrer or url_for('inventory.devices.devicelist')
            return flask.redirect(next_url)


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()

            next_url = request.referrer or url_for('inventory.devices.devicelist')
            return flask.redirect(next_url)


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', lot_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', lot_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)


class UploadSnapshotView(View):
    methods = ['GET', 'POST']
    decorators = [login_required]
    template_name = 'inventory/upload_snapshot.html'

    def dispatch_request(self):
        context = {'page_title': 'Upload Snapshot'}
        lots = Lot.query.filter(Lot.owner_id == current_user.id).all()
        form = UploadSnapshotForm()
        if form.validate_on_submit():
            form.save()

        return flask.render_template(self.template_name, form=form, lots=lots, **context)


class DeviceCreateView(View):
    methods = ['GET', 'POST']
    decorators = [login_required]
    template_name = 'inventory/device_create.html'

    def dispatch_request(self):
        context = {'page_title': 'New Device'}
        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, **context)


class TagListView(View):
    methods = ['GET']
    decorators = [login_required]
    template_name = 'inventory/tag_list.html'

    def dispatch_request(self):
        tags = Tag.query.filter(Tag.owner_id == current_user.id)
        context = {
            'lots': [],
            'tags': tags,
            'page_title': 'Tags Management',
        }
        return flask.render_template(self.template_name, **context)


class TagAddView(View):
    methods = ['GET', 'POST']
    decorators = [login_required]
    template_name = 'inventory/tag_create.html'

    def dispatch_request(self):
        context = {'page_title': 'New Tag'}
        form = TagForm()
        if form.validate_on_submit():
            form.save()
            next_url = url_for('inventory.devices.taglist')
            return flask.redirect(next_url)

        return flask.render_template(self.template_name, form=form, **context)


class TagAddUnnamedView(View):
    methods = ['GET', 'POST']
    decorators = [login_required]
    template_name = 'inventory/tag_create_unnamed.html'

    def dispatch_request(self):
        context = {'page_title': 'New Unnamed Tag'}
        form = TagUnnamedForm()
        if form.validate_on_submit():
            form.save()
            next_url = url_for('inventory.devices.taglist')
            return flask.redirect(next_url)

        return flask.render_template(self.template_name, form=form, **context)


class TagDetailView(View):
    decorators = [login_required]
    template_name = 'inventory/tag_detail.html'

    def dispatch_request(self, id):
        lots = Lot.query.filter(Lot.owner_id == current_user.id)
        tag = Tag.query.filter(
            Tag.owner_id == current_user.id).filter(Tag.id == id).one()

        context = {
            'lots': lots,
            'tag': tag,
            'page_title': '{} Tag'.format(tag.code),
        }
        return flask.render_template(self.template_name, **context)


class TagLinkDeviceView(View):
    methods = ['POST']
    decorators = [login_required]
    # template_name = 'inventory/device_list.html'

    def dispatch_request(self):
        form = TagDeviceForm()
        if form.validate_on_submit():
            form.save()

            return flask.redirect(request.referrer)


class TagUnlinkDeviceView(View):
    methods = ['POST', 'GET']
    decorators = [login_required]
    template_name = 'inventory/tag_unlink_device.html'

    def dispatch_request(self, id):
        form = TagDeviceForm(delete=True, device=id)
        if form.validate_on_submit():
            form.remove()

            next_url = url_for('inventory.devices.devicelist')
            return flask.redirect(next_url)

        return flask.render_template(self.template_name, form=form, referrer=request.referrer)


class NewActionView(View):
    methods = ['POST']
    decorators = [login_required]
    form_class = NewActionForm

    def dispatch_request(self):
        self.form = self.form_class()

        if self.form.validate_on_submit():
            instance = self.form.save()
            messages.success('Action "{}" created successfully!'.format(self.form.type.data))

            next_url = self.get_next_url()
            return flask.redirect(next_url)

    def get_next_url(self):
        lot_id = self.form.lot.data

        if lot_id:
            return url_for('inventory.devices.lotdevicelist', lot_id=lot_id)

        return url_for('inventory.devices.devicelist')


class NewAllocateView(NewActionView, DeviceListMix):
    methods = ['POST']
    form_class = AllocateForm

    def dispatch_request(self):
        self.form = self.form_class()

        if self.form.validate_on_submit():
            instance = self.form.save()
            messages.success('Action "{}" created successfully!'.format(self.form.type.data))

            next_url = self.get_next_url()
            return flask.redirect(next_url)

        lot_id = self.form.lot.data
        self.get_context(lot_id)
        self.context['form_new_allocate'] = self.form
        return flask.render_template(self.template_name, **self.context)


class NewDataWipeView(NewActionView, DeviceListMix):
    methods = ['POST']
    form_class = DataWipeForm

    def dispatch_request(self):
        self.form = self.form_class()

        if self.form.validate_on_submit():
            instance = self.form.save()
            messages.success('Action "{}" created successfully!'.format(self.form.type.data))

            next_url = self.get_next_url()
            return flask.redirect(next_url)

        lot_id = self.form.lot.data
        self.get_context(lot_id)
        self.context['form_new_datawipe'] = self.form
        return flask.render_template(self.template_name, **self.context)


class NewTradeView(NewActionView, DeviceListMix):
    methods = ['POST']
    form_class = TradeForm

    def dispatch_request(self):
        self.form = self.form_class()

        if self.form.validate_on_submit():
            instance = self.form.save()
            messages.success('Action "{}" created successfully!'.format(self.form.type.data))

            next_url = self.get_next_url()
            return flask.redirect(next_url)

        lot_id = self.form.lot.data
        self.get_context(lot_id)
        self.context['form_new_trade'] = self.form
        return flask.render_template(self.template_name, **self.context)


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)

        if self.form.validate_on_submit():
            self.form.save()
            messages.success('Document created successfully!')
            next_url = url_for('inventory.devices.lotdevicelist', lot_id=lot_id)
            return flask.redirect(next_url)

        return flask.render_template(self.template_name, form=self.form, title=self.title)


class ExportsView(View):
    methods = ['GET']
    decorators = [login_required]

    # def dispatch_request(self, export_id):
    def dispatch_request(self):
        import pdb; pdb.trace()
        export_id = 'metrics'
        export_ids = {
            'metrics': self.metrics(),
            'devices-list': self.metrics(),
            'certificates': self.metrics()
        }

        if export_id not in export_ids:
            return NotFound()
        return export_ids[export_id]()

    def metrics(self):
        """Get device query and put information in csv format."""
        data = StringIO()
        # cw = csv.writer(data, delimiter=';', lineterminator="\n", quotechar='"')
        # first = True
        # document_ids = self.get_documents_id()
        # for device in query:
        #     d = DeviceRow(device, document_ids)
        #     if first:
        #         cw.writerow(d.keys())
        #         first = False
        #     cw.writerow(d.values())
        bfile = data.getvalue().encode('utf-8')
        output = make_response(bfile)
        # insert_hash(bfile)
        output.headers['Content-Disposition'] = 'attachment; filename=export.csv'
        output.headers['Content-type'] = 'text/csv'
        return output


devices.add_url_rule('/action/add/', view_func=NewActionView.as_view('action_add'))
devices.add_url_rule('/action/trade/add/', view_func=NewTradeView.as_view('trade_add'))
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'))
devices.add_url_rule('/device/', view_func=DeviceListView.as_view('devicelist'))
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/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/<string:id>/del/', view_func=LotDeleteView.as_view('lot_del'))
devices.add_url_rule('/lot/<string:id>/', view_func=LotUpdateView.as_view('lot_edit'))
devices.add_url_rule('/upload-snapshot/', view_func=UploadSnapshotView.as_view('upload_snapshot'))
devices.add_url_rule('/device/add/', view_func=DeviceCreateView.as_view('device_add'))
devices.add_url_rule('/tag/', view_func=TagListView.as_view('taglist'))
devices.add_url_rule('/tag/add/', view_func=TagAddView.as_view('tag_add'))
devices.add_url_rule('/tag/unnamed/add/', view_func=TagAddUnnamedView.as_view('tag_unnamed_add'))
devices.add_url_rule('/tag/<string:id>/', view_func=TagDetailView.as_view('tag_details'))
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'))
devices.add_url_rule('/export/', view_func=ExportsView.as_view('export'))