获取两个日期之间的日期范围,不包括周末

日期范围 = B - A

不包括周末的日期范围 = 日期范围 - (周末数量 * 2)

import datetime

def get_date_range(start_date, end_date):
    date_range = (end_date - start_date).days + 1
    weekends = 0
    for n in range(date_range):
        day = start_date + datetime.timedelta(days=n)
        if day.weekday() >= 5:
            weekends += 1
    return date_range - weekends

start_date = datetime.date(2022, 1, 1)
end_date = datetime.date(2022, 1, 31)
date_range = get_date_range(start_date, end_date)
print(date_range)

import json
import datetime

def main_handler(event, context):
    start_date = event['start_date']
    end_date = event['end_date']
    date_range = get_date_range(start_date, end_date)
    return {
        'statusCode': 200,
        'body': json.dumps({
            'date_range': date_range
        })
    }

def get_date_range(start_date, end_date):
    date_range = (end_date - start_date).days + 1
    weekends = 0
    for n in range(date_range):
        day = start_date + datetime.timedelta(days=n)
        if day.weekday() >= 5:
            weekends += 1
    return date_range - weekends