end_date added

This commit is contained in:
Arvind Tiwari 2017-09-30 17:55:49 +05:30
commit bd91883369
3 changed files with 15 additions and 2 deletions

View file

@ -1,5 +1,7 @@
from datetime import datetime
import os
import logging
from dateutil.relativedelta import relativedelta
from django.db import models
from django.utils.functional import cached_property
@ -172,3 +174,9 @@ class VMDetail(models.Model):
ipv6 = models.TextField(default='')
created_at = models.DateTimeField(auto_now_add=True)
terminated_at = models.DateTimeField(null=True)
def end_date(self):
end_date = self.terminated_at if self.terminated_at else datetime.now()
months = relativedelta(end_date, self.created_at)
end_date = self.created_at + relativedelta(months=months)
return end_date