From 447ddc7c2eec77b7651e2d79779b773fcaa13e47 Mon Sep 17 00:00:00 2001
From: Levi <levinoelvm@gmail.com>
Date: Thu, 29 Sep 2016 00:04:40 -0500
Subject: [PATCH] finishing digitalglarus booking

---
 .../migrations/0018_auto_20160928_0424.py     | 27 +++++++++++
 .../migrations/0019_auto_20160929_0324.py     | 25 ++++++++++
 digitalglarus/mixins.py                       |  4 +-
 digitalglarus/models.py                       | 36 ++++++++++++--
 .../digitalglarus/booking_orders_detail.html  | 10 +++-
 .../digitalglarus/booking_payment.html        | 11 +++--
 digitalglarus/views.py                        | 48 +++++++++++++------
 7 files changed, 137 insertions(+), 24 deletions(-)
 create mode 100644 digitalglarus/migrations/0018_auto_20160928_0424.py
 create mode 100644 digitalglarus/migrations/0019_auto_20160929_0324.py

diff --git a/digitalglarus/migrations/0018_auto_20160928_0424.py b/digitalglarus/migrations/0018_auto_20160928_0424.py
new file mode 100644
index 00000000..10a92271
--- /dev/null
+++ b/digitalglarus/migrations/0018_auto_20160928_0424.py
@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.9.4 on 2016-09-28 04:24
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('digitalglarus', '0017_membership_active'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='bookingorder',
+            name='membership_required_months',
+            field=models.IntegerField(default=0),
+            preserve_default=False,
+        ),
+        migrations.AddField(
+            model_name='bookingorder',
+            name='membership_required_months_price',
+            field=models.FloatField(default=0),
+            preserve_default=False,
+        ),
+    ]
diff --git a/digitalglarus/migrations/0019_auto_20160929_0324.py b/digitalglarus/migrations/0019_auto_20160929_0324.py
new file mode 100644
index 00000000..f78cf3e7
--- /dev/null
+++ b/digitalglarus/migrations/0019_auto_20160929_0324.py
@@ -0,0 +1,25 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.9.4 on 2016-09-29 03:24
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('digitalglarus', '0018_auto_20160928_0424'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='bookingorder',
+            name='membership_required_months',
+            field=models.IntegerField(default=0),
+        ),
+        migrations.AlterField(
+            model_name='bookingorder',
+            name='membership_required_months_price',
+            field=models.FloatField(default=0),
+        ),
+    ]
diff --git a/digitalglarus/mixins.py b/digitalglarus/mixins.py
index 40777bf3..013cdd78 100644
--- a/digitalglarus/mixins.py
+++ b/digitalglarus/mixins.py
@@ -9,7 +9,7 @@ class MembershipRequiredMixin(object):
 
     def dispatch(self, request, *args, **kwargs):
         from .models import Membership
-        if not Membership.is_digitalglarus_member(request.user):
+        if not Membership.is_digitalglarus_active_member(request.user):
             return HttpResponseRedirect(self.membership_redirect_url)
 
         return super(MembershipRequiredMixin, self).dispatch(request, *args, **kwargs)
@@ -20,7 +20,7 @@ class IsNotMemberMixin(object):
 
     def dispatch(self, request, *args, **kwargs):
         from .models import Membership
-        if Membership.is_digitalglarus_member(request.user):
+        if Membership.is_digitalglarus_active_member(request.user):
             return HttpResponseRedirect(self.already_member_redirect_url)
 
         return super(IsNotMemberMixin, self).dispatch(request, *args, **kwargs)
diff --git a/digitalglarus/models.py b/digitalglarus/models.py
index 5fdf6607..0147d53b 100644
--- a/digitalglarus/models.py
+++ b/digitalglarus/models.py
@@ -16,8 +16,9 @@ from .mixins import Ordereable
 
 class MembershipType(models.Model):
 
