patchwork_csv_utils 0.1.14-x86_64-linux → 0.1.16-x86_64-linux

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 30d63239e5166c8534e462461e4973a5dadd39ab450226b7dbd6c3c7cddaa7d0
-  data.tar.gz: cab1593e1171eea64b7c529665be9114547e0264df60eba459c0de524f79fa91
+  metadata.gz: ccd0a5a1fac5fc48723b28f4b0922ee10468b2bcf26bb6da77a7abdcb9c3882f
+  data.tar.gz: b9be825c7c520ef319c940814863f71e71c41d0ed2be062179e665110dfcb0d6
 SHA512:
-  metadata.gz: 4a7fbc1d06a848cf787460005b0824f9bb7a0b8d66616b8568467fcf9c118d102571100edf5b66c663854afdf97a0deedd0ff174d1c03ab72d8a71146f09fb76
-  data.tar.gz: 2b2a9dd72e61a2fec34e5e52fe75ca1da66e3a4b481e902b4581e824eda187271c08d85ac5859a5d3ef6dc61a7949faf632801f6055cf47bd5bc1adfebbe61df
+  metadata.gz: f2419b94893570ef960dd9be99ed2d530b3b8d201b05a8373dd8a5a2b679e6c9f078e26ebc6f8a43f649158e2984ed9c10dad5d8c7ebac80ffe6b927ca2250fc
+  data.tar.gz: bb87188b17f9e95e3cdf7db797cf9ae067bb322151069f70b6caaa7c8749ec9c2ecdb068136c55c0a010a0f67b5de9990192b4ca94fd6ec5b8e0d73da71a2ed9

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    patchwork_csv_utils (0.1.14)
+    patchwork_csv_utils (0.1.16)
 
 GEM
   remote: https://rubygems.org/

data/README.md

@@ -14,8 +14,8 @@ gem install patchwork_csv_utils
 ```irb
 require 'csv_utils'
 CsvUtils.dedup('file1.csv', 'file2.csv', 'output.csv', ['mandatory_headers'])
-CsvUtils.to_csv('file1.xls', 'output_file1.csv', ['request_ids_to_skip'], ['mandatory_headers'])
-CsvUtils.transform_csv('file1.xls', 'output_file1.csv', ['request_ids_to_skip'], ['mandatory_headers'])
+CsvUtils.to_csv('file1.xls', 'output_file1.csv', ['request_ids_to_skip'], ['mandatory_headers'], ['status_exclusions'])
+CsvUtils.transform_csv('file1.xls', 'output_file1.csv', ['request_ids_to_skip'], ['mandatory_headers'], ['status_exclusions'])
 ```
 
 ## Release

data/ext/csv_utils/src/lib.rs

@@ -9,7 +9,7 @@ pub mod utils;
 fn init() -> Result<(), magnus::Error> {
     let module = define_module("CsvUtils")?;
     module.define_singleton_method("dedup", function!(dedup, 4))?;
-    module.define_singleton_method("to_csv", function!(to_csv, 4))?;
-    module.define_singleton_method("transform_csv", function!(transform_csv, 4))?;
+    module.define_singleton_method("to_csv", function!(to_csv, 6))?;
+    module.define_singleton_method("transform_csv", function!(transform_csv, 6))?;
     Ok(())
 }

data/ext/csv_utils/src/utils/csv.rs

@@ -1,20 +1,22 @@
 use std::collections::HashMap;
 use std::fs::File;
-
 use chrono::{NaiveDate, NaiveDateTime, NaiveTime, Utc};
 use csv::{Reader, StringRecord, Writer};
 use magnus::{Error, RArray, Ruby};
 
-use crate::utils::{FileExtension, magnus_err, missing_header, to_datetime_error, check_mandatory_headers, create_header_map, missing_value, headers_as_byte_record, index_of_header_in_mandatory_list};
+use crate::utils::{FileExtension, magnus_err, missing_header, to_datetime_error, check_mandatory_headers, create_header_map, missing_value, headers_as_byte_record, index_of_header_in_mandatory_list, validate_trust_name};
 
 pub fn transform_csv(ruby: &Ruby, csv_path: String,
                      target_path: String, exclusions: RArray,
-                     mandatory_headers: RArray, ) -> magnus::error::Result<()> {
+                     mandatory_headers: RArray,
+                     status_exclusions: RArray,
+                     expected_trust_name: String,) -> magnus::error::Result<()> {
     if !csv_path.has_extension(&["csv"]) {
         return Err(Error::new(ruby.exception_standard_error(), "csv_path must be a csv file".to_string()));
     }
 
     let exclusions = RArray::to_vec(exclusions)?;
+    let status_exclusions = RArray::to_vec(status_exclusions)?;
     let mandatory_headers: Vec<String> = RArray::to_vec(mandatory_headers)?;
 
     let csv_file = File::open(csv_path).map_err(|e| magnus_err(ruby, e, "csv_path"))?;
@@ -38,12 +40,15 @@ pub fn transform_csv(ruby: &Ruby, csv_path: String,
     let end = header_map.get("End").ok_or(missing_header(ruby, "End"))?;
     let actual_start = header_map.get("Actual Start").ok_or(missing_header(ruby, "Actual Start"))?;
     let actual_end = header_map.get("Actual End").ok_or(missing_header(ruby, "Actual End"))?;
+    let status = header_map.get("Status");
+    let trust_name = header_map.get("Trust").ok_or(missing_header(ruby, "Trust"))?;
 
     let mandatory_records = get_mandatory_records(&ruby, &mut csv, &headers_list, &mandatory_headers)?;
 
     for (ri, record) in mandatory_records.iter().enumerate() {
 
         if skip_excluded_rows(request_id, &record, &exclusions) { continue; }
+        if skip_excluded_status_rows(&status, &record, &status_exclusions) { continue; }
         if has_empty_row_skip(&record) { continue; }
         if has_empty_first_col_skip_row(&record) { continue; }
 
@@ -55,6 +60,8 @@ pub fn transform_csv(ruby: &Ruby, csv_path: String,
             let column_value = record.get(*column_index).ok_or(missing_value(ruby, column))?;
             let column_value = column_value.trim_end();
 
+            validate_trust_name(ruby, &expected_trust_name, trust_name, ri, i, &column_value.to_string())?;
+
             if i == *date {
                 let current = string_to_datetime(column_value).ok_or(to_datetime_error(ruby, column_value, ri, "Date"))?;
                 date_value = current;
@@ -124,6 +131,12 @@ fn skip_excluded_rows(request_id: &usize, r: &StringRecord, exclusions: &Vec<Str
     exclusions.contains(&value.to_string())
 }
 
+fn skip_excluded_status_rows(status: &Option<&usize>, r: &StringRecord, exclusions: &Vec<String>) -> bool {
+    status
+        .map(|index| exclusions.contains(&r[*index].to_string()))
+        .unwrap_or(false)
+}
+
 fn string_to_datetime(s: &str) -> Option<NaiveDateTime> {
     let maybe_correct = correct_datetime(s);
     if maybe_correct.is_some() { return maybe_correct; }

data/ext/csv_utils/src/utils/mod.rs

@@ -9,6 +9,15 @@ pub mod csv;
 pub mod dedup;
 pub mod xls;
 
+fn validate_trust_name(ruby: &Ruby, expected_trust_name: &String, trust_name: &usize, ri: usize, i: usize, s: &String) -> magnus::error::Result<()> {
+    if ri > 0 && i == *trust_name {
+        if s != &expected_trust_name.clone() {
+            return Err(magnus::Error::new(ruby.exception_standard_error(), format!("Trust actual name: '{}' is not as expected: '{}'", s, expected_trust_name)));
+        }
+    }
+    Ok(())
+}
+
 fn missing_header(ruby: &Ruby, header: &str) -> magnus::Error {
     magnus::Error::new(ruby.exception_standard_error(), format!("Missing '{}' header", header))
 }

data/ext/csv_utils/src/utils/xls.rs

@@ -6,12 +6,14 @@ use calamine::{Data, open_workbook, Range, Reader, Xls};
 use chrono::{NaiveDateTime, Utc};
 use magnus::{RArray, Ruby};
 
-use crate::utils::{FileExtension, magnus_err, missing_header, to_datetime_error, check_mandatory_headers, missing_value, index_of_header_in_mandatory_list};
+use crate::utils::{FileExtension, magnus_err, missing_header, to_datetime_error, check_mandatory_headers, missing_value, index_of_header_in_mandatory_list, validate_trust_name};
 
 pub fn to_csv(ruby: &Ruby, xls_path: String,
               target_path: String,
               exclusions: RArray,
-              mandatory_headers: RArray
+              mandatory_headers: RArray,
+              status_exclusions: RArray,
+              expected_trust_name: String,
 ) -> magnus::error::Result<()> {
     if !xls_path.has_extension(&["xls"]) {
         return Err(magnus::Error::new(ruby.exception_standard_error(), "xls_path must be an xls file".to_string()));
@@ -19,6 +21,7 @@ pub fn to_csv(ruby: &Ruby, xls_path: String,
 
     let exclusions = RArray::to_vec(exclusions)?;
     let mandatory_headers: Vec<String> = RArray::to_vec(mandatory_headers)?;
+    let status_exclusions = RArray::to_vec(status_exclusions)?;
 
     let mut workbook: Xls<_> = open_workbook(xls_path.clone()).map_err(|e| magnus_err(ruby, e, format!("could not open xls: {}", xls_path).as_str()))?;
     let range = workbook.worksheet_range_at(0)
@@ -26,7 +29,7 @@ pub fn to_csv(ruby: &Ruby, xls_path: String,
         .and_then(|r| r.map_err(|e| magnus_err(ruby, e, "could not read worksheet range")))?;
 
     let headers = range.headers().ok_or(magnus::Error::new(ruby.exception_standard_error(), "no headers found in xls".to_string()))?;
-    let headers_list : Vec<String> = headers.iter().map(|h| h.to_string()).collect();
+    let headers_list: Vec<String> = headers.iter().map(|h| h.to_string()).collect();
 
     if let Some(value) =
         check_mandatory_headers(ruby, &headers_list, &mandatory_headers, "csv") { return value; }
@@ -35,13 +38,15 @@ pub fn to_csv(ruby: &Ruby, xls_path: String,
     let csv_out_file = File::create(target_path.clone()).map_err(|e| magnus_err(ruby, e, format!("could not create csv file: {}", target_path).as_str()))?;
     let mut dest = BufWriter::new(csv_out_file);
 
-    write_csv(ruby, &mut dest, &range, header_map, exclusions, mandatory_headers, headers_list)
+    write_csv(ruby, &mut dest, &range, header_map, exclusions, mandatory_headers, headers_list, status_exclusions, expected_trust_name)
 }
 
 fn write_csv<W: Write>(ruby: &Ruby, dest: &mut W, range: &Range<Data>,
                        header_map: HashMap<String, usize>, exclusions: Vec<String>,
                        mandatory_headers: Vec<String>,
-                       headers_list: Vec<String>) -> magnus::error::Result<()> {
+                       headers_list: Vec<String>,
+                       status_exclusions: Vec<String>,
+                       expected_trust_name: String) -> magnus::error::Result<()> {
     let n = mandatory_headers.len() - 1;
     let request_id = header_map.get("Request Id").ok_or(missing_header(ruby, "Request Id"))?;
     let date = header_map.get("Date").ok_or(missing_header(ruby, "Date"))?;
@@ -49,6 +54,8 @@ fn write_csv<W: Write>(ruby: &Ruby, dest: &mut W, range: &Range<Data>,
     let end = header_map.get("End").ok_or(missing_header(ruby, "End"))?;
     let actual_start = header_map.get("Actual Start").ok_or(missing_header(ruby, "Actual Start"))?;
     let actual_end = header_map.get("Actual End").ok_or(missing_header(ruby, "Actual End"))?;
+    let status = header_map.get("Status");
+    let trust_name = header_map.get("Trust").ok_or(missing_header(ruby, "Trust"))?;
 
     let mandatory_rows = get_mandatory_records(ruby, range, &headers_list, &mandatory_headers)?;
 
@@ -56,6 +63,7 @@ fn write_csv<W: Write>(ruby: &Ruby, dest: &mut W, range: &Range<Data>,
         let mut date_value = Utc::now().naive_utc();
 
         if skip_excluded_rows(&request_id, &r, &exclusions) { continue; }
+        if skip_excluded_status_rows(&status, &r, &status_exclusions) { continue; }
         if skip_empty_rows(&r) { continue; }
         if skip_rows_with_no_request_id(&request_id, &r) { continue; }
         if date_value_is_not_present(&date, &r) {
@@ -63,13 +71,13 @@ fn write_csv<W: Write>(ruby: &Ruby, dest: &mut W, range: &Range<Data>,
         }
 
         for (i, c) in mandatory_headers.iter().enumerate() {
-
             let column_index = header_map.get(c).ok_or(missing_header(ruby, c))?;
             let c = r.get(*column_index).ok_or(missing_value(ruby, c))?;
 
             match *c {
                 Data::Empty => Ok(()),
                 Data::String(ref s) | Data::DateTimeIso(ref s) | Data::DurationIso(ref s) => {
+                    validate_trust_name(ruby, &expected_trust_name, trust_name, ri, i, s)?;
                     handle_commas(dest, s)
                 }
                 Data::Float(ref f) => write!(dest, "{}", f),
@@ -125,6 +133,12 @@ fn skip_excluded_rows(request_id: &usize, r: &Vec<&Data>, exclusions: &Vec<Strin
     exclusions.contains(&value.to_string())
 }
 
+fn skip_excluded_status_rows(status: &Option<&usize>, r: &Vec<&Data>, exclusions: &Vec<String>) -> bool {
+    status
+        .map(|index| exclusions.contains(&r[*index].to_string()))
+        .unwrap_or(false)
+}
+
 fn skip_empty_rows(r: &Vec<&Data>) -> bool {
     r.into_iter().all(|c| c == &&Data::Empty)
 }

data/lib/csv_utils/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module CsvUtils
-  VERSION = '0.1.14'
+  VERSION = '0.1.16'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: patchwork_csv_utils
 version: !ruby/object:Gem::Version
-  version: 0.1.14
+  version: 0.1.16
 platform: x86_64-linux
 authors:
 - kingsley.hendrickse
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-09-17 00:00:00.000000000 Z
+date: 2024-11-25 00:00:00.000000000 Z
 dependencies: []
 description: Deduplication of CSV files and XLS to CSV conversion.
 email: