update

[cas.git] / src / services / views.py

diff --git a/src/services/views.py b/src/services/views.py
index f599592..c2c4a06 100644 (file)
--- a/src/services/views.py
+++ b/src/services/views.py
@@ -1,4 +1,5 @@
-from django.views.generic import DetailView
+from django.contrib.auth.mixins import LoginRequiredMixin
+from django.views.generic import DetailView, ListView
 from .models import Service
 
 
@@ -13,3 +14,12 @@ class SshAuthorizedKeysView(DetailView):
             obj = None
         return obj
 
+
+class ServicesView(LoginRequiredMixin, ListView):
+    def get_queryset(self):
+        return Service.for_user(self.request.user)
+
+
+class ServiceDetail(LoginRequiredMixin, DetailView):
+    def get_queryset(self):
+        return Service.for_user(self.request.user)