From bd6aea011a223dce7f4f140a1e5ead63da621bb5 Mon Sep 17 00:00:00 2001
From: Ivlev Denis <di-erz@yandex.ru>
Date: Fri, 11 May 2018 13:05:39 +0300
Subject: [PATCH] Update school buy logic

---
 apps/payment/views.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/apps/payment/views.py b/apps/payment/views.py
index d719beb9..7d9aeded 100644
--- a/apps/payment/views.py
+++ b/apps/payment/views.py
@@ -1,5 +1,6 @@
-import logging
+import arrow
 import json
+import logging
 
 from datetime import timedelta
 from urllib.parse import urlsplit
@@ -203,11 +204,11 @@ class PaymentwallCallbackView(View):
                             date_start = now()
                             date_end = school_payment.date_end
                         else:
-                            date_start = school_payment.date_end + timedelta(days=1)
-                            date_end = date_start + timedelta(days=30)
+                            date_start = arrow.get(school_payment.date_end).shift(days=1).datetime
+                            date_end = arrow.get(date_start).shift(months=1).datetime
                     else:
                         date_start = now()
-                        date_end = now() + timedelta(days=30)
+                        date_end = arrow.get(date_start).shift(months=1).datetime
                     payment.date_start = date_start
                     payment.date_end = date_end
             if product_type_name == 'course':