Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Calculate the end of the previous quarter

Tags:

python

I'm looking for an elegant and pythonic way to get the date of the end of the previous quarter.

Something like this:

def previous_quarter(reference_date):
   ...


>>> previous_quarter(datetime.date(2013, 5, 31))
datetime.date(2013, 3, 31)

>>> previous_quarter(datetime.date(2013, 2, 1))
datetime.date(2012, 12, 31)

>>> previous_quarter(datetime.date(2013, 3, 31))
datetime.date(2012, 12, 31)

>>> previous_quarter(datetime.date(2013, 11, 1))
datetime.date(2013, 9, 30)

Edit: Have I tried anything?

Yes, this seems to work:

def previous_quarter(ref_date):
    current_date = ref_date - timedelta(days=1)
    while current_date.month % 3:
        current_date -= timedelta(days=1)
    return current_date

But it seems unnecessarily iterative.

like image 709
dgel Avatar asked May 31 '13 19:05

dgel


2 Answers

You can do it the "hard way" by just looking at the month you receive:

def previous_quarter(ref):
    if ref.month < 4:
        return datetime.date(ref.year - 1, 12, 31)
    elif ref.month < 7:
        return datetime.date(ref.year, 3, 31)
    elif ref.month < 10:
        return datetime.date(ref.year, 6, 30)
    return datetime.date(ref.year, 9, 30)
like image 153
Justin Ethier Avatar answered Sep 19 '22 14:09

Justin Ethier


Using dateutil:

import datetime as DT
import dateutil.rrule as rrule

def previous_quarter(date):
    date = DT.datetime(date.year, date.month, date.day)
    rr = rrule.rrule(
        rrule.DAILY,
        bymonth=(3,6,9,12),    # the month must be one of these
        bymonthday=-1,         # the day has to be the last of the month
        dtstart = date-DT.timedelta(days=100))
    result = rr.before(date, inc=False)  # inc=False ensures result < date
    return result.date()

print(previous_quarter(DT.date(2013, 5, 31)))
# 2013-03-31
print(previous_quarter(DT.date(2013, 2, 1)))
# 2012-12-31
print(previous_quarter(DT.date(2013, 3, 31)))
# 2012-12-31
print(previous_quarter(DT.date(2013, 11, 1)))
# 2013-09-30
like image 35
unutbu Avatar answered Sep 16 '22 14:09

unutbu