Use top-level semantic detection for E402

crates/ruff/resources/test/fixtures/pycodestyle/E402.py

@@ -30,3 +30,10 @@ def foo() -> None:
 
 if __name__ == "__main__":
     import g
+
+import h; import i
+
+
+if __name__ == "__main__":
+    import j; \
+import k

crates/ruff/src/checkers/ast/analyze/statement.rs

@@ -520,11 +520,7 @@ pub(crate) fn statement(stmt: &Stmt, checker: &mut Checker) {
                 pycodestyle::rules::multiple_imports_on_one_line(checker, stmt, names);
             }
             if checker.enabled(Rule::ModuleImportNotAtTopOfFile) {
-                pycodestyle::rules::module_import_not_at_top_of_file(
-                    checker,
-                    stmt,
-                    checker.locator,
-                );
+                pycodestyle::rules::module_import_not_at_top_of_file(checker, stmt);
             }
             if checker.enabled(Rule::GlobalStatement) {
                 for name in names {
@@ -680,11 +676,7 @@ pub(crate) fn statement(stmt: &Stmt, checker: &mut Checker) {
             let module = module.as_deref();
             let level = level.map(|level| level.to_u32());
             if checker.enabled(Rule::ModuleImportNotAtTopOfFile) {
-                pycodestyle::rules::module_import_not_at_top_of_file(
-                    checker,
-                    stmt,
-                    checker.locator,
-                );
+                pycodestyle::rules::module_import_not_at_top_of_file(checker, stmt);
             }
             if checker.enabled(Rule::GlobalStatement) {
                 for name in names {

crates/ruff/src/rules/pycodestyle/rules/imports.rs

@@ -1,8 +1,6 @@
-use ruff_python_ast::{Alias, Ranged, Stmt};
-
 use ruff_diagnostics::{Diagnostic, Violation};
 use ruff_macros::{derive_message_formats, violation};
-use ruff_source_file::Locator;
+use ruff_python_ast::{Alias, Ranged, Stmt};
 
 use crate::checkers::ast::Checker;
 
@@ -81,12 +79,8 @@ pub(crate) fn multiple_imports_on_one_line(checker: &mut Checker, stmt: &Stmt, n
 }
 
 /// E402
-pub(crate) fn module_import_not_at_top_of_file(
-    checker: &mut Checker,
-    stmt: &Stmt,
-    locator: &Locator,
-) {
-    if checker.semantic().seen_import_boundary() && locator.is_at_start_of_line(stmt.start()) {
+pub(crate) fn module_import_not_at_top_of_file(checker: &mut Checker, stmt: &Stmt) {
+    if checker.semantic().seen_import_boundary() && checker.semantic().at_top_level() {
         checker
             .diagnostics
             .push(Diagnostic::new(ModuleImportNotAtTopOfFile, stmt.range()));

crates/ruff/src/rules/pycodestyle/snapshots/ruff__rules__pycodestyle__tests__E402_E402.py.snap

@@ -9,4 +9,20 @@ E402.py:24:1: E402 Module level import not at top of file
    | ^^^^^^^^ E402
    |
 
+E402.py:34:1: E402 Module level import not at top of file
+   |
+32 |     import g
+33 | 
+34 | import h; import i
+   | ^^^^^^^^ E402
+   |
+
+E402.py:34:11: E402 Module level import not at top of file
+   |
+32 |     import g
+33 | 
+34 | import h; import i
+   |           ^^^^^^^^ E402
+   |
+