Working invites in the Driver's list page

1 files changed, 18 insertions, 1 deletions

diff --git a/app/dispatch/views.py b/app/dispatch/views.py
index 62fc73f..784da9e 100644
--- a/app/dispatch/views.py
+++ b/app/dispatch/views.py
@@ -14,7 +14,7 @@ from django.views.generic.edit import CreateView, UpdateView, DeleteView
 from django.core.urlresolvers import reverse_lazy
 from dispatch.models import Customer, Load, Paperwork, \
         Invoice, UserInvoiceNumber, Identity, Settings
-from dispatch.forms import AddPaperworkForm
+from dispatch.forms import AddPaperworkForm, InviteForm
 from django.contrib.auth import get_user_model
 from django.contrib.auth.mixins import UserPassesTestMixin
 # from django.http import HttpResponseRedirect
@@ -362,6 +362,23 @@ class LoadDelete(UserPassesTestMixin, FilteredDeleteView):
         return self.request.user.is_superuser
 
 
+def InviteUser(request):
+    if request.method == 'POST':
+        form = InviteForm(request.POST)
+        if form.is_valid():
+            request.user.send_invite(
+                    form.cleaned_data.get('first_name'),
+                    form.cleaned_data.get('email_address')
+                    )
+            return redirect(reverse('driver_list'))
+    else:
+        form = InviteForm()
+
+    ctx = {'form': form}
+
+    return render(request, 'dispatch/invite_user.html', ctx)
+
+
 # Paperwork Uploads
 
 def PaperworkUpload(request, load_id):