From 21941174774c9a4397db27a0443e25df9aa44738 Mon Sep 17 00:00:00 2001 From: Ben Sima Date: Wed, 28 Nov 2018 14:18:28 -0800 Subject: Handle leap years --- progress.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'progress.py') diff --git a/progress.py b/progress.py index 411f605..bb5c571 100755 --- a/progress.py +++ b/progress.py @@ -9,12 +9,14 @@ class Progress(object): def __init__(self): self.now = datetime.now() weekdays_left = timedelta(7 - self.now.isoweekday()) - month_max = max(list(Calendar().itermonthdays(self.now.year, self.now.month))) + cal = Calendar() + month_max = max(list(cal.itermonthdays(self.now.year, self.now.month))) + year_max = 366 if isleap(self.now.year) else 365 self.dura = { 'day': timedelta(hours=8), 'week': timedelta(days=7), 'month': timedelta(days=month_max), - 'year': timedelta(days=365), + 'year': timedelta(days=year_max), } self.end = { 'year': datetime(self.now.year+1, 1, 1), -- cgit v1.2.3