[flake8-pyi] Implement PYI017

crates/ruff/resources/test/fixtures/flake8_pyi/PYI017.py

@@ -0,0 +1,11 @@
+a = int # Ok
+
+b = c = int  # Ok
+
+a.b = int  # Ok
+
+d, e = int, str # Ok
+
+f, g, h = int, str, TypeVar("T") # Ok
+
+i = int | str # Ok

crates/ruff/resources/test/fixtures/flake8_pyi/PYI017.pyi

@@ -0,0 +1,16 @@
+a = int # Ok
+
+
+b = c = int  # PYI017
+
+
+a.b = int  # PYI017
+
+
+d, e = int, str # PYI017
+
+
+f, g, h = int, str, TypeVar("T") # PYI017
+
+
+i = int | str # Ok

crates/ruff/src/checkers/ast/mod.rs

@@ -1492,7 +1492,7 @@ where
                     tryceratops::rules::error_instead_of_exception(self, handlers);
                 }
             }
-            Stmt::Assign(ast::StmtAssign { targets, value, .. }) => {
+            Stmt::Assign(stmt_assign @ ast::StmtAssign { targets, value, .. }) => {
                 if self.enabled(Rule::LambdaAssignment) {
                     if let [target] = &targets[..] {
                         pycodestyle::rules::lambda_assignment(self, target, value, None, stmt);
@@ -1557,6 +1557,7 @@ where
                         Rule::UnprefixedTypeParam,
                         Rule::AssignmentDefaultInStub,
                         Rule::UnannotatedAssignmentInStub,
+                        Rule::ComplexAssignment,
                         Rule::TypeAliasWithoutAnnotation,
                     ]) {
                         // Ignore assignments in function bodies; those are covered by other rules.
@@ -1576,6 +1577,9 @@ where
                                     self, targets, value,
                                 );
                             }
+                            if self.enabled(Rule::ComplexAssignment) {
+                                flake8_pyi::rules::complex_assignment(self, stmt_assign);
+                            }
                             if self.enabled(Rule::TypeAliasWithoutAnnotation) {
                                 flake8_pyi::rules::type_alias_without_annotation(
                                     self, value, targets,

crates/ruff/src/codes.rs

@@ -629,6 +629,7 @@ pub fn code_to_rule(linter: Linter, code: &str) -> Option<(RuleGroup, Rule)> {
         (Flake8Pyi, "014") => (RuleGroup::Unspecified, rules::flake8_pyi::rules::ArgumentDefaultInStub),
         (Flake8Pyi, "015") => (RuleGroup::Unspecified, rules::flake8_pyi::rules::AssignmentDefaultInStub),
         (Flake8Pyi, "016") => (RuleGroup::Unspecified, rules::flake8_pyi::rules::DuplicateUnionMember),
+        (Flake8Pyi, "017") => (RuleGroup::Unspecified, rules::flake8_pyi::rules::ComplexAssignment),
         (Flake8Pyi, "020") => (RuleGroup::Unspecified, rules::flake8_pyi::rules::QuotedAnnotationInStub),
         (Flake8Pyi, "021") => (RuleGroup::Unspecified, rules::flake8_pyi::rules::DocstringInStub),
         (Flake8Pyi, "024") => (RuleGroup::Unspecified, rules::flake8_pyi::rules::CollectionsNamedTuple),

crates/ruff/src/rules/flake8_pyi/mod.rs

@@ -25,6 +25,8 @@ mod tests {
     #[test_case(Rule::BadVersionInfoComparison, Path::new("PYI006.pyi"))]
     #[test_case(Rule::CollectionsNamedTuple, Path::new("PYI024.py"))]
     #[test_case(Rule::CollectionsNamedTuple, Path::new("PYI024.pyi"))]
+    #[test_case(Rule::ComplexAssignment, Path::new("PYI017.py"))]
+    #[test_case(Rule::ComplexAssignment, Path::new("PYI017.pyi"))]
     #[test_case(Rule::ComplexIfStatementInStub, Path::new("PYI002.py"))]
     #[test_case(Rule::ComplexIfStatementInStub, Path::new("PYI002.pyi"))]
     #[test_case(Rule::DocstringInStub, Path::new("PYI021.py"))]

crates/ruff/src/rules/flake8_pyi/rules/complex_assignment.rs

@@ -0,0 +1,49 @@
+use rustpython_parser::ast::{Expr, StmtAssign};
+
+use ruff_diagnostics::{Diagnostic, Violation};
+use ruff_macros::{derive_message_formats, violation};
+
+use crate::checkers::ast::Checker;
+
+/// ## What it does
+/// Checks for assignments with multiple or non-name targets
+///
+/// ## Why is this bad?
+/// Stub files are not intended to ever be executed. As such, it's useful to enforce that only a
+/// subset of Python syntax is allowed in a stub file, to ensure that everything in the stub is
+/// 100% unambiguous when it comes to how the type checker is supposed to interpret it. Only
+/// allowing simple assignments is one such restriction.
+///
+/// ## Example
+/// ```python
+/// a = b = int
+/// a.b = int
+/// ```
+///
+/// Use instead:
+/// ```python
+/// a = int
+/// b = int
+///
+/// class a:
+///     b: int
+/// ```
+#[violation]
+pub struct ComplexAssignment;
+
+impl Violation for ComplexAssignment {
+    #[derive_message_formats]
+    fn message(&self) -> String {
+        format!("Stubs should not contain assignments to attributes or multiple targets.")
+    }
+}
+
+/// PYI017
+pub(crate) fn complex_assignment(checker: &mut Checker, stmt: &StmtAssign) {
+    if let [Expr::Name(_)] = stmt.targets[..] {
+        return;
+    }
+    checker
+        .diagnostics
+        .push(Diagnostic::new(ComplexAssignment, stmt.range));
+}

crates/ruff/src/rules/flake8_pyi/rules/mod.rs

@@ -1,6 +1,7 @@
 pub(crate) use any_eq_ne_annotation::*;
 pub(crate) use bad_version_info_comparison::*;
 pub(crate) use collections_named_tuple::*;
+pub(crate) use complex_assignment::*;
 pub(crate) use complex_if_statement_in_stub::*;
 pub(crate) use docstring_in_stubs::*;
 pub(crate) use duplicate_union_member::*;
@@ -31,6 +32,7 @@ pub(crate) use unrecognized_version_info::*;
 mod any_eq_ne_annotation;
 mod bad_version_info_comparison;
 mod collections_named_tuple;
+mod complex_assignment;
 mod complex_if_statement_in_stub;
 mod docstring_in_stubs;
 mod duplicate_union_member;

crates/ruff/src/rules/flake8_pyi/snapshots/ruff__rules__flake8_pyi__tests__PYI017_PYI017.py.snap

@@ -0,0 +1,4 @@
+---
+source: crates/ruff/src/rules/flake8_pyi/mod.rs
+---
+

crates/ruff/src/rules/flake8_pyi/snapshots/ruff__rules__flake8_pyi__tests__PYI017_PYI017.pyi.snap

@@ -0,0 +1,28 @@
+---
+source: crates/ruff/src/rules/flake8_pyi/mod.rs
+---
+PYI017.pyi:4:1: PYI017 Stubs should not contain assignments to attributes or multiple targets.
+  |
+4 | b = c = int  # PYI017
+  | ^^^^^^^^^^^ PYI017
+  |
+
+PYI017.pyi:7:1: PYI017 Stubs should not contain assignments to attributes or multiple targets.
+  |
+7 | a.b = int  # PYI017
+  | ^^^^^^^^^ PYI017
+  |
+
+PYI017.pyi:10:1: PYI017 Stubs should not contain assignments to attributes or multiple targets.
+   |
+10 | d, e = int, str # PYI017
+   | ^^^^^^^^^^^^^^^ PYI017
+   |
+
+PYI017.pyi:13:1: PYI017 Stubs should not contain assignments to attributes or multiple targets.
+   |
+13 | f, g, h = int, str, TypeVar("T") # PYI017
+   | ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ PYI017
+   |
+
+