+    STANDARD = 'standard'
     MEMBERSHIP_TYPES = (
-        ('standard', 'Standard'),
+        (STANDARD, 'Standard'),
 
     )
     name = models.CharField(choices=MEMBERSHIP_TYPES, max_length=20)
@@ -63,7 +64,7 @@ class Membership(models.Model):
     active = models.BooleanField(default=True)
 
     @classmethod
-    def is_digitalglarus_member(cls, user):
+    def is_digitalglarus_active_member(cls, user):
         past_month = (datetime.today() - relativedelta(months=1)).month
         has_booking_current_month = Q(membershiporder__customer__user=user,
                                       membershiporder__created_at__month=datetime.today().month)
@@ -158,6 +159,14 @@ class Booking(models.Model):
         total_free_days = months * TWO_DAYS + free_days_this_month
         return total_free_days
 
+    @staticmethod
+    def membership_required_booking_months(start_date, end_date):
+        start_month = start_date.month
+        end_month = end_date.month
+        months = abs(start_month - (end_month + 12) if end_month < start_month
+                     else end_month - start_month)
+        return months
+
     @classmethod
     def booking_price(cls, user, start_date, end_date):
 
@@ -178,13 +187,34 @@ class Booking(models.Model):
         free_days = cls.get_ramaining_free_days(user, start_date, end_date)
         final_booking_price = normal_price - (free_days * price_per_day)
 
-        return normal_price, final_booking_price, free_days
+        # Calculating membership required months price for booking
+        required_membership_months = 0
+        membership_booking_price = 0.0
+        if BookingOrder.user_has_not_bookings(user):
+            today = datetime.today().date()
+            membership_price = MembershipType.objects.get(name=MembershipType.STANDARD).price
+            required_membership_months = cls.membership_required_booking_months(today, end_date)
+            membership_booking_price = membership_price * required_membership_months
+
+        # Add required membership months to final prices
+        final_booking_price += membership_booking_price
+
+        return normal_price, final_booking_price, free_days,\
+            required_membership_months, membership_booking_price
 
 
 class BookingOrder(Ordereable, models.Model):
     booking = models.OneToOneField(Booking)
     original_price = models.FloatField()
     special_month_price = models.FloatField()
+    membership_required_months = models.IntegerField(default=0)
+    membership_required_months_price = models.FloatField(default=0)
+
+
+    @classmethod
+    def user_has_not_bookings(cls, user):
+        return cls.objects.filter(customer__user=user).exists()
+
 
     def booking_days(self):
         return (self.booking.end_date - self.booking.start_date).days + 1
diff --git a/digitalglarus/templates/digitalglarus/booking_orders_detail.html b/digitalglarus/templates/digitalglarus/booking_orders_detail.html
index d4dac3b7..f3d7b049 100644
--- a/digitalglarus/templates/digitalglarus/booking_orders_detail.html
+++ b/digitalglarus/templates/digitalglarus/booking_orders_detail.html
@@ -44,6 +44,10 @@
             </h2>
             <h2 class="col-xs-6 payment-total text-left">Total days {{booking_days}}</h2>
             <h2 class="order-sum">{{original_price|floatformat}}CHF</h2>
+            {% if membership_required_months and membership_required_months_price %}
+              <h2 class="col-xs-6 payment-total text-left">Required Membership months: {{membership_required_months}}</h2>
+              <h2 class="order-sum"><span>{{membership_required_months_price|floatformat}}CHF</span></h2>
+            {% endif %}
             {% if free_days %}
               <h2 class="col-xs-6 payment-total text-left">Free days {{free_days}} </h2>
               <h2 class="order-sum">-{{total_discount|floatformat}}CHF</h2>
@@ -55,7 +59,7 @@
           </div>
     </div>              
      
