From f1790445cd5b9ea7f179c06730fa868e382eb7a0 Mon Sep 17 00:00:00 2001 From: Arvind Tiwari Date: Sat, 30 Sep 2017 18:05:02 +0530 Subject: [PATCH] timezone aware --- hosting/models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/hosting/models.py b/hosting/models.py index d901e962..db68f522 100644 --- a/hosting/models.py +++ b/hosting/models.py @@ -1,9 +1,9 @@ -from datetime import datetime import os import logging from dateutil.relativedelta import relativedelta from django.db import models +from django.utils import timezone from django.utils.functional import cached_property from Crypto.PublicKey import RSA from membership.models import StripeCustomer, CustomUser @@ -176,7 +176,7 @@ class VMDetail(models.Model): terminated_at = models.DateTimeField(null=True) def end_date(self): - end_date = self.terminated_at if self.terminated_at else datetime.now() + end_date = self.terminated_at if self.terminated_at else timezone.now() months = relativedelta(end_date, self.created_at) end_date = self.created_at + relativedelta(months=months) return end_date