Support adding websites to your profile.

nkdsu/apps/vote/views/profiles.py

@@ -4,6 +4,7 @@
 from django.contrib.auth import get_user_model
 from django.contrib.auth.base_user import AbstractBaseUser
 from django.contrib.auth.mixins import LoginRequiredMixin
+from django.core.exceptions import ValidationError
 from django.db.models import QuerySet
 from django.http import HttpRequest, HttpResponse
 from django.shortcuts import get_object_or_404, redirect
@@ -51,6 +52,19 @@ def post(self, request: HttpRequest, username: str) -> HttpResponse:
                 messages.success(self.request, f"website {website.url!r} removed from your profile")
                 return redirect('.')
 
+        if request.POST.get('add-website') == 'yes' and request.POST.get('url'):
+            if user.profile.has_max_websites():
+                messages.warning(self.request, "don't you think you have enough websites already")
+                return redirect('.')
+            else:
+                try:
+                    website = user.profile.websites.create(url=request.POST['url'])
+                except ValidationError as e:
+                    messages.warning(self.request, ', '.join(e.messages))
+                    return redirect('.')
+                messages.success(self.request, f"website {website.url} added to your profile")
+                return redirect('.')
+
         return redirect('.')
 
     def get_voter(self) -> Profile:

nkdsu/templates/include/voter_meta.html

@@ -74,7 +74,7 @@ <h2>
               <input type="text" name="url" placeholder="https://mycoolwebsite.gov"/>
             </p>
             <p class="submit">
-              <button class="button" type="submit">Add website</button>
+              <button class="button" name="add-website" value="yes" type="submit">Add website</button>
             </p>
           </form>
         </details>