62 lines
2.5 KiB
Python
62 lines
2.5 KiB
Python
from django.views import View
|
|
from django.shortcuts import redirect, get_object_or_404
|
|
from django.contrib import messages
|
|
from action.forms import AddStateForm
|
|
from django.views.generic.edit import DeleteView
|
|
from django.urls import reverse_lazy
|
|
from action.models import State, StateDefinition
|
|
from device.models import Device
|
|
import logging
|
|
|
|
|
|
device_logger = logging.getLogger('device_log')
|
|
|
|
|
|
class NewActionView(View):
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
form = AddStateForm(request.POST)
|
|
|
|
if form.is_valid():
|
|
state_definition_id = form.cleaned_data['state_id']
|
|
state_definition = get_object_or_404(StateDefinition, pk=state_definition_id)
|
|
snapshot_uuid = form.cleaned_data['snapshot_uuid']
|
|
#TODO: implement notes
|
|
note = form.cleaned_data.get('note', '')
|
|
|
|
state = State.objects.create(
|
|
snapshot_uuid=snapshot_uuid,
|
|
state=state_definition.state,
|
|
user=request.user,
|
|
institution=request.user.institution,
|
|
)
|
|
#TODO: also change logger for full fledged table
|
|
device_logger.info(f"<Updated> State to '{state_definition.state}', for device '{snapshot_uuid}') by user {self.request.user}.")
|
|
|
|
messages.success(request, f"Action to '{state_definition.state}' has been added.")
|
|
return redirect(request.META.get('HTTP_REFERER'))
|
|
else:
|
|
messages.error(request, "There was an error with your submission.")
|
|
return redirect(request.META.get('HTTP_REFERER'))
|
|
|
|
|
|
class ActionUndoView(DeleteView):
|
|
model = State
|
|
|
|
def delete(self, request, *args, **kwargs):
|
|
self.object = self.get_object()
|
|
time_since_creation = timezone.now() - self.object.date
|
|
if time_since_creation.total_seconds() <= 3600: # 1 hour = 3600 seconds
|
|
data_to_return= super().delete(request, *args, **kwarg)
|
|
|
|
messages.success(request, f"Action to state'{self.object.state}' has been delete.")
|
|
return data_to_return
|
|
else:
|
|
#TODO: revise if condition is correct
|
|
messages.error(request, "You can undo an action within one hour of its creation.")
|
|
return reverse_lazy(self.get_success_url())
|
|
|
|
def get_success_url(self):
|
|
device_logger.info(f"<Deleted> State '{self.object.state}', for device '{self.object.snapshot_uuid}') by user {self.request.user}.")
|
|
return self.request.META.get('HTTP_REFERER', reverse_lazy('device:details', args=[self.object.snapshot_uuid]))
|