|
10 | 10 | from django.core.mail import send_mail |
11 | 11 | from django.db.models import Count |
12 | 12 | from django.db.models import Q |
13 | | -from django.http import HttpResponse |
14 | 13 | from django.http.response import Http404 |
15 | 14 | from django.shortcuts import render |
16 | 15 | from django.urls import reverse_lazy |
17 | 16 | from django.views import View |
18 | 17 | from django.views import generic |
19 | 18 | from django.views.generic.detail import DetailView |
| 19 | +from django.contrib import messages |
20 | 20 | from django.views.generic.list import ListView |
21 | 21 | from packageurl import PackageURL |
22 | 22 |
|
@@ -235,18 +235,26 @@ class ApiUserCreateView(generic.CreateView): |
235 | 235 | template_name = "api_user_creation_form.html" |
236 | 236 |
|
237 | 237 | def form_valid(self, form): |
238 | | - super().form_valid(form) |
| 238 | + response = super().form_valid(form) |
239 | 239 |
|
240 | 240 | send_mail( |
241 | 241 | subject="VulnerableCode.io API key token", |
242 | 242 | message=f"Here is your VulnerableCode.io API key token: {self.object.auth_token}", |
243 | 243 | from_email=env.str("FROM_EMAIL", default=""), |
244 | 244 | recipient_list=[self.object.email], |
| 245 | + fail_silently=True, |
245 | 246 | ) |
246 | 247 |
|
247 | | - return HttpResponse( |
248 | | - f"Check your email for VulnerableCode.io API key token: {self.object.email}" |
249 | | - ) |
| 248 | + messages.success(self.request, f"API key token sent to your email address {self.object.email}.") |
| 249 | + |
| 250 | + return response |
| 251 | + |
| 252 | + def form_invalid(self, form): |
| 253 | + response = super().form_invalid(form) |
| 254 | + |
| 255 | + messages.error(self.request, str(form.errors)) |
| 256 | + |
| 257 | + return response |
250 | 258 |
|
251 | 259 | def get_success_url(self): |
252 | | - return reverse_lazy("home") |
| 260 | + return reverse_lazy("api_user_request") |
0 commit comments