use requests for . imports

This commit is contained in:
Ruben Fiszel
2023-03-30 01:07:24 +02:00
parent fbe5c18da0
commit 31d0d102eb
@@ -236,13 +236,11 @@ pub fn parse_python_imports(code: &str) -> error::Result<Vec<String>> {
.map(replace_import)
.collect::<Vec<String>>(),
),
StmtKind::ImportFrom { level, module: Some(mod_), names: _ } => {
let imprt = if level.is_some() && level.unwrap() > 0 {
".".to_string()
} else {
mod_.split('.').next().unwrap_or("").replace("_", "-")
};
StmtKind::ImportFrom { level: Some(i), .. } if i > 0 => {
Some(vec!["requests".to_string()])
}
StmtKind::ImportFrom { level: _, module: Some(mod_), names: _ } => {
let imprt = mod_.split('.').next().unwrap_or("").replace("_", "-");
Some(vec![replace_import(imprt)])
}
_ => None,