mirror of
https://github.com/rzmk/czv.git
synced 2025-12-19 08:09:24 +00:00
feat: add czv, czv-wasm, and czv-python (init release)
This commit is contained in:
commit
9799ab694b
40 changed files with 70383 additions and 0 deletions
33
czv-python/src/count.rs
Normal file
33
czv-python/src/count.rs
Normal file
|
|
@ -0,0 +1,33 @@
|
|||
use crate::Result;
|
||||
use csv::ReaderBuilder;
|
||||
use pyo3::pyfunction;
|
||||
|
||||
/// Returns a count of the total number of rows.
|
||||
///
|
||||
/// ## Arguments
|
||||
///
|
||||
/// * `file_data` - CSV file data.
|
||||
/// * `include_header_row` - Specify whether to include the header row (first row) in the row count.
|
||||
#[pyfunction]
|
||||
pub fn row_count(file_data: String, include_header_row: Option<bool>) -> Result<usize> {
|
||||
let mut rdr = ReaderBuilder::new();
|
||||
|
||||
rdr.has_headers(!include_header_row.unwrap_or(false));
|
||||
return Ok(rdr.from_reader(file_data.as_bytes()).records().count());
|
||||
}
|
||||
|
||||
/// Returns a count of the total number of columns (fields).
|
||||
///
|
||||
/// ## Arguments
|
||||
///
|
||||
/// * `file_data` - CSV file data.
|
||||
#[pyfunction]
|
||||
pub fn column_count(file_data: Option<String>) -> Result<usize> {
|
||||
let rdr = ReaderBuilder::new();
|
||||
|
||||
if let Some(file_data) = file_data {
|
||||
return Ok(rdr.from_reader(file_data.as_bytes()).headers()?.len());
|
||||
} else {
|
||||
bail!("Could not determine a file path or file data for column_count_builder.");
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue