dashboard: notify owners about ownership transfer

circle/dashboard/templates/dashboard/notifications/ownership-accepted.html

+{%load i18n%}
+{%blocktrans with instance=instance.name user=user.name%}
+Your ownership offer of {{instance}} has been accepted by {{user}}.
+{%endblocktrans%}

circle/dashboard/templates/dashboard/notifications/ownership-offer.html

+{%load i18n%}
+{%blocktrans with instance=instance.name user=user.name%}
+{{user}} offered you to take the ownership of his/her virtual machine
+called {{instance}}.{%endblocktrans%} 
+<a href="{{token}}" class="btn btn-success btn-small">{%trans "Accept"%}</a>

circle/dashboard/views.py

@@ -37,7 +37,7 @@ from vm.models import (Instance, InstanceTemplate, InterfaceTemplate,
                        InstanceActivity, Node, instance_activity, Lease,
                        InstanceActivity, Node, instance_activity, Lease,
                        Interface, NodeActivity)
                        Interface, NodeActivity)
 from firewall.models import Vlan, Host, Rule
 from firewall.models import Vlan, Host, Rule
-from dashboard.models import Favourite
+from dashboard.models import Favourite, Profile
 
 
 logger = logging.getLogger(__name__)
 logger = logging.getLogger(__name__)
 
 
@@ -1490,7 +1490,12 @@ class TransferOwnershipView(LoginRequiredMixin, DetailView):
         try:
         try:
             new_owner = User.objects.get(username=request.POST['name'])
             new_owner = User.objects.get(username=request.POST['name'])
         except User.DoesNotExist:
         except User.DoesNotExist:
-            raise Http404()
+            new_owner = User.objects.get(email=request.POST['name'])
+        except User.DoesNotExist:
+            new_owner = User.objects.get(profile__org_id=request.POST['name'])
+        except User.DoesNotExist:
+            messages.error(request, _('Can not find specified user.'))
+            return self.get(request, *args, **kwargs)
         except KeyError:
         except KeyError:
             raise SuspiciousOperation()
             raise SuspiciousOperation()
 
 
@@ -1501,9 +1506,22 @@ class TransferOwnershipView(LoginRequiredMixin, DetailView):
 
 
         token = signing.dumps((obj.pk, new_owner.pk),
         token = signing.dumps((obj.pk, new_owner.pk),
                               salt=TransferOwnershipConfirmView.get_salt())
                               salt=TransferOwnershipConfirmView.get_salt())
-        return HttpResponse("%s?key=%s" % (
-            reverse('dashboard.views.vm-transfer-ownership-confirm'), token),
-            content_type="text/plain")
+        token_path = reverse(
+            'dashboard.views.vm-transfer-ownership-confirm', args=[token])
+        try:
+            new_owner.profile.notify(
+                _('Ownership offer'),
+                'dashboard/notifications/ownership-offer.html',
+                {'instance': obj, 'token': token_path})
+        except Profile.DoesNotExist:
+            messages.error(request, _('Can not notify selected user.'))
+        else:
+            messages.success(request,
+                             _('User %s is notified about the offer.') % (
+                                 unicode(new_owner), ))
+
+        return redirect(reverse_lazy("dashboard.views.detail",
+                                     kwargs={'pk': obj.pk}))
 
 
 
 
 class TransferOwnershipConfirmView(LoginRequiredMixin, View):
 class TransferOwnershipConfirmView(LoginRequiredMixin, View):
@@ -1546,6 +1564,11 @@ class TransferOwnershipConfirmView(LoginRequiredMixin, View):
         messages.success(request, self.success_message)
         messages.success(request, self.success_message)
         logger.info('Ownership of %s transferred from %s to %s.',
         logger.info('Ownership of %s transferred from %s to %s.',
                     unicode(instance), unicode(old), unicode(request.user))
                     unicode(instance), unicode(old), unicode(request.user))
+        if old.profile:
+            old.profile.notify(
+                _('Ownership accepted'),
+                'dashboard/notifications/ownership-accepted.html',
+                {'instance': instance})
         return HttpResponseRedirect(instance.get_absolute_url())
         return HttpResponseRedirect(instance.get_absolute_url())
 
 
     def get_instance(self, key, user):
     def get_instance(self, key, user):
@@ -1555,15 +1578,7 @@ class TransferOwnershipConfirmView(LoginRequiredMixin, View):
             instance, new_owner = (
             instance, new_owner = (
                 signing.loads(key, max_age=self.max_age,
                 signing.loads(key, max_age=self.max_age,
                               salt=self.get_salt()))
                               salt=self.get_salt()))
-        except signing.BadSignature as e:
-            logger.error('Tried invalid token. Token: %s, user: %s. %s',
-                         key, unicode(user), unicode(e))
-            raise SuspiciousOperation()
-        except ValueError as e:
-            logger.error('Tried invalid token. Token: %s, user: %s. %s',
-                         key, unicode(user), unicode(e))
-            raise SuspiciousOperation()
-        except TypeError as e:
+        except (signing.BadSignature, ValueError, TypeError) as e:
             logger.error('Tried invalid token. Token: %s, user: %s. %s',
             logger.error('Tried invalid token. Token: %s, user: %s. %s',
                          key, unicode(user), unicode(e))
                          key, unicode(user), unicode(e))
             raise SuspiciousOperation()
             raise SuspiciousOperation()