|
|
@@ -98,13 +98,14 @@ class SecretEditView(generic.ObjectEditView):
|
|
|
|
|
|
if form.is_valid():
|
|
|
logger.debug("Form validation was successful")
|
|
|
+ secret = form.save(commit=False)
|
|
|
|
|
|
- # We must have a session key in order to create a secret or update the plaintext of an existing secret
|
|
|
- if (form.cleaned_data['plaintext'] or secret.pk is None) and session_key is None:
|
|
|
+ # We must have a session key in order to set the plaintext of a Secret
|
|
|
+ if form.cleaned_data['plaintext'] and session_key is None:
|
|
|
logger.debug("Unable to proceed: No session key was provided with the request")
|
|
|
form.add_error(None, "No session key was provided with the request. Unable to encrypt secret data.")
|
|
|
|
|
|
- else:
|
|
|
+ elif form.cleaned_data['plaintext']:
|
|
|
master_key = None
|
|
|
try:
|
|
|
sk = SessionKey.objects.get(userkey__user=request.user)
|
|
|
@@ -115,19 +116,18 @@ class SecretEditView(generic.ObjectEditView):
|
|
|
|
|
|
if master_key is not None:
|
|
|
logger.debug("Successfully resolved master key for encryption")
|
|
|
- secret = form.save(commit=False)
|
|
|
- if form.cleaned_data['plaintext']:
|
|
|
- secret.plaintext = str(form.cleaned_data['plaintext'])
|
|
|
+ secret.plaintext = str(form.cleaned_data['plaintext'])
|
|
|
secret.encrypt(master_key)
|
|
|
- secret.save()
|
|
|
- form.save_m2m()
|
|
|
|
|
|
- msg = '{} secret'.format('Created' if not form.instance.pk else 'Modified')
|
|
|
- logger.info(f"{msg} {secret} (PK: {secret.pk})")
|
|
|
- msg = '{} <a href="{}">{}</a>'.format(msg, secret.get_absolute_url(), escape(secret))
|
|
|
- messages.success(request, mark_safe(msg))
|
|
|
+ secret.save()
|
|
|
+ form.save_m2m()
|
|
|
+
|
|
|
+ msg = '{} secret'.format('Created' if not form.instance.pk else 'Modified')
|
|
|
+ logger.info(f"{msg} {secret} (PK: {secret.pk})")
|
|
|
+ msg = f'{msg} <a href="{secret.get_absolute_url()}">{escape(secret)}</a>'
|
|
|
+ messages.success(request, mark_safe(msg))
|
|
|
|
|
|
- return redirect(self.get_return_url(request, secret))
|
|
|
+ return redirect(self.get_return_url(request, secret))
|
|
|
|
|
|
else:
|
|
|
logger.debug("Form validation failed")
|