Nulo/preciazo
1
0
Fork 0
mirror of https://github.com/catdevnull/preciazo.git synced 2024-11-26 03:26:19 +00:00
Code Issues Projects Releases Packages Wiki Activity

no requerir telegram para auto

Browse source
This commit is contained in:
Cat /dev/Nulo 2024-01-25 16:56:27 -03:00
parent f7bc0a9db8
commit cce34571f1
1 changed files with 30 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
scraper-rs/src/main.rs
View file

@ -296,11 +296,16 @@ async fn scrap_url(
} }
} }
#[derive(Clone)]
struct AutoTelegram {
token: String,
chat_id: String,
}
#[derive(Clone)] #[derive(Clone)]
struct Auto { struct Auto {
pool: Pool, pool: Pool,
telegram_token: String, telegram: Option<AutoTelegram>,
telegram_chat_id: String,
} }
impl Auto { impl Auto {
async fn download_supermercado(self: Self, supermercado: Supermercado) -> anyhow::Result<()> { async fn download_supermercado(self: Self, supermercado: Supermercado) -> anyhow::Result<()> {
@ -383,28 +388,35 @@ impl Auto {
async fn inform(self: &Self, msg: &str) { async fn inform(self: &Self, msg: &str) {
println!("{}", msg); println!("{}", msg);
let u = Url::parse_with_params( if let Some(telegram) = &self.telegram {
&format!( let u = Url::parse_with_params(
"https://api.telegram.org/bot{}/sendMessage", &format!("https://api.telegram.org/bot{}/sendMessage", telegram.token),
self.telegram_token &[
), ("chat_id", telegram.chat_id.clone()),
&[ ("text", msg.to_string()),
("chat_id", self.telegram_chat_id.clone()), ],
("text", msg.to_string()), )
], .unwrap();
) reqwest::get(u).await.unwrap();
.unwrap(); }
reqwest::get(u).await.unwrap();
} }
} }
async fn auto_cli() -> anyhow::Result<()> { async fn auto_cli() -> anyhow::Result<()> {
let db = connect_db(); let db = connect_db();
let auto = Auto { let telegram = {
pool: db, match (
telegram_token: env::var("TELEGRAM_BOT_TOKEN")?, env::var("TELEGRAM_BOT_TOKEN"),
telegram_chat_id: env::var("TELEGRAM_BOT_CHAT_ID")?, env::var("TELEGRAM_BOT_CHAT_ID"),
) {
(Ok(token), Ok(chat_id)) => Some(AutoTelegram { token, chat_id }),
_ => {
tracing::warn!("No token or chat_id for telegram");
None
}
}
}; };
let auto = Auto { pool: db, telegram };
auto.inform("[auto] Empezando scrap").await; auto.inform("[auto] Empezando scrap").await;
let handles: Vec<_> = Supermercado::value_variants() let handles: Vec<_> = Supermercado::value_variants()
.iter() .iter()