businesstimedelta_working_hours_calculation_with_holidays.py

1import datetime
2import pytz
3import businesstimedelta
4import holidays as pyholidays
5
6# Define a working day (e.g., 9 AM to 5 PM)
7workday = businesstimedelta.WorkDayRule(
8    start_time=datetime.time(9),
9    end_time=datetime.time(17),
10    working_days=[0, 1, 2, 3, 4] # Monday to Friday
11)
12
13# Take US holidays into account
14vic_holidays = pyholidays.US()
15holidays = businesstimedelta.HolidayRule(vic_holidays)
16
17# Combine rules into a business time configuration
18business_hours = businesstimedelta.Rules([workday, holidays])
19
20# Define two datetimes
21tz = pytz.timezone('US/Eastern')
22dt1 = tz.localize(datetime.datetime(2016, 1, 1, 8)) # New Year's Day (Friday)
23dt2 = tz.localize(datetime.datetime(2016, 1, 4, 10)) # Following Monday
24
25# Calculate the business time difference
26bdiff = business_hours.difference(dt1, dt2)
27
28print(f"Business time: {bdiff.hours} hours, {bdiff.seconds} seconds")
29# Output: Business time: 1 hours, 0 seconds