月范围计算不返回预期结果

时间:2019-02-12 06:11:59

标签: python python-3.x datetime python-dateutil relativedelta

如果我将日期设置为以下日期:

from datetime import datetime
from dateutil import relativedelta
class count_month_range:
    'Get the month range between 2 specified date based on the timezone'
    def __init__(self, start_date, end_date, timezone):
        self.start_date = start_date
        self.end_date = end_date
        self.timezone = timezone
    def count(self):
        start_date = datetime.strptime(str(self.start_date), '%Y-%m-%dT%H:%M:%S'+self.timezone)
        end_date = datetime.strptime(str(self.end_date), '%Y-%m-%dT%H:%M:%S'+self.timezone)
        r = relativedelta.relativedelta(end_date, start_date)
        print (r.months)
        return r.months
month_range = count_month_range("2018-12-01T00:00:00Z", "2019-12-01T00:00:00Z", "Z")
test = month_range.count()
print(test)

它将返回意外结果,如下所示:

0
0

我希望它返回12

我正在尝试获取月份范围。

例如:2018-12-01到2019-10-31将给我10个月的结果。

我有以下python test.py文件:

from datetime import datetime
from dateutil import relativedelta
class count_month_range:
    'Get the month range between 2 specified date based on the timezone'
    def __init__(self, start_date, end_date, timezone):
        self.start_date = start_date
        self.end_date = end_date
        self.timezone = timezone
    def count(self):
        start_date = datetime.strptime(str(self.start_date), '%Y-%m-%dT%H:%M:%S'+self.timezone)
        end_date = datetime.strptime(str(self.end_date), '%Y-%m-%dT%H:%M:%S'+self.timezone)
        r = relativedelta.relativedelta(end_date, start_date)
        print (r.months)
        return r.months
month_range = count_month_range("2018-12-01T00:00:00Z", "2019-10-31T00:00:00Z", "Z")
test = month_range.count()
print(test)

当我运行test.py文件时,它将返回以下内容

10
10

那是预期的结果。

如果2018年12月1日至2019年10月31日返回10个月是正确的,我希望在输入时有相同的正确计算: 2018年12月1日至2019年12月1日。

每当我在start_date和end_date中输入相同的月份(无论年份不同)时,都会发生这种问题。 我应该怎么做才能使其按预期工作? 非常感谢您的帮助。

谢谢。

1 个答案:

答案 0 :(得分:3)

使用提供的测试数据时,您会得到相对delta(years = + 1)。 因此,当您返回relativedelta对象时,您需要将年份转换为月份并将其与月份相加:

total_months = r.years * 12 + r.months

另一个例子。以下测试返回此结果: relativedelta(years = + 1,months = + 1)

count_month_range("2018-11-01T00:00:00Z", "2019-12-01T00:00:00Z", "Z")