X-Git-Url: https://git.mdrn.pl/cas.git/blobdiff_plain/3a65fcbfa423d5627e41c2f9e17c464d4a594087..5c901ee485c3abdf11dec71bdb7534f1f6a9eaa9:/src/ssh_keys/views.py?ds=inline

diff --git a/src/ssh_keys/views.py b/src/ssh_keys/views.py
new file mode 100644
index 0000000..684dcf6
--- /dev/null
+++ b/src/ssh_keys/views.py
@@ -0,0 +1,27 @@
+from django.contrib.auth.mixins import LoginRequiredMixin
+from django.views.generic import ListView, CreateView, DeleteView
+from .models import SSHKey
+
+
+class SSHKeysView(LoginRequiredMixin, ListView):
+    def get_queryset(self):
+        return SSHKey.objects.filter(user=self.request.user)
+
+
+class AddSSHKeyView(LoginRequiredMixin, CreateView):
+    fields = ['key']
+    model = SSHKey
+    success_url = '/ssh/'
+    template_name = 'ssh_keys/sshkey_add.html'
+
+    def form_valid(self, form):
+        form.instance.user = self.request.user
+        return super().form_valid(form)
+
+    
+class DeleteSSHKeyView(LoginRequiredMixin, DeleteView):
+    success_url = '/ssh/'
+
+    def get_queryset(self):
+        return SSHKey.objects.filter(user=self.request.user)
+