polars_plan/dsl/functions/
business.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
use super::*;

#[cfg(feature = "dtype-date")]
pub fn business_day_count(
    start: Expr,
    end: Expr,
    week_mask: [bool; 7],
    holidays: Vec<i32>,
) -> Expr {
    let input = vec![start, end];

    Expr::Function {
        input,
        function: FunctionExpr::Business(BusinessFunction::BusinessDayCount {
            week_mask,
            holidays,
        }),
        options: FunctionOptions {
            flags: FunctionFlags::default() | FunctionFlags::ALLOW_RENAME,
            ..Default::default()
        },
    }
}