-    <div class="col-xs-12 col-sm-4 col-lg-4 wow fadeInDown"> 
+    <div class="col-xs-12 col-sm-4 col-lg-4 wow fadeInDown hidden-print"> 
         <div class="order-summary">
            <div class="header text-center">
               <h2 class="order-name">Order Summary</h2>
@@ -65,6 +69,10 @@
               
               <h2 class="col-xs-6 payment-total">Total days {{booking_days}}</h2>
               <h2 class="order-sum">{{original_price|floatformat}}CHF</h2>
+              {% if membership_required_months and membership_required_months_price %}
+                <h2 class="col-xs-6 payment-total">Required Membership months: {{membership_required_months}}</h2>
+                <h2 class="order-sum"><span>{{membership_required_months_price|floatformat}}CHF</span></h2>
+              {% endif %}
               {% if free_days %}
                 <h2 class="col-xs-6 payment-total">Free days {{free_days}}</h2>
                 <h2 class="order-sum">-{{total_discount|floatformat}}CHF</h2>
diff --git a/digitalglarus/templates/digitalglarus/booking_payment.html b/digitalglarus/templates/digitalglarus/booking_payment.html
index 336ebf70..fda2a0e2 100644
--- a/digitalglarus/templates/digitalglarus/booking_payment.html
+++ b/digitalglarus/templates/digitalglarus/booking_payment.html
@@ -121,15 +121,20 @@
                   <br/>
 
                   <hr class="greyline">
-                  <h2 class="col-xs-6 payment-total">Total days {{booking_days}} </h2>
+                  <h2 class="col-xs-6 payment-total">Total days: {{booking_days}} </h2>
                   <h2 class="order-sum">{{original_price|floatformat}}CHF</h2>
+                  {% if membership_required_months and membership_required_months_price %}
+                    <h2 class="col-xs-6 payment-total">Required Membership months: {{membership_required_months}}</h2>
+                    <h2 class="order-sum"><span>{{membership_required_months_price|floatformat}}CHF</span></h2>
+                  {% endif %}
+                  <br/>
                   {% if free_days %}
-                    <h2 class="col-xs-6 payment-total">Free days {{free_days}}</h2>
+                    <h2 class="col-xs-6 payment-total">Free days: {{free_days}}</h2>
                     <h2 class="order-sum"><span class="text-danger">-{{total_discount|floatformat}}CHF</span></h2>
                   {% endif %}
                   <hr class="greyline">
                   <h2 class="col-xs-6 payment-total">Total</h2>
-                  <h2 class="order-result">{{discount_price|floatformat}}CHF</h2>
+                  <h2 class="order-result">{{final_price|floatformat}}CHF</h2>
                   <div class="text-center">
                   <label class="custom-control custom-checkbox">
                            <br/>
