def test_add_months(self): """ Test two cases of add_month method """ date = datetime.datetime(2015, 10, 30, 20, 0, 0) self.assertEquals(CommonService.add_months(date, 1), datetime.datetime(2015, 11, 30, 20, 0, 0)) self.assertEquals(CommonService.add_months(date, 4), datetime.datetime(2016, 02, 29, 20, 0, 0))
def test_add_months(self):
"""
Test two cases of add_month method
"""
date = datetime.datetime(2015, 10, 30, 20, 0, 0)
self.assertEquals(CommonService.add_months(date, 1),
datetime.datetime(2015, 11, 30, 20, 0, 0))
self.assertEquals(CommonService.add_months(date, 4),
datetime.datetime(2016, 02, 29, 20, 0, 0))
def extend_subscription(user_profile, duration):
"""
Extend profile subscription
:param user: User instance
:param duration: {int} duration in month
:return:
"""
user_profile = UserProfile.objects.get(pk=user_profile.pk)
if user_profile.date_subscribe_ended and user_profile.date_subscribe_ended.replace(tzinfo=pytz.utc).replace(tzinfo=None) > datetime.datetime.now(tz=pytz.timezone(pytz.country_timezones['RU'][1])).replace(tzinfo=None):
new_date = CommonService.add_months(user_profile.date_subscribe_ended, duration)
else:
new_date = CommonService.add_months(datetime.datetime.now(), duration)
user_profile.date_subscribe_ended = new_date
user_profile.save()
def confirm(self, **kwargs): """ Execute confirm transaction :return: transact """ self.__transact.status_id = 2 self.__transact.expiration_date = CommonService.add_months(datetime.datetime.now(), self.__transact.duration) return self.__save(self.__extend_subscription)
def confirm(self, **kwargs):
"""
Execute confirm transaction
:return: transact
"""
self.__transact.status_id = 2
self.__transact.expiration_date = CommonService.add_months(
datetime.datetime.now(), self.__transact.duration)
return self.__save(self.__extend_subscription)