From 31d0d102ebd6de440bfb03def2c719933bc96607 Mon Sep 17 00:00:00 2001 From: Ruben Fiszel Date: Thu, 30 Mar 2023 01:07:24 +0200 Subject: [PATCH] use requests for . imports --- backend/parsers/windmill-parser-py/src/lib.rs | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/backend/parsers/windmill-parser-py/src/lib.rs b/backend/parsers/windmill-parser-py/src/lib.rs index 0616ec9588..9131cb35fa 100644 --- a/backend/parsers/windmill-parser-py/src/lib.rs +++ b/backend/parsers/windmill-parser-py/src/lib.rs @@ -236,13 +236,11 @@ pub fn parse_python_imports(code: &str) -> error::Result> { .map(replace_import) .collect::>(), ), - 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,