move mesh control into mesh module and add permissions
This commit is contained in:
Laura Klünder
parent
ce8f5f0084
commit
88d6f07eaf
27 changed files with 504 additions and 432 deletions
|
|
@ -1,290 +0,0 @@
|
|||
from functools import cached_property
|
||||
from uuid import uuid4
|
||||
|
||||
from django.contrib import messages
|
||||
from django.contrib.messages.views import SuccessMessageMixin
|
||||
from django.db.models import Max
|
||||
from django.http import Http404
|
||||
from django.shortcuts import redirect
|
||||
from django.urls import reverse
|
||||
from django.utils.translation import gettext_lazy as _
|
||||
from django.views.generic import DetailView, FormView, ListView, TemplateView, UpdateView
|
||||
|
||||
from c3nav.control.forms import MeshMessageFilterForm
|
||||
from c3nav.control.views.base import ControlPanelMixin
|
||||
from c3nav.mesh.forms import MeshMessageForm, MeshNodeForm
|
||||
from c3nav.mesh.messages import MeshMessage, MeshMessageType
|
||||
from c3nav.mesh.models import FirmwareBuild, FirmwareVersion, MeshNode, NodeMessage
|
||||
from c3nav.mesh.utils import get_node_names, group_msg_type_choices
|
||||
|
||||
|
||||
class MeshNodeListView(ControlPanelMixin, ListView):
|
||||
model = MeshNode
|
||||
template_name = "control/mesh_nodes.html"
|
||||
ordering = "address"
|
||||
context_object_name = "nodes"
|
||||
|
||||
def get_queryset(self):
|
||||
return super().get_queryset().annotate(last_msg=Max('received_messages__datetime')).prefetch_last_messages()
|
||||
|
||||
def get_context_data(self, *args, **kwargs):
|
||||
return {
|
||||
**super().get_context_data(*args, **kwargs),
|
||||
"send_msg_types": group_msg_type_choices({msg_type for msg_type in MeshMessageForm.msg_types.keys()})
|
||||
}
|
||||
|
||||
def post(self, request):
|
||||
return redirect(
|
||||
reverse("control.mesh.send", kwargs={"msg_type": request.POST.get("send_msg_type", "")})
|
||||
)
|
||||
|
||||
|
||||
class MeshNodeDetailView(ControlPanelMixin, DetailView):
|
||||
model = MeshNode
|
||||
template_name = "control/mesh_node_detail.html"
|
||||
pk_url_kwargs = "address"
|
||||
context_object_name = "node"
|
||||
|
||||
def get_queryset(self):
|
||||
return super().get_queryset().annotate(last_msg=Max('received_messages__datetime')).prefetch_last_messages()
|
||||
|
||||
|
||||
class MeshNodeEditView(ControlPanelMixin, SuccessMessageMixin, UpdateView):
|
||||
model = MeshNode
|
||||
form_class = MeshNodeForm
|
||||
template_name = "control/form.html"
|
||||
success_message = _('Name updated successfully')
|
||||
|
||||
def get_context_data(self, **kwargs):
|
||||
return {
|
||||
**super().get_context_data(),
|
||||
'title': _('Editing mesh node: %s') % self.get_object(),
|
||||
}
|
||||
|
||||
def get_success_url(self):
|
||||
return reverse('control.mesh.node.detail', kwargs={'pk': self.get_object().pk})
|
||||
|
||||
|
||||
class MeshMessageListView(ControlPanelMixin, ListView):
|
||||
model = NodeMessage
|
||||
template_name = "control/mesh_messages.html"
|
||||
ordering = "-datetime"
|
||||
paginate_by = 20
|
||||
context_object_name = "mesh_messages"
|
||||
|
||||
def get_queryset(self):
|
||||
qs = super().get_queryset()
|
||||
|
||||
self.form = MeshMessageFilterForm(self.request.GET)
|
||||
if self.form.is_valid():
|
||||
if self.form.cleaned_data['message_types']:
|
||||
qs = qs.filter(message_type__in=self.form.cleaned_data['message_types'])
|
||||
if self.form.cleaned_data['src_nodes']:
|
||||
qs = qs.filter(src_node__in=self.form.cleaned_data['src_nodes'])
|
||||
|
||||
return qs
|
||||
|
||||
def get_context_data(self, *args, **kwargs):
|
||||
ctx = super().get_context_data(*args, **kwargs)
|
||||
|
||||
form_data = self.request.GET.copy()
|
||||
form_data.pop('page', None)
|
||||
|
||||
ctx.update({
|
||||
"node_names": get_node_names(),
|
||||
'form': self.form,
|
||||
'form_data': form_data.urlencode(),
|
||||
})
|
||||
return ctx
|
||||
|
||||
|
||||
class MeshMessageSendView(ControlPanelMixin, FormView):
|
||||
template_name = "control/mesh_message_send.html"
|
||||
|
||||
@cached_property
|
||||
def msg_type(self):
|
||||
return MeshMessageType[self.kwargs['msg_type']]
|
||||
|
||||
def get_form_class(self):
|
||||
try:
|
||||
return MeshMessageForm.get_form_for_type(self.msg_type)
|
||||
except KeyError:
|
||||
raise Http404('unknown message type')
|
||||
|
||||
def get_form_kwargs(self):
|
||||
return {
|
||||
**super().get_form_kwargs(),
|
||||
'recipient': self.kwargs.get('recipient', None),
|
||||
}
|
||||
|
||||
def get_initial(self):
|
||||
if 'recipient' in self.kwargs and self.msg_type.name.startswith('CONFIG_'):
|
||||
try:
|
||||
node = MeshNode.objects.get(address=self.kwargs['recipient'])
|
||||
except MeshNode.DoesNotExist:
|
||||
pass
|
||||
else:
|
||||
initial = MeshMessage.get_type(self.msg_type).tojson(
|
||||
node.last_messages[self.msg_type].parsed
|
||||
)
|
||||
while keys := tuple(key for key, value in initial.items() if isinstance(value, dict)):
|
||||
for key in keys:
|
||||
subdict = initial.pop(key)
|
||||
for subkey, value in subdict.items():
|
||||
initial[key+"_"+subkey.removeprefix(key).lstrip('_')] = value
|
||||
return initial
|
||||
|
||||
if 'address' in self.request.GET and self.msg_type == MeshMessageType.MESH_ROUTE_REQUEST:
|
||||
return {"address": self.request.GET["address"]}
|
||||
|
||||
return {}
|
||||
|
||||
def get_success_url(self):
|
||||
if 'recipient' in self.kwargs and False:
|
||||
return reverse('control.mesh.node.detail', kwargs={'pk': self.kwargs['recipient']})
|
||||
else:
|
||||
return self.request.path
|
||||
|
||||
def form_valid(self, form):
|
||||
if 'noscript' in self.request.POST:
|
||||
form.send()
|
||||
messages.success(self.request, _('Message sent successfully(?)'))
|
||||
super().form_valid(form)
|
||||
uuid = uuid4()
|
||||
self.request.session["mesh_msg_%s" % uuid] = {
|
||||
"success_url": self.get_success_url(),
|
||||
"recipients": form.get_recipients(),
|
||||
"msg_data": form.get_msg_data(),
|
||||
}
|
||||
return redirect(reverse('control.mesh.sending', kwargs={'uuid': uuid}))
|
||||
|
||||
|
||||
class MeshMessageSendingView(ControlPanelMixin, TemplateView):
|
||||
template_name = "control/mesh_message_sending.html"
|
||||
|
||||
def get_context_data(self, uuid):
|
||||
try:
|
||||
data = self.request.session["mesh_msg_%s" % uuid]
|
||||
except KeyError:
|
||||
raise Http404
|
||||
node_names = get_node_names()
|
||||
return {
|
||||
**super().get_context_data(),
|
||||
"node_names": node_names,
|
||||
"send_uuid": uuid,
|
||||
**data,
|
||||
"node_name": node_names.get(data["msg_data"].get("address"), ""),
|
||||
"recipients": [(address, node_names[address]) for address in data["recipients"]],
|
||||
"msg_type": MeshMessageType[data["msg_data"]["msg_type"]].pretty_name,
|
||||
}
|
||||
|
||||
|
||||
class MeshLogView(ControlPanelMixin, TemplateView):
|
||||
template_name = "control/mesh_logs.html"
|
||||
|
||||
def get_context_data(self, **kwargs):
|
||||
return {
|
||||
**super().get_context_data(),
|
||||
"node_names": get_node_names(),
|
||||
}
|
||||
|
||||
|
||||
class MeshFirmwaresListView(ControlPanelMixin, ListView):
|
||||
model = FirmwareVersion
|
||||
template_name = "control/mesh_firmwares.html"
|
||||
ordering = "-created"
|
||||
context_object_name = "firmwares"
|
||||
paginate_by = 20
|
||||
|
||||
|
||||
class MeshFirmwaresCurrentListView(ControlPanelMixin, TemplateView):
|
||||
template_name = "control/mesh_firmwares_current.html"
|
||||
|
||||
def get_context_data(self, **kwargs):
|
||||
nodes = list(MeshNode.objects.all().prefetch_firmwares())
|
||||
|
||||
firmwares = {}
|
||||
for node in nodes:
|
||||
firmwares.setdefault(node.firmware_desc.get_lookup(), (node.firmware_desc, []))[1].append(node)
|
||||
|
||||
firmwares = sorted(firmwares.values(), key=lambda k: k[0].created, reverse=True)
|
||||
|
||||
print(firmwares)
|
||||
|
||||
return {
|
||||
**super().get_context_data(),
|
||||
"firmwares": firmwares,
|
||||
}
|
||||
|
||||
|
||||
class MeshFirmwareDetailView(ControlPanelMixin, DetailView):
|
||||
model = FirmwareVersion
|
||||
template_name = "control/mesh_firmware_detail.html"
|
||||
context_object_name = "firmware"
|
||||
|
||||
def get_queryset(self):
|
||||
return super().get_queryset().prefetch_related('builds', 'builds__firmwarebuildboard_set')
|
||||
|
||||
def get_context_data(self, **kwargs):
|
||||
ctx = super().get_context_data()
|
||||
|
||||
nodes = list(MeshNode.objects.all().prefetch_firmwares().prefetch_last_messages(
|
||||
MeshMessageType.CONFIG_BOARD,
|
||||
))
|
||||
builds = self.get_object().builds.all()
|
||||
|
||||
build_lookups = set(build.get_firmware_description().get_lookup() for build in builds)
|
||||
|
||||
installed_nodes = []
|
||||
compatible_nodes = []
|
||||
for node in nodes:
|
||||
if node.firmware_desc.get_lookup() in build_lookups:
|
||||
installed_nodes.append(node)
|
||||
else:
|
||||
node.compatible_builds = []
|
||||
for build in builds:
|
||||
if node.board in build.boards:
|
||||
node.compatible_builds.append(build)
|
||||
if node.compatible_builds:
|
||||
compatible_nodes.append(node)
|
||||
|
||||
ctx.update({
|
||||
'builds': builds,
|
||||
'installed_nodes': installed_nodes,
|
||||
'compatible_nodes': compatible_nodes,
|
||||
})
|
||||
return ctx
|
||||
|
||||
|
||||
class MeshFirmwareBuildDetailView(ControlPanelMixin, DetailView):
|
||||
model = FirmwareBuild
|
||||
template_name = "control/mesh_firmware_build_detail.html"
|
||||
context_object_name = "build"
|
||||
|
||||
def get_queryset(self):
|
||||
return super().get_queryset().prefetch_related('firmwarebuildboard_set')
|
||||
|
||||
def get_context_data(self, **kwargs):
|
||||
ctx = super().get_context_data()
|
||||
|
||||
nodes = list(MeshNode.objects.all().prefetch_firmwares().prefetch_last_messages(
|
||||
MeshMessageType.CONFIG_BOARD,
|
||||
))
|
||||
|
||||
build_lookup = self.get_object().get_firmware_description().get_lookup()
|
||||
build_boards = self.get_object().boards
|
||||
|
||||
installed_nodes = []
|
||||
compatible_nodes = []
|
||||
for node in nodes:
|
||||
if node.firmware_desc.get_lookup() == build_lookup:
|
||||
installed_nodes.append(node)
|
||||
else:
|
||||
if node.board in build_boards:
|
||||
compatible_nodes.append(node)
|
||||
|
||||
ctx.update({
|
||||
'installed_nodes': installed_nodes,
|
||||
'compatible_nodes': compatible_nodes,
|
||||
})
|
||||
return ctx
|
||||
Loading…
Add table
Add a link
Reference in a new issue