feat(config): expose wait for options

Cargo.toml

@@ -1,7 +1,7 @@
 [package]
 edition = "2021"
 name = "spider_rs"
-version = "0.0.45"
+version = "0.0.46"
 repository = "https://github.com/spider-rs/spider-py"
 license = "MIT"
 description = "The fastest web crawler and indexer."

src/website.rs

@@ -3,7 +3,7 @@ use indexmap::IndexMap;
 use pyo3::prelude::*;
 use pyo3::types::PyDict;
 use spider::compact_str::CompactString;
-use spider::configuration::WaitForIdleNetwork;
+use spider::configuration::{WaitForDelay, WaitForIdleNetwork, WaitForSelector};
 use spider::tokio::select;
 use spider::tokio::task::JoinHandle;
 use spider::utils::shutdown;
@@ -718,6 +718,46 @@ impl Website {
     slf
   }
 
+  /// Wait for a delay. Should only be used for testing. This method does nothing if the `chrome` feature is not enabled.
+  pub fn with_wait_for_delay(
+    mut slf: PyRefMut<'_, Self>,
+    timeout: Option<u64>,
+  ) -> PyRefMut<'_, Self> {
+    slf
+      .inner
+      .configuration
+      .with_wait_for_delay(if timeout.is_some() {
+        let duration = Duration::from_millis(timeout.unwrap_or_default());
+        Some(WaitForDelay::new(Some(duration)))
+      } else {
+        None
+      });
+
+    slf
+  }
+
+  /// Wait for a CSS query selector. This method does nothing if the `chrome` feature is not enabled.
+  pub fn with_wait_for_selector(
+    mut slf: PyRefMut<'_, Self>,
+    selector: Option<String>,
+    timeout: Option<u64>,
+  ) -> PyRefMut<'_, Self> {
+    slf
+      .inner
+      .configuration
+      .with_wait_for_selector(if timeout.is_some() {
+        let duration = Duration::from_millis(timeout.unwrap_or_default());
+        Some(WaitForSelector::new(
+          Some(duration),
+          selector.unwrap_or_default().to_string(),
+        ))
+      } else {
+        None
+      });
+
+    slf
+  }
+
   /// add external domains
   pub fn with_external_domains(
     mut slf: PyRefMut<'_, Self>,