research/psboard-qaqc-postprocess
1
0
Fork 0
mirror of https://gitlab.cern.ch/wotsubo/psboard-qaqc-postprocess.git synced 2025-07-02 01:29:28 +09:00
Code Issues Projects Releases Packages Wiki Activity Actions

update: organize commands to subcommands and rename bin

Browse source 
- existing functions are moved to subcommand `add-master-log`
This commit is contained in:
Wataru Otsubo 2024-07-21 21:04:46 +09:00
parent bdb415e6e6
commit a94a871e52
5 changed files with 81 additions and 56 deletions
Download patch file Download diff file Expand all files Collapse all files

119
src/main.rs
View file

@ -10,26 +10,35 @@ use std::{
use anyhow::{anyhow, Context, Result};
use chrono::{DateTime, Local, Utc};
use clap::Parser;
use clap::{Parser, Subcommand};
use log::{debug, error, info, trace, warn};
use regex::Regex;
use semver::Version;
use serde::{Deserialize, Serialize};
use serde_with::{serde_as, DisplayFromStr};
/// Parse master jathub logfile for PS Board QAQC and write out to CSV.
/// PS Board QAQC shift related commands.
#[derive(Parser, Debug)]
#[command(version, about, long_about = None)]
struct Args {
/// Master log file.
master_log: path::PathBuf,
/// Output CSV file.
// #[arg(default_value = get_default_log_path().into_os_string())]
outfile: PathBuf,
#[command(subcommand)]
command: Commands,
#[command(flatten)]
verbose: clap_verbosity_flag::Verbosity,
}
#[derive(Subcommand, Debug)]
pub enum Commands {
/// Parse master jathub logfile for PS Board QAQC and write out to CSV.
AddMasterLog {
/// Master log file.
master_log: path::PathBuf,
/// Output CSV file.
// #[arg(default_value = get_default_log_path().into_os_string())]
outfile: PathBuf,
},
}
/// Layer
#[derive(Debug, PartialEq, Eq, Hash, Clone, PartialOrd, Ord)]
enum PositionLayer {
@ -444,55 +453,58 @@ fn main() -> Result<()> {
" 1".split_whitespace().next().unwrap().parse::<u8>()
);
let result = {
let file = File::open(args.master_log.clone())?;
let reader = BufReader::new(file);
MasterLogResult::parse_file(
reader,
args.master_log
.file_name()
.unwrap()
.to_str()
.unwrap()
.to_string(),
)?
};
debug!("{:?}", result);
match args.command {
Commands::AddMasterLog {
master_log,
outfile,
} => {
let result = {
let file = File::open(master_log.clone())?;
let reader = BufReader::new(file);
MasterLogResult::parse_file(
reader,
master_log
.file_name()
.unwrap()
.to_str()
.unwrap()
.to_string(),
)?
};
debug!("{:?}", result);
// {
// let file = File::options()
// .read(true)
// .open(args.outfile)?;
//
// let mut rdr = csv::Reader::from_reader(file);
// rdr.records();
// }
// {
// let file = File::options()
// .read(true)
// .open(args.outfile)?;
//
// let mut rdr = csv::Reader::from_reader(file);
// rdr.records();
// }
let expanded_results = PsbQaqcResult::from_masterlogresult(result);
let expanded_results = PsbQaqcResult::from_masterlogresult(result);
let mut wtr = match args.outfile.exists() {
true => {
let file = File::options().read(true).append(true).open(args.outfile)?;
csv::WriterBuilder::new()
.has_headers(false)
.from_writer(file)
let mut wtr = match outfile.exists() {
true => {
let file = File::options().read(true).append(true).open(outfile)?;
csv::WriterBuilder::new()
.has_headers(false)
.from_writer(file)
}
false => {
println!("Creating new file: {}", outfile.display());
let file = File::options().create_new(true).write(true).open(outfile)?;
csv::WriterBuilder::new()
.has_headers(true)
.from_writer(file)
}
};
for result in expanded_results {
wtr.serialize(result)?;
}
wtr.flush()?;
}
false => {
println!("Creating new file: {}", args.outfile.display());
let file = File::options()
.create_new(true)
.write(true)
.open(args.outfile)?;
csv::WriterBuilder::new()
.has_headers(true)
.from_writer(file)
}
};
for result in expanded_results {
wtr.serialize(result)?;
}
wtr.flush()?;
Ok(())
}
@ -625,4 +637,9 @@ mod test {
)
);
}
// #[test]
// fn parse_file() {
// let logfile = r"""""";
// }
}