1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
//! # Polars rows iterator
//!
//! Simple and convenient iteration of polars dataframe rows.
//!
//! This crate provides two main approaches for iterating over DataFrame rows:
//! - **Struct-based iteration** using `#[derive(FromDataFrameRow)]` - best for complex rows with many columns
//! - **Tuple-based iteration** using the [`df_rows_iter!`] macro - best for quick, simple iterations
//!
//! ## Tuple-based iteration with `df_rows_iter!`
//!
//! For simple use cases where you don't need a dedicated struct, use the [`df_rows_iter!`] macro
//! to iterate over rows as tuples:
//!
//! ```rust
//! use polars::prelude::*;
//! use polars_rows_iter::*;
//!
//! let df = df!(
//! "name" => ["Alice", "Bob", "Charlie"],
//! "age" => [25i32, 30, 35],
//! "score" => [Some(95.5f64), None, Some(87.0)]
//! ).unwrap();
//!
//! let score_col = format!("sco{}", "re"); // dynamic column name
//!
//! let iter = df_rows_iter!(
//! &df,
//! "name" => &str, // string literal
//! "age" => i32,
//! score_col => Option<f64> // variable
//! ).unwrap();
//!
//! for row in iter {
//! let (name, age, score) = row.unwrap();
//! println!("{name}: age {age}, score {score:?}");
//! }
//! ```
//!
//! The macro supports tuples of up to 10 elements. Each element is specified as `column_name => Type`.
//! Column names can be string literals or any expression that implements `AsRef<str>`.
//!
//! ## Struct-based iteration with `FromDataFrameRow`
//!
//! For more complex use cases, derive `FromDataFrameRow` on a struct:
//! ```rust
//!use polars::prelude::*;
//!use polars_rows_iter::*;
//!
//!fn main() {
//! #[derive(Debug, FromDataFrameRow)]
//! #[derive(PartialEq)] // for assert_eq
//! struct MyRow<'a>
//! {
//! #[column("col_a")]
//! a: i32,
//! // the column name defaults to the field name if no explicit name given
//! col_b: &'a str,
//! col_c: String,
//! #[column("col_d")]
//! optional: Option<f64>
//! }
//!
//! let df = df!(
//! "col_a" => [1i32, 2, 3, 4, 5],
//! "col_b" => ["a", "b", "c", "d", "e"],
//! "col_c" => ["A", "B", "C", "D", "E"],
//! "col_d" => [Some(1.0f64), None, None, Some(2.0), Some(3.0)]
//! ).unwrap();
//!
//! let rows_iter = df.rows_iter::<MyRow>().unwrap(); // ready to use row iterator
//! // collect to vector for assert_eq
//! let rows_vec = rows_iter.collect::<PolarsResult<Vec<MyRow>>>().unwrap();
//!
//! assert_eq!(
//! rows_vec,
//! [
//! MyRow { a: 1, col_b: "a", col_c: "A".to_string(), optional: Some(1.0) },
//! MyRow { a: 2, col_b: "b", col_c: "B".to_string(), optional: None },
//! MyRow { a: 3, col_b: "c", col_c: "C".to_string(), optional: None },
//! MyRow { a: 4, col_b: "d", col_c: "D".to_string(), optional: Some(2.0) },
//! MyRow { a: 5, col_b: "e", col_c: "E".to_string(), optional: Some(3.0) },
//! ]
//! );
//!}
//! ```
//! Every row is wrapped with a PolarsError, in case of an unexpected null value the row creation fails and the iterator
//! returns an Err(...) for the row. One can decide to cancel the iteration or to skip the affected row.
//!
//! ## Column Name Transformations
//!
//! The `#[from_dataframe(...)]` attribute allows automatic transformation of field names to column names:
//!
//! ```rust
//! use polars::prelude::*;
//! use polars_rows_iter::*;
//!
//! #[derive(Debug, FromDataFrameRow)]
//! #[from_dataframe(convert_case(Pascal), prefix("col_"))]
//! struct MyRow {
//! user_name: String, // maps to column "col_UserName"
//! age: i32, // maps to column "col_Age"
//! }
//!
//! let df = df!(
//! "col_UserName" => ["Alice", "Bob"],
//! "col_Age" => [25i32, 30]
//! ).unwrap();
//!
//! let rows: Vec<MyRow> = df.rows_iter::<MyRow>()
//! .unwrap()
//! .collect::<PolarsResult<Vec<_>>>()
//! .unwrap();
//!
//! assert_eq!(rows[0].user_name, "Alice");
//! assert_eq!(rows[0].age, 25);
//! ```
//!
//! ### Available options:
//!
//! - `convert_case(Case)` - Convert field names using a case style. Supported cases:
//! `Upper`, `Lower`, `Title`, `Toggle`, `Camel`, `Pascal`, `UpperCamel`, `Snake`,
//! `UpperSnake`, `ScreamingSnake`, `Kebab`, `Cobol`, `UpperKebab`, `Train`, `Flat`,
//! `UpperFlat`, `Alternating`
//! - `prefix("str")` - Add a prefix to all column names
//! - `postfix("str")` - Add a postfix/suffix to all column names
//!
//! These can be combined: `#[from_dataframe(convert_case(Snake), prefix("data_"), postfix("_col"))]`
//!
//! Individual fields can still override with `#[column("explicit_name")]`.
//!
//! ## Supported types
//!
//! |State|Rust Type|Supported Polars DataType|Feature Flag|
//! |--|--|--|--|
//! |✓|`bool`|`Boolean`
//! |✓|`u8`|`UInt8`
//! |✓|`u16`|`UInt16`
//! |✓|`u32`|`UInt32`
//! |✓|`u64`|`UInt64`
//! |✓|`i8`|`Int8`
//! |✓|`i16`|`Int16`
//! |✓|`i32`|`Int32`
//! |✓|`i32`|`Date`
//! |✓|`i64`|`Int64`
//! |✓|`i64`|`Datetime(..)`
//! |✓|`i64`|`Duration(..)`
//! |✓|`i64`|`Time`
//! |✓|`f32`|`Float32`
//! |✓|`f64`|`Float64`
//! |✓|`&str`|`String`
//! |✓|`&str`|`Categorical(..)`|`dtype-categorical`
//! |✓|`&str`|`Enum(..)`|`dtype-categorical`
//! |✓|`String`|`String`
//! |✓|`String`|`Categorical(..)`|`dtype-categorical`
//! |✓|`String`|`Enum(..)`|`dtype-categorical`
//! |✓|`&[u8]`|`Binary`
//! |✓|`&[u8]`|`BinaryOffset`
//! |✓|`chrono::NaiveDateTime`|`Datetime(..)`|`chrono`
//! |✓|`chrono::DateTime<Utc>`|`Datetime(..)`|`chrono`
//! |✓|`chrono::Date`|`Date`|`chrono`|
//! |✓|`polars::prelude::Series`|`List(..)`
//! |✓|`Vec<T>`|`List(..)`
//! |X|`Vec<&str>`|`List(..)`
//! |X|`Vec<&[u8]>`|`List(..)`
//! |?|?|`Array(..)`|
//! |?|?|`Decimal(..)`|
//! |?|?|`Struct(..)`|
//! |X|X|`Null`
//! |X|X|`Unknown(..)`|
//! |X|X|`Object(..)`|
//!
//! TODO: Support is planned <br>
//! ?: Support not yet certain<br>
//! X: No Support
extern crate self as polars_rows_iter;
pub use convert_case;
pub use *;
pub use *;
pub use *;
use impl_tuple_rows_iter;
pub use FromDataFrameRow;
impl_tuple_rows_iter!;