1
1
from django .http .response import HttpResponseRedirect
2
2
from django .contrib import messages
3
3
from django .views .generic import ListView , DetailView , CreateView , FormView
4
- from django_tables2 import RequestConfig
5
4
from django .urls import reverse
6
5
from django .utils .translation import ugettext as _
7
6
7
+ from django_tables2 import RequestConfig
8
8
from django_tables2 .export .views import ExportMixin
9
9
10
10
from . import models , tables , filters , forms
@@ -45,6 +45,26 @@ class PublicKeyDetailView(AdminOrAuthorityQsMixin, DetailView):
45
45
model = models .PublicKey
46
46
47
47
48
+ class KeyPairCreateView (CreateView ):
49
+ model = models .PublicKey
50
+ form_class = forms .KeyPairCreateForm
51
+ template_name = 'keys/key_pair_create.html'
52
+
53
+ def dispatch (self , request , * args , ** kwargs ):
54
+ self .user_authority = self .request .user .get_authority ()
55
+ if not self .user_authority :
56
+ messages .error (
57
+ self .request ,
58
+ "Key pair creation is not allowed from users without an authority!" ,
59
+ )
60
+ return HttpResponseRedirect (reverse ("home" ))
61
+ return super ().dispatch (request , * args , ** kwargs )
62
+
63
+
64
+ def form_valid (self , form ):
65
+ a += 1
66
+
67
+
48
68
class PublicKeyCreateView (CreateView ):
49
69
model = models .PublicKey
50
70
form_class = forms .PublicKeyCreateForm
@@ -58,7 +78,7 @@ def form_valid(self, form):
58
78
)
59
79
return HttpResponseRedirect (reverse ("home" ))
60
80
form .instance .authority = user_authority
61
- obj = form .save ()
81
+ _obj = form .save ()
62
82
63
83
messages .add_message (
64
84
self .request ,
@@ -72,18 +92,22 @@ def form_valid(self, form):
72
92
73
93
class LoadPrivateKey (FormView ):
74
94
form_class = forms .LoadPrivateKeyForm
75
- template_name = ' keys/load_private_key.html'
95
+ template_name = " keys/load_private_key.html"
76
96
77
97
def form_valid (self , form ):
78
- fingerprint = form .cleaned_data [' fingerprint' ]
79
- user_id = form .cleaned_data [' user_id' ]
80
- self .request .session [' private_key_data' ] = {
81
- ' fingerprint' : fingerprint ,
82
- ' user_id' : user_id ,
98
+ fingerprint = form .cleaned_data [" fingerprint" ]
99
+ user_id = form .cleaned_data [" user_id" ]
100
+ self .request .session [" private_key_data" ] = {
101
+ " fingerprint" : fingerprint ,
102
+ " user_id" : user_id ,
83
103
}
84
- messages .add_message (self .request , messages .SUCCESS , _ (
85
- "Private Key has been loaded. User id: {0}, fingerprint {1}" .format (
86
- user_id , fingerprint
87
- )
88
- ))
104
+ messages .add_message (
105
+ self .request ,
106
+ messages .SUCCESS ,
107
+ _ (
108
+ "Private Key has been loaded. User id: {0}, fingerprint {1}" .format (
109
+ user_id , fingerprint
110
+ )
111
+ ),
112
+ )
89
113
return HttpResponseRedirect (reverse ("home" ))
0 commit comments