From 7acfc33b7477e0f43c1c26e1fd4a27e5b280348e Mon Sep 17 00:00:00 2001 From: ByteDice Date: Tue, 24 Feb 2026 16:25:43 +0100 Subject: [PATCH] made bot usable again --- src/gen.rs | 6 +++++- src/messages.rs | 5 +++-- src/python.rs | 2 ++ 3 files changed, 10 insertions(+), 3 deletions(-) diff --git a/src/gen.rs b/src/gen.rs index d616c7f..03e6497 100644 --- a/src/gen.rs +++ b/src/gen.rs @@ -51,7 +51,11 @@ pub async fn gen_bot(data: Data, args: Args) -> Client { let token_end_len = token[peek_len..].len(); rs_println!("Token: {}{}", token_peek, "*".repeat(token_end_len)); - let own: HashSet = data.owners.clone().into_iter().map(UserId::from).collect(); + let own: HashSet = data.owners + .clone() + .into_iter() + .filter_map(|i| if i == 0 { None } else { Some(UserId::from(i))}) + .collect(); let framework = poise::Framework::builder() .options(poise::FrameworkOptions { diff --git a/src/messages.rs b/src/messages.rs index d5002bb..197bcef 100644 --- a/src/messages.rs +++ b/src/messages.rs @@ -199,7 +199,7 @@ pub async fn http_edit_msg( } -pub async fn send_dm(msg: String, args: Args, owners: Vec) { +pub async fn send_dm(msg: String, args: Args, receivers: Vec) { let token: String = if !args.test { env::var("ASSISTANT_TOKEN") .expect("Missing ASSISTANT_TOKEN env var!") } else { env::var("ASSISTANT_TOKEN_TEST").expect("Missing ASSISTANT_TOKEN_TEST env var!") }; @@ -208,7 +208,8 @@ pub async fn send_dm(msg: String, args: Args, owners: Vec) { let c_msg = CreateMessage::new().content(msg); - for uid in owners { + for uid in receivers { + if uid == 0 { continue; } let user = UserId::new(uid); let _ = user.dm(http.as_ref(), c_msg.clone()).await; } diff --git a/src/python.rs b/src/python.rs index 6e87f77..ce8cee0 100644 --- a/src/python.rs +++ b/src/python.rs @@ -36,6 +36,8 @@ pub async fn start(args: Args) -> PyResult<()> { let mut traceback: String = String::new(); let mut is_error = false; + Python::initialize(); + let _ = Python::attach(|py| -> Result<(), PyErr> { let syspath = py.import("sys")?.getattr("path")?.cast_into::()?; syspath.insert(0, path)?;