Ignore relative imports in banned-api rules

astral-sh · Apr 19, 2023 · 53431e9 · 53431e9
1 parent 13fda30
commit 53431e9
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/crates/ruff/resources/test/fixtures/flake8_tidy_imports/TID251.py b/crates/ruff/resources/test/fixtures/flake8_tidy_imports/TID251.py
@@ -31,3 +31,6 @@
 # import aliases are resolved
 import typing as totally_not_typing
 totally_not_typing.TypedDict
+
+# relative imports are respected
+from .typing import TypedDict
diff --git a/crates/ruff/src/checkers/ast/mod.rs b/crates/ruff/src/checkers/ast/mod.rs
@@ -1155,13 +1155,15 @@ where
                 }
 
                 if self.settings.rules.enabled(Rule::BannedApi) {
-                    if let Some(module) = module {
-                        for name in names {
-                            flake8_tidy_imports::banned_api::name_is_banned(self, module, name);
+                    if level.map_or(true, |level| level == 0) {
+                        if let Some(module) = module {
+                            for name in names {
+                                flake8_tidy_imports::banned_api::name_is_banned(self, module, name);
+                            }
+                            flake8_tidy_imports::banned_api::name_or_parent_is_banned(
+                                self, module, stmt,
+                            );
                         }
-                        flake8_tidy_imports::banned_api::name_or_parent_is_banned(
-                            self, module, stmt,
-                        );
                     }
                 }