diff --git a/src/flamegraph/color/palettes.rs b/src/flamegraph/color/palettes.rs index d83cdbe9..457e3e0a 100644 --- a/src/flamegraph/color/palettes.rs +++ b/src/flamegraph/color/palettes.rs @@ -61,16 +61,23 @@ pub(super) mod perl { pub(super) mod python { use crate::flamegraph::color::BasicPalette; + use std::path::Path; pub fn resolve(name: &str) -> BasicPalette { if name.starts_with("native@") { // austin-specific format for native calls return BasicPalette::Aqua; - } else if name.contains("/site-packages/") || name.contains("\\site-packages\\") { + } else if Path::new(name).iter().any(|c| c == "site-packages") { return BasicPalette::Yellow; - } else if (name.contains("/python") && name.contains("/lib/")) - || (name.contains("\\Python") && name.contains("\\lib\\")) - || name.starts_with("