reworked invite system

cookbook/views/new.py

@@ -1,7 +1,11 @@
 import re
-from datetime import datetime
+from datetime import datetime, timedelta
+from html import escape
+from smtplib import SMTPException
 
 from django.contrib import messages
+from django.contrib.auth.models import Group
+from django.core.mail import send_mail, BadHeaderError
 from django.http import HttpResponseRedirect
 from django.shortcuts import get_object_or_404, redirect, render
 from django.urls import reverse, reverse_lazy
@@ -154,7 +158,8 @@ class MealPlanCreate(GroupRequiredMixin, CreateView, SpaceFormMixing):
 
     def get_form(self, form_class=None):
         form = self.form_class(**self.get_form_kwargs())
-        form.fields['meal_type'].queryset = MealType.objects.filter(created_by=self.request.user, space=self.request.space).all()
+        form.fields['meal_type'].queryset = MealType.objects.filter(created_by=self.request.user,
+                                                                    space=self.request.space).all()
         return form
 
     def get_initial(self):
@@ -207,6 +212,32 @@ class InviteLinkCreate(GroupRequiredMixin, CreateView):
         obj.created_by = self.request.user
         obj.space = self.request.space
         obj.save()
+        if obj.email:
+            try:
+                if InviteLink.objects.filter(space=self.request.space, created_at__gte=datetime.now() - timedelta(hours=4)).count() < 20:
+                    message = _('Hello') + '!\n\n' + _('You have been invited by ') + escape(self.request.user.username)
+                    message += _(' to join their Tandoor Recipes space ') + escape(self.request.space.name) + '.\n\n'
+                    message += _('Click the following link to activate your account: ') + self.request.build_absolute_uri(reverse('view_signup', args=[str(obj.uuid)])) + '\n\n'
+                    message += _('If the link does not work use the following code to manually join the space: ') + str(obj.uuid) + '\n\n'
+                    message += _('The invitation is valid until ') + obj.valid_until + '\n\n'
+                    message += _('Tandoor Recipes is an Open Source recipe manager. Check it out on GitHub ') + 'https://github.com/vabene1111/recipes/'
+
+                    send_mail(
+                        _('Tandoor Recipes Invite'),
+                        message,
+                        None,
+                        [obj.email],
+                        fail_silently=False,
+
+                    )
+                    messages.add_message(self.request, messages.SUCCESS,
+                                         _('Invite link successfully send to user.'))
+                else:
+                    messages.add_message(self.request, messages.ERROR,
+                                         _('You have send to many emails, please share the link manually or wait a few hours.'))
+            except (SMTPException, BadHeaderError, TimeoutError):
+                messages.add_message(self.request, messages.ERROR, _('Email to user could not be send, please share link manually.'))
+
         return HttpResponseRedirect(reverse('list_invite_link'))
 
     def get_context_data(self, **kwargs):
@@ -218,3 +249,9 @@ class InviteLinkCreate(GroupRequiredMixin, CreateView):
         kwargs = super().get_form_kwargs()
         kwargs.update({'user': self.request.user})
         return kwargs
+
+    def get_initial(self):
+        return dict(
+            space=self.request.space,
+            group=Group.objects.get(name='user')
+        )