use super::super::AstStrategy;
use crate::services::context::FileContext;
use crate::services::file_classifier::FileClassifier;
use anyhow::Result;
use async_trait::async_trait;
use std::path::Path;
#[cfg(feature = "python-ast")]
pub struct PythonStrategy;
#[cfg(feature = "python-ast")]
impl Default for PythonStrategy {
fn default() -> Self {
Self::new()
}
}
impl PythonStrategy {
#[must_use]
pub fn new() -> Self {
Self
}
}
#[cfg(feature = "python-ast")]
#[async_trait]
impl AstStrategy for PythonStrategy {
async fn analyze(&self, file_path: &Path, _classifier: &FileClassifier) -> Result<FileContext> {
let context = crate::services::ast_python::analyze_python_file(file_path)
.await
.map_err(|e| anyhow::anyhow!("Python analysis failed: {e}"))?;
Ok(context)
}
fn primary_extension(&self) -> &'static str {
"py"
}
fn supported_extensions(&self) -> Vec<&'static str> {
vec!["py", "pyi", "pyw"]
}
fn language_name(&self) -> &'static str {
"Python"
}
}
#[cfg_attr(coverage_nightly, coverage(off))]
#[cfg(test)]
mod property_tests {
use proptest::prelude::*;
proptest! {
#[test]
fn basic_property_stability(_input in ".*") {
prop_assert!(true);
}
#[test]
fn module_consistency_check(_x in 0u32..1000) {
prop_assert!(_x < 1001);
}
}
}