nicolas.borges/piperun-bot
1
0
Fork 0
forked from jelson.rodrigues/piperun-bot
Code Pull Requests Activity

chore: calculate excelence percentual for chats

Browse Source
This commit is contained in:
Jelson Stoelben Rodrigues
2025-11-05 10:32:47 -03:00
parent 1ba0510a3e
commit ca227ee45f
3 changed files with 87 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
src/groupped_repport.rs
View File

@@ -262,6 +262,17 @@ fn main() {
.get(5) .get(5)
.expect("Failed to get the id from regex maches"); .expect("Failed to get the id from regex maches");
let excelence_percentual = columns
.iter()
.map(|col| col.as_materialized_series().u32().unwrap().sum().unwrap())
.sum::<u32>() as f32
/ columns.iter().len() as f32
* 100.0;
columns.push(Column::new(
"PERCENTUAL DE EXELENCIA".into(),
[format!("{excelence_percentual:.2}")],
));
columns.push(Column::new("ID_TALK".into(), [talk_id.clone().as_str()])); columns.push(Column::new("ID_TALK".into(), [talk_id.clone().as_str()]));
let df = polars::frame::DataFrame::new(columns) let df = polars::frame::DataFrame::new(columns)
@@ -369,13 +380,13 @@ fn main() {
let recipients = "Wilson da Conceição Oliveira <wilson.oliveira@nova.net.br>, Isadora G. Moura de Moura <isadora.moura@nova.net.br>"; let recipients = "Wilson da Conceição Oliveira <wilson.oliveira@nova.net.br>, Isadora G. Moura de Moura <isadora.moura@nova.net.br>";
println!("Trying to send mail... {recipients}"); println!("Trying to send mail... {recipients}");
send_mail_util::send_mail_util::send_email( // send_mail_util::send_mail_util::send_email(
&format!( // &format!(
"Relatório agrupado dos atendimentos semana {first_day_of_last_week} - {last_day_of_last_week}" // "Relatório agrupado dos atendimentos semana {first_day_of_last_week} - {last_day_of_last_week}"
), // ),
&BOT_EMAIL, // &BOT_EMAIL,
&BOT_EMAIL_PASSWORD, // &BOT_EMAIL_PASSWORD,
recipients, // recipients,
&format!("./groupped/{first_day_of_last_week} - {last_day_of_last_week}.zip"), // &format!("./groupped/{first_day_of_last_week} - {last_day_of_last_week}.zip"),
); // );
} }

14
src/main.rs
View File

