From 1858777c698214b3223aeb2c51d310ceef9b8d9a Mon Sep 17 00:00:00 2001 From: Guillem Borrell Date: Tue, 17 Jan 2023 12:36:09 +0000 Subject: [PATCH] Handle empty columns in schema summary too --- src/schema.rs | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/src/schema.rs b/src/schema.rs index 7e9e3e1..65bd905 100644 --- a/src/schema.rs +++ b/src/schema.rs @@ -5,8 +5,15 @@ use sea_query::*; pub fn print_schema(ldf: LazyFrame) { let schema = ldf.schema().expect("Could not retreive schema"); for f in schema.iter_fields() { - let n = f.name(); + let mut unnamed_cols_counter = 0; let d = f.data_type().to_string(); + let n = if f.name.is_empty() { + unnamed_cols_counter += 1; + format!("Column{}", unnamed_cols_counter) + } else { + f.name + }; + println!("{n} ({d})"); } }