devicehub-django/device/views.py

138 lines
4.2 KiB
Python
Raw Normal View History

2024-07-15 14:23:14 +00:00
import json
2024-07-05 13:32:07 +00:00
from django.urls import reverse_lazy
from django.shortcuts import get_object_or_404
from django.utils.translation import gettext_lazy as _
from django.views.generic.edit import (
CreateView,
UpdateView,
2024-07-19 15:40:01 +00:00
FormView,
2024-07-05 13:32:07 +00:00
)
2024-07-15 14:23:14 +00:00
from django.views.generic.base import TemplateView
2024-07-19 15:40:01 +00:00
from dashboard.mixins import DashboardView
2024-07-15 14:23:14 +00:00
from snapshot.models import Annotation
from snapshot.xapian import search
2024-07-18 15:21:22 +00:00
from lot.models import LotTag
from device.models import Device
2024-07-19 15:40:01 +00:00
from device.forms import DeviceFormSet
2024-07-05 13:32:07 +00:00
2024-07-19 15:40:01 +00:00
class NewDeviceView(DashboardView, FormView):
2024-07-05 13:32:07 +00:00
template_name = "new_device.html"
title = _("New Device")
breadcrumb = "Device / New Device"
2024-07-22 11:48:10 +00:00
success_url = reverse_lazy('device:add')
# success_url = reverse_lazy('dashboard:unassigned_devices')
2024-07-19 15:40:01 +00:00
form_class = DeviceFormSet
2024-07-05 13:32:07 +00:00
def form_valid(self, form):
2024-07-22 11:48:10 +00:00
# import pdb; pdb.set_trace()
# form.instance.owner = self.request.user
2024-07-05 13:32:07 +00:00
response = super().form_valid(form)
return response
2024-07-18 15:21:22 +00:00
2024-07-22 11:48:10 +00:00
def form_invalid(self, form):
import pdb; pdb.set_trace()
response = super().form_invalid(form)
return response
2024-07-05 13:32:07 +00:00
2024-07-19 15:40:01 +00:00
# class AddToLotView(DashboardView, FormView):
# template_name = "list_lots.html"
# title = _("Add to lots")
# breadcrumb = "lot / add to lots"
# success_url = reverse_lazy('dashboard:unassigned_devices')
# form_class = LotsForm
# def get_context_data(self, **kwargs):
# context = super().get_context_data(**kwargs)
# lots = Lot.objects.filter(owner=self.request.user)
# lot_tags = LotTag.objects.filter(owner=self.request.user)
# context.update({
# 'lots': lots,
# 'lot_tags':lot_tags,
# })
# return context
# def get_form(self):
# form = super().get_form()
# form.fields["lots"].queryset = Lot.objects.filter(owner=self.request.user)
# return form
# def form_valid(self, form):
# form.devices = self.get_session_devices()
# form.save()
# response = super().form_valid(form)
# return response
2024-07-08 10:59:32 +00:00
class EditDeviceView(DashboardView, UpdateView):
template_name = "new_device.html"
title = _("Update Device")
breadcrumb = "Device / Update Device"
success_url = reverse_lazy('dashboard:unassigned_devices')
2024-07-19 15:40:01 +00:00
model = Annotation
2024-07-08 10:59:32 +00:00
def get_form_kwargs(self):
pk = self.kwargs.get('pk')
self.object = get_object_or_404(self.model, pk=pk)
self.success_url = reverse_lazy('device:details', args=[pk])
kwargs = super().get_form_kwargs()
return kwargs
2024-07-05 13:32:07 +00:00
2024-07-19 15:40:01 +00:00
class DetailsView(DashboardView, TemplateView):
2024-07-15 14:23:14 +00:00
template_name = "details.html"
title = _("Device")
breadcrumb = "Device / Details"
2024-07-19 15:40:01 +00:00
model = Annotation
def get(self, request, *args, **kwargs):
self.pk = kwargs['pk']
self.object = Device(id=self.pk)
return super().get(request, *args, **kwargs)
2024-07-15 14:23:14 +00:00
def get_context_data(self, **kwargs):
context = super().get_context_data(**kwargs)
2024-07-18 15:21:22 +00:00
self.object.initial()
lot_tags = LotTag.objects.filter(owner=self.request.user)
2024-07-15 14:23:14 +00:00
context.update({
2024-07-19 15:40:01 +00:00
'object': self.object,
2024-07-18 15:21:22 +00:00
'snapshot': self.object.get_last_snapshot(),
'lot_tags': lot_tags,
2024-07-15 14:23:14 +00:00
})
return context
2024-07-05 13:32:07 +00:00
2024-07-18 15:21:22 +00:00
class AddAnnotationView(DashboardView, CreateView):
template_name = "new_device.html"
title = _("New annotation")
breadcrumb = "Device / New annotation"
2024-07-05 13:32:07 +00:00
success_url = reverse_lazy('dashboard:unassigned_devices')
2024-07-18 15:21:22 +00:00
model = Annotation
fields = ("key", "value")
2024-07-05 13:32:07 +00:00
2024-07-18 15:21:22 +00:00
def form_valid(self, form):
self.device.get_annotations()
self.device.get_uuids()
form.instance.owner = self.request.user
form.instance.device = self.device
form.instance.uuid = self.device.uuids[0]
form.instance.type = Annotation.Type.USER
response = super().form_valid(form)
return response
2024-07-08 10:59:32 +00:00
def get_form_kwargs(self):
pk = self.kwargs.get('pk')
2024-07-05 13:32:07 +00:00
self.device = get_object_or_404(Device, pk=pk)
2024-07-18 15:21:22 +00:00
self.success_url = reverse_lazy('device:details', args=[pk])
2024-07-08 10:59:32 +00:00
kwargs = super().get_form_kwargs()
2024-07-05 13:32:07 +00:00
return kwargs
2024-07-18 15:21:22 +00:00
def get_success_url(self):
url = super().get_success_url()
import pdb; pdb.set_trace()
return url
2024-06-12 07:32:49 +00:00