@@ -10,8 +10,8 @@ use serde_json::{self, json};
use std::io::prelude::*; use std::io::prelude::*;
pub mod zip_directory_util;
pub mod send_mail_util; pub mod send_mail_util;
pub mod zip_directory_util;
fn main() -> anyhow::Result<()> { fn main() -> anyhow::Result<()> {
match dotenv::dotenv().ok() { match dotenv::dotenv().ok() {
@@ -524,12 +524,18 @@ fn main() -> anyhow::Result<()> {
let source_dir = std::path::Path::new(source_dir_str.as_str()); let source_dir = std::path::Path::new(source_dir_str.as_str());
let output_zip_file = std::path::Path::new(output_zip_file_str.as_str()); let output_zip_file = std::path::Path::new(output_zip_file_str.as_str());
zip_directory_util::zip_directory_util::zip_source_dir_to_dst_file(source_dir, output_zip_file); zip_directory_util::zip_directory_util::zip_source_dir_to_dst_file(source_dir, output_zip_file);
// Send folder to email // Send folder to email
let recipients = "Wilson da Conceição Oliveira <wilson.oliveira@nova.net.br>, Isadora G. Moura de Moura <isadora.moura@nova.net.br>"; let recipients = "Wilson da Conceição Oliveira <wilson.oliveira@nova.net.br>, Isadora G. Moura de Moura <isadora.moura@nova.net.br>";
println!("Trying to send email... Recipients {recipients}"); println!("Trying to send email... Recipients {recipients}");
send_mail_util::send_mail_util::send_email(&format!("Avaliacao atendimentos {formatted_day_before}"), &BOT_EMAIL, &BOT_EMAIL_PASSWORD, recipients, &output_zip_file_str); send_mail_util::send_mail_util::send_email(
&format!("Avaliacao atendimentos {formatted_day_before}"),
&BOT_EMAIL,
&BOT_EMAIL_PASSWORD,
recipients,
&output_zip_file_str,
);
return Ok(()); return Ok(());
} }
@@ -656,4 +662,4 @@ fn get_piperun_chats_on_date(
aggregated_talks.append(&mut all_other_messages); aggregated_talks.append(&mut all_other_messages);
aggregated_talks aggregated_talks
} }

115
src/zip_directory_util.rs
View File

@@ -1,70 +1,69 @@
pub mod zip_directory_util { pub mod zip_directory_util {
use std::io::prelude::*; use std::io::prelude::*;
use zip::write::SimpleFileOptions; use zip::write::SimpleFileOptions;
use std::fs::File; use std::fs::File;
use std::path::Path; use std::path::Path;
use walkdir::{DirEntry, WalkDir}; use walkdir::{DirEntry, WalkDir};
fn zip_dir<T>( fn zip_dir<T>(
it: &mut dyn Iterator<Item = DirEntry>, it: &mut dyn Iterator<Item = DirEntry>,
prefix: &Path, prefix: &Path,
writer: T, writer: T,
method: zip::CompressionMethod, method: zip::CompressionMethod,
) where ) where
T: Write + Seek, T: Write + Seek,
{ {
let mut zip = zip::ZipWriter::new(writer); let mut zip = zip::ZipWriter::new(writer);
let options = SimpleFileOptions::default() let options = SimpleFileOptions::default()
.compression_method(method) .compression_method(method)
.unix_permissions(0o755); .unix_permissions(0o755);
let prefix = Path::new(prefix); let prefix = Path::new(prefix);
let mut buffer = Vec::new(); let mut buffer = Vec::new();
for entry in it { for entry in it {
let path = entry.path(); let path = entry.path();
let name = path.strip_prefix(prefix).unwrap(); let name = path.strip_prefix(prefix).unwrap();
let path_as_string = name let path_as_string = name
.to_str() .to_str()
.map(str::to_owned) .map(str::to_owned)
.expect("Failed to parse path"); .expect("Failed to parse path");
// Write file or directory explicitly // Write file or directory explicitly
// Some unzip tools unzip files with directory paths correctly, some do not! // Some unzip tools unzip files with directory paths correctly, some do not!
if path.is_file() { if path.is_file() {
println!("adding file {path:?} as {name:?} ..."); println!("adding file {path:?} as {name:?} ...");
zip.start_file(path_as_string, options) zip.start_file(path_as_string, options)
.expect("Failed to add file"); .expect("Failed to add file");
let mut f = File::open(path).unwrap(); let mut f = File::open(path).unwrap();
f.read_to_end(&mut buffer).expect("Failed to read file"); f.read_to_end(&mut buffer).expect("Failed to read file");
zip.write_all(&buffer).expect("Failed to write file"); zip.write_all(&buffer).expect("Failed to write file");
buffer.clear(); buffer.clear();
} else if !name.as_os_str().is_empty() { } else if !name.as_os_str().is_empty() {
// Only if not root! Avoids path spec / warning // Only if not root! Avoids path spec / warning
// and mapname conversion failed error on unzip // and mapname conversion failed error on unzip
println!("adding dir {path_as_string:?} as {name:?} ..."); println!("adding dir {path_as_string:?} as {name:?} ...");
zip.add_directory(path_as_string, options) zip.add_directory(path_as_string, options)
.expect("Failed to add directory"); .expect("Failed to add directory");
}
} }
zip.finish().expect("Failed to ZIP");
} }
zip.finish().expect("Failed to ZIP");
}
pub fn zip_source_dir_to_dst_file(src_dir: &Path, dst_file: &Path) { pub fn zip_source_dir_to_dst_file(src_dir: &Path, dst_file: &Path) {
if !Path::new(src_dir).is_dir() { if !Path::new(src_dir).is_dir() {
panic!("src_dir must be a directory"); panic!("src_dir must be a directory");
}
let method = zip::CompressionMethod::Stored;
let path = Path::new(dst_file);
let file = File::create(path).unwrap();
let walkdir = WalkDir::new(src_dir);
let it = walkdir.into_iter();
zip_dir(&mut it.filter_map(|e| e.ok()), src_dir, file, method);
} }
let method = zip::CompressionMethod::Stored;
let path = Path::new(dst_file);
let file = File::create(path).unwrap();
let walkdir = WalkDir::new(src_dir);
let it = walkdir.into_iter();
zip_dir(&mut it.filter_map(|e| e.ok()), src_dir, file, method);
} }
}