From 399c91d636dbe775ccdc00614a74484c6d7dd91c Mon Sep 17 00:00:00 2001
From: PCoder <purple.coder@yahoo.co.uk>
Date: Sat, 7 Apr 2018 12:51:40 +0200
Subject: [PATCH] Add HostingContextMixin that sets the REDIRECT_TO_CMS and
 MULTISITE_CMS_FALLBACK urls

---
 hosting/mixins.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/hosting/mixins.py b/hosting/mixins.py
index 666cbcbe..7ef05581 100644
--- a/hosting/mixins.py
+++ b/hosting/mixins.py
@@ -1,4 +1,7 @@
+from cms.models.pagemodel import Page
 from django.shortcuts import redirect
+from django.conf import settings
+from django.contrib.sites.models import Site
 from django.core.urlresolvers import reverse
 
 from opennebula_api.serializers import VirtualMachineTemplateSerializer
@@ -24,3 +27,17 @@ class ProcessVMSelectionMixin(object):
             request.session['next'] = reverse('hosting:payment')
             return redirect(reverse('hosting:login'))
         return redirect(reverse('hosting:payment'))
+
+
+class HostingContextMixin(object):
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        context['REDIRECT_TO_CMS'] = (Page
+                                      .objects
+                                      .filter(site_id=Site
+                                              .objects
+                                              .get_current()
+                                              .id)
+                                      .count())
+        context['MULTISITE_CMS_FALLBACK'] = settings.MULTISITE_CMS_FALLBACK
+        return context