feat: set python logging

Author: gadomski

Cargo.lock

@@ -27,19 +27,20 @@ serde_json = "1.0.138"
 stac = { features = [
     "geoparquet-compression",
     "object-store-all",
-], git = "https://github.com/stac-utils/stac-rs", branch = "optional-tracing-cli" }
+], git = "https://github.com/stac-utils/stac-rs", branch = "main" }
 stac-api = { features = [
     "client",
     "python",
-], git = "https://github.com/stac-utils/stac-rs", branch = "optional-tracing-cli" }
+], git = "https://github.com/stac-utils/stac-rs", branch = "main" }
 stac-cli = { git = "https://github.com/stac-utils/stac-rs", features = [
     "pgstac",
     "duckdb",
-], branch = "optional-tracing-cli" }
-stac-duckdb = { git = "https://github.com/stac-utils/stac-rs", branch = "optional-tracing-cli" }
-thiserror = "2.0.11"
+], branch = "main" }
+stac-duckdb = { git = "https://github.com/stac-utils/stac-rs", branch = "main" }
+thiserror = "2.0.12"
 tokio = { version = "1.43.0", features = ["rt-multi-thread"] }
 pyo3-log = "0.12.1"
+tracing = "0.1.41"
 
 [patch.crates-io]
 duckdb = { git = "https://github.com/duckdb/duckdb-rs", rev = "5eeb1f01c278790ce1e2d24045f0096e9e2528e4" }

Cargo.toml

@@ -1,6 +1,11 @@
 use clap::Parser;
-use pyo3::{pyfunction, types::PyAnyMethods, PyResult, Python};
+use pyo3::{
+    pyfunction,
+    types::{PyAnyMethods, PyDict},
+    PyResult, Python,
+};
 use stac_cli::Stacrs;
+use tracing::Level;
 
 #[pyfunction]
 pub fn main(py: Python<'_>) -> PyResult<i64> {
@@ -9,6 +14,16 @@ pub fn main(py: Python<'_>) -> PyResult<i64> {
         .getattr("signal")?
         .call1((signal.getattr("SIGINT")?, signal.getattr("SIG_DFL")?))?;
     let args = Stacrs::parse_from(std::env::args_os().skip(1));
+    let logging = py.import("logging")?;
+    let kwargs = PyDict::new(py);
+    kwargs.set_item(
+        "format",
+        "'%(levelname)s %(name)s %(asctime)-15s %(filename)s:%(lineno)d %(message)s'",
+    )?;
+    logging.getattr("basicConfig")?.call((), Some(&kwargs))?;
+    let logger = logging.getattr("getLogger")?.call0()?;
+    let level = args.log_level().unwrap_or(Level::INFO);
+    logger.call_method1("setLevel", (level.to_string(),))?;
     std::process::exit(
         tokio::runtime::Builder::new_multi_thread()
             .enable_all()