pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use UTC tz for determining TO status

Browse Source
This commit is contained in:
richard-dds 2019-06-10 14:54:05 -04:00
parent 6eb64d4aae
commit a2d201e43b
2 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
atst/models/task_order.py
View File

@ -8,6 +8,7 @@ from werkzeug.datastructures import FileStorage
from atst.models import Attachment, Base, mixins, types from atst.models import Attachment, Base, mixins, types
from atst.models.clin import JEDICLINType from atst.models.clin import JEDICLINType
from atst.utils.clock import Clock
class Status(Enum): class Status(Enum):
@ -81,7 +82,7 @@ class TaskOrder(Base, mixins.TimestampsMixin):
@property @property
def status(self): def status(self):
today = date.today() today = Clock.today()
if not self.is_completed and not self.is_signed: if not self.is_completed and not self.is_signed:
return Status.DRAFT return Status.DRAFT
@ -105,7 +106,7 @@ class TaskOrder(Base, mixins.TimestampsMixin):
@property @property
def days_to_expiration(self): def days_to_expiration(self):
if self.end_date: if self.end_date:
return (self.end_date - date.today()).days return (self.end_date - Clock.today()).days
@property @property
def total_obligated_funds(self): def total_obligated_funds(self):

10
atst/utils/clock.py Normal file
View File

@ -0,0 +1,10 @@
import pendulum
class Clock(object):
@classmethod
def today(cls, tz="UTC"):
return pendulum.today(tz=tz).date()
@classmethod
def now(cls, tz="UTC"):
return pendulum.now(tz=tz)