46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
from django.contrib.messages.views import SuccessMessageMixin
|
|
from django.shortcuts import redirect
|
|
from django.urls import reverse
|
|
from django.utils.translation import gettext_lazy as _
|
|
from django.views.generic import DetailView, ListView, UpdateView
|
|
|
|
from c3nav.mesh.forms import MeshNodeForm
|
|
from c3nav.mesh.models import MeshNode
|
|
from c3nav.mesh.views.base import MeshControlMixin
|
|
|
|
|
|
class NodeListView(MeshControlMixin, ListView):
|
|
model = MeshNode
|
|
template_name = "mesh/nodes.html"
|
|
ordering = "address"
|
|
context_object_name = "nodes"
|
|
|
|
def get_queryset(self):
|
|
return super().get_queryset().prefetch_last_messages().prefetch_firmwares().prefetch_ranging_beacon()
|
|
|
|
|
|
|
|
class NodeDetailView(MeshControlMixin, DetailView):
|
|
model = MeshNode
|
|
template_name = "mesh/node_detail.html"
|
|
pk_url_kwargs = "address"
|
|
context_object_name = "node"
|
|
|
|
def get_queryset(self):
|
|
return super().get_queryset().prefetch_last_messages().prefetch_firmwares()
|
|
|
|
|
|
class NodeEditView(MeshControlMixin, 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('mesh.node.detail', kwargs={'pk': self.get_object().pk})
|