devicehub-django/snapshot/views.py

from django.utils.translation import gettext_lazy as _
from django.views.generic.base import TemplateView
from django.views.generic.edit import FormView
from django.urls import reverse_lazy

from dashboard.mixins import  DashboardView
from snapshot.models import Snapshot
# from snapshot.forms import UploadForm
# from django.shortcuts import render
# from rest_framework import viewsets
# from snapshot.serializers import SnapshotSerializer


# class SnapshotViewSet(viewsets.ModelViewSet):
#     queryset = Snapshot.objects.all()
#     serializer_class = SnapshotSerializer


class ListSnapshotsView(DashboardView, TemplateView):
    template_name = "snapshots.html"
    section = "snapshots"
    title = _("Snapshots")
    breadcrumb = "Snapshots"

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        snapshots = Snapshot.objects.filter(owner=self.request.user)
        context.update({
            'snapshots': snapshots,
        })
        return context


# class UploadView(DashboardView, FormView):
#     template_name = "upload.html"
#     section = "snapshots"
#     title = _("Upload Snapshot")
#     breadcrumb = "Snapshots / Upload"
#     success_url = reverse_lazy('snashot:list')
#     form_class = UploadForm