def to_quarterly_bucket()

in scripts/community-dataset/convert-mbox-to-solr-docs.py [0:0]

• 19 lines of code
• 8 McCabe index (conditional complexity)

def to_quarterly_bucket(date_obj):
    month = date_obj.month
    year = date_obj.year

    if month >= 2 and month <= 4:
        quarter = "Q4"
        fiscal_year = year
    elif month >= 5 and month <= 7:
        quarter = "Q1"
        fiscal_year = year + 1
    elif month >= 8 and month <= 10:
        quarter = "Q2"
        fiscal_year = year + 1
    else: # month = 11, 12, 1
        quarter = "Q3"
        if month == 1:
            fiscal_year = year
        else:
            fiscal_year = year + 1
    return str(year) + "-" + quarter