diff --git a/digitalglarus/views.py b/digitalglarus/views.py
index e87b61aa..0148620c 100644
--- a/digitalglarus/views.py
+++ b/digitalglarus/views.py
@@ -100,11 +100,22 @@ class BookingSelectDatesView(LoginRequiredMixin, MembershipRequiredMixin, FormVi
         start_date = form.cleaned_data.get('start_date')
         end_date = form.cleaned_data.get('end_date')
         booking_days = (end_date - start_date).days + 1
-        original_price, discount_price, free_days = Booking.\
+
+        price_per_day = BookingPrice.objects.get().price_per_day
+
+        original_price, final_price, free_days,\
+            membership_required_months, membership_required_months_price = Booking.\
             booking_price(user, start_date, end_date)
+
+        total_discount = price_per_day * free_days
+
         self.request.session.update({
             'original_price': original_price,
-            'discount_price': discount_price,
+            'final_price': final_price,
+            'total_discount': total_discount,
+            'membership_required_months_price': membership_required_months_price,
+            'membership_required_months': membership_required_months,
+            'booking_price_per_day': price_per_day,
             'booking_days': booking_days,
             'free_days': free_days,
             'start_date': start_date.strftime('%m/%d/%Y'),
@@ -118,8 +129,10 @@ class BookingPaymentView(LoginRequiredMixin, MembershipRequiredMixin, FormView):
     form_class = BookingBillingForm
     membership_redirect_url = reverse_lazy('digitalglarus:membership_pricing')
     # success_url = reverse_lazy('digitalglarus:booking_payment')
-    booking_needed_fields = ['original_price', 'discount_price', 'booking_days', 'free_days',
-                             'start_date', 'end_date']
+    booking_needed_fields = ['original_price', 'final_price', 'booking_days', 'free_days',
+                             'start_date', 'end_date', 'membership_required_months_price',
+                             'membership_required_months', 'booking_price_per_day',
+                             'total_discount']
 
     def dispatch(self, request, *args, **kwargs):
         from_booking = all(field in request.session.keys()
@@ -138,7 +151,7 @@ class BookingPaymentView(LoginRequiredMixin, MembershipRequiredMixin, FormView):
             'initial': {
                 'start_date': self.request.session.get('start_date'),
                 'end_date': self.request.session.get('end_date'),
-                'price': self.request.session.get('discount_price'),
+                'price': self.request.session.get('final_price'),
             }
         })
         return form_kwargs
@@ -148,11 +161,11 @@ class BookingPaymentView(LoginRequiredMixin, MembershipRequiredMixin, FormView):
 
         booking_data = {key: self.request.session.get(key)
                         for key in self.booking_needed_fields}
-        booking_price_per_day = BookingPrice.objects.get().price_per_day
-        total_discount = booking_price_per_day * booking_data.get('free_days')
+        # booking_price_per_day = BookingPrice.objects.get().price_per_day
+        # total_discount = booking_price_per_day * booking_data.get('free_days')
         booking_data.update({
-            'booking_price_per_day': booking_price_per_day,
-            'total_discount': total_discount,
+            # 'booking_price_per_day': booking_price_per_day,
+            # 'total_discount': total_discount,
             'stripe_key': settings.STRIPE_API_PUBLIC_KEY
         })
         context.update(booking_data)
@@ -165,7 +178,8 @@ class BookingPaymentView(LoginRequiredMixin, MembershipRequiredMixin, FormView):
         start_date = data.get('start_date')
         end_date = data.get('end_date')
 
-        normal_price, final_price, free_days = Booking.\
+        normal_price, final_price, free_days, membership_required_months,\
+            membership_required_months_price = Booking.\
             booking_price(self.request.user, start_date, end_date)
 
         # Get or create stripe customer
@@ -213,14 +227,12 @@ class BookingPaymentView(LoginRequiredMixin, MembershipRequiredMixin, FormView):
             'stripe_charge': charge,
             'amount': final_price,
             'original_price': normal_price,
-            'special_month_price': BookingPrice.objects.last().special_month_price
+            'special_month_price': BookingPrice.objects.last().special_month_price,
+            'membership_required_months': membership_required_months,
+            'membership_required_months_price': membership_required_months_price,
         }
         order = BookingOrder.create(order_data)
 
-        # request.session.update({
-        #     'membership_price': membership.type.first_month_price,
-        #     'membership_dates': membership.type.first_month_formated_range
-        # })
         return HttpResponseRedirect(self.get_success_url(order.id))
 
 
@@ -439,9 +451,15 @@ class OrdersBookingDetailView(LoginRequiredMixin, DetailView):
         booking_days = (end_date - start_date).days + 1
         original_price = booking.price
         final_price = booking.final_price
+
+        membership_required_months = bookig_order.membership_required_months
+        membership_required_months_price = bookig_order.membership_required_months_price
+
         context.update({
             'original_price': original_price,
             'total_discount': original_price - final_price,
+            'membership_required_months': membership_required_months,
+            'membership_required_months_price': membership_required_months_price,
             'final_price': final_price,
             'booking_days': booking_days,
             'free_days': free_days,