made counter.rs

src/counter.rs

@@ -0,0 +1,22 @@
+use crate::{ballot::Ballot, header::Header};
+
+
+#[derive(Debug)]
+pub struct Counter {
+	pub header: Header,
+	pub ballots: Vec<Ballot>,
+}
+
+impl Counter {
+	pub fn new<T: std::io::BufRead>(mut csv: quick_csv::Csv<T>) -> Result<Self, Box<dyn std::error::Error>> {
+		let header = Header::parse(csv.next().ok_or("csv header missing")??)?;
+		let ballots = Ballot::parse_rows(csv, &header)?;
+
+		Ok(Counter {
+			header,
+			ballots,
+		})
+	}
+
+}
+

src/main.rs

@@ -1,11 +1,13 @@
 use std::env;
 
 use ballot::{Ballot, BallotType};
+use counter::Counter;
 use header::Header;
 
 pub mod candidate;
-mod ballot;
-mod header;
+pub mod ballot;
+pub mod header;
+mod counter;
 
 fn main() {
 	let csv_path = {
@@ -16,18 +18,17 @@ fn main() {
 		}
 		args[1].clone()
 	};
-	let mut csv = quick_csv::Csv::from_file(csv_path).unwrap().flexible(true);
-	let header = Header::parse(csv.next().unwrap().unwrap()).unwrap();
-	let ballots = Ballot::parse_rows(csv, &header).unwrap();
+	let csv = quick_csv::Csv::from_file(csv_path).unwrap().flexible(true);
+	let counter = Counter::new(csv).unwrap();
 
-	println!("Parties: {}", header.parties.len());
-	println!("Candidates: {}", header.candidates.len());
+	println!("Parties: {}", counter.header.parties.len());
+	println!("Candidates: {}", counter.header.candidates.len());
 
-	let total = ballots.len() as u32;
+	let total = counter.ballots.len() as u32;
 	let mut total_atl = 0u32;
 	let mut total_btl = 0u32;
 
-	for ballot in ballots.iter() {
+	for ballot in counter.ballots.iter() {
 		*match ballot.ty {
 			BallotType::Party => &mut total_atl,
 			BallotType::Candidate => &mut total_btl,