Implement C404

README.md

@@ -276,6 +276,7 @@ The 🛠 emoji indicates that a rule is automatically fixable by the `--fix` com
 | A002 | BuiltinArgumentShadowing | Argument `...` is shadowing a python builtin |  |  |
 | A003 | BuiltinAttributeShadowing | Class attribute `...` is shadowing a python builtin |  |  |
 | C403 | UnnecessaryListComprehensionSet | Unnecessary list comprehension - rewrite as a set comprehension |  |  |
+| C404 | UnnecessaryListComprehensionDict | Unnecessary list comprehension - rewrite as a dict comprehension |  |  |
 | SPR001 | SuperCallWithParameters | Use `super()` instead of `super(__class__, self)` |  | 🛠 |
 | T201 | PrintFound | `print` found |  | 🛠 |
 | T203 | PPrintFound | `pprint` found |  | 🛠 |

resources/test/fixtures/C404.py

@@ -0,0 +1 @@
+d = dict([(i, i) for i in range(3)])

src/ast/checks.rs

@@ -695,12 +695,26 @@ pub fn is_super_call_with_arguments(func: &Expr, args: &Vec<Expr>) -> bool {
 // flakes8-comprehensions
 pub fn unnecessary_list_comprehension(expr: &Expr, func: &Expr, args: &Vec<Expr>) -> Option<Check> {
     if let ExprKind::Name { id, .. } = &func.node {
-        if id == "set" && args.len() == 1 {
-            if let ExprKind::ListComp { .. } = &args[0].node {
-                return Some(Check::new(
-                    CheckKind::UnnecessaryListComprehensionSet,
-                    Range::from_located(expr),
-                ));
+        if (id == "set" || id == "dict") && args.len() == 1 {
+            if let ExprKind::ListComp { elt, .. } = &args[0].node {
+                if id == "set" {
+                    return Some(Check::new(
+                        CheckKind::UnnecessaryListComprehensionSet,
+                        Range::from_located(expr),
+                    ));
+                }
+
+                if id == "dict" {
+                    match &elt.node {
+                        ExprKind::Tuple { elts, .. } if elts.len() == 2 => {
+                            return Some(Check::new(
+                                CheckKind::UnnecessaryListComprehensionDict,
+                                Range::from_located(expr),
+                            ));
+                        }
+                        _ => {}
+                    }
+                }
             }
         }
     }

src/check_ast.rs

@@ -766,7 +766,9 @@ where
                 }
 
                 // flake8-comprehensions
-                if self.settings.enabled.contains(&CheckCode::C403) {
+                if self.settings.enabled.contains(&CheckCode::C403)
+                    || self.settings.enabled.contains(&CheckCode::C404)
+                {
                     if let Some(check) = checks::unnecessary_list_comprehension(expr, func, args) {
                         self.checks.push(check);
                     };

src/checks.rs

@@ -53,7 +53,7 @@ pub const DEFAULT_CHECK_CODES: [CheckCode; 42] = [
     CheckCode::F901,
 ];
 
-pub const ALL_CHECK_CODES: [CheckCode; 53] = [
+pub const ALL_CHECK_CODES: [CheckCode; 54] = [
     // pycodestyle
     CheckCode::E402,
     CheckCode::E501,
@@ -104,6 +104,7 @@ pub const ALL_CHECK_CODES: [CheckCode; 53] = [
     CheckCode::A003,
     // flake8-comprehensions
     CheckCode::C403,
+    CheckCode::C404,
     // flake8-super
     CheckCode::SPR001,
     // flake8-print
@@ -170,6 +171,7 @@ pub enum CheckCode {
     A003,
     // flake8-comprehensions
     C403,
+    C404,
     // flake8-super
     SPR001,
     // flake8-print
@@ -239,6 +241,7 @@ impl FromStr for CheckCode {
             "A003" => Ok(CheckCode::A003),
             // flake8-comprehensions
             "C403" => Ok(CheckCode::C403),
+            "C404" => Ok(CheckCode::C404),
             // flake8-super
             "SPR001" => Ok(CheckCode::SPR001),
             // flake8-print
@@ -309,6 +312,7 @@ impl CheckCode {
             CheckCode::A003 => "A003",
             // flake8-comprehensions
             CheckCode::C403 => "C403",
+            CheckCode::C404 => "C404",
             // flake8-super
             CheckCode::SPR001 => "SPR001",
             // flake8-print
@@ -388,6 +392,7 @@ impl CheckCode {
             CheckCode::A003 => CheckKind::BuiltinAttributeShadowing("...".to_string()),
             // flake8-comprehensions
             CheckCode::C403 => CheckKind::UnnecessaryListComprehensionSet,
+            CheckCode::C404 => CheckKind::UnnecessaryListComprehensionDict,
             // flake8-super
             CheckCode::SPR001 => CheckKind::SuperCallWithParameters,
             // flake8-print
@@ -471,6 +476,7 @@ pub enum CheckKind {
     BuiltinAttributeShadowing(String),
     // flakes8-comprehensions
     UnnecessaryListComprehensionSet,
+    UnnecessaryListComprehensionDict,
     // flake8-super
     SuperCallWithParameters,
     // flake8-print
@@ -530,6 +536,7 @@ impl CheckKind {
             CheckKind::BuiltinAttributeShadowing(_) => "BuiltinAttributeShadowing",
             // flake8-comprehensions
             CheckKind::UnnecessaryListComprehensionSet => "UnnecessaryListComprehensionSet",
+            CheckKind::UnnecessaryListComprehensionDict => "UnnecessaryListComprehensionDict",
             // flake8-super
             CheckKind::SuperCallWithParameters => "SuperCallWithParameters",
             // flake8-print
@@ -596,6 +603,7 @@ impl CheckKind {
             CheckKind::BuiltinAttributeShadowing(_) => &CheckCode::A003,
             // flake8-comprehensions
             CheckKind::UnnecessaryListComprehensionSet => &CheckCode::C403,
+            CheckKind::UnnecessaryListComprehensionDict => &CheckCode::C404,
             // flake8-super
             CheckKind::SuperCallWithParameters => &CheckCode::SPR001,
             // flake8-print
@@ -753,6 +761,9 @@ impl CheckKind {
             CheckKind::UnnecessaryListComprehensionSet => {
                 "Unnecessary list comprehension - rewrite as a set comprehension".to_string()
             }
+            CheckKind::UnnecessaryListComprehensionDict => {
+                "Unnecessary list comprehension - rewrite as a dict comprehension".to_string()
+            }
             // flake8-super
             CheckKind::SuperCallWithParameters => {
                 "Use `super()` instead of `super(__class__, self)`".to_string()

src/linter.rs

@@ -798,6 +798,18 @@ mod tests {
         Ok(())
     }
 
+    #[test]
+    fn c404() -> Result<()> {
+        let mut checks = check_path(
+            Path::new("./resources/test/fixtures/C404.py"),
+            &settings::Settings::for_rule(CheckCode::C404),
+            &fixer::Mode::Generate,
+        )?;
+        checks.sort_by_key(|check| check.location);
+        insta::assert_yaml_snapshot!(checks);
+        Ok(())
+    }
+
     #[test]
     fn spr001() -> Result<()> {
         let mut checks = check_path(

src/snapshots/ruff__linter__tests__c404.snap

@@ -0,0 +1,13 @@
+---
+source: src/linter.rs
+expression: checks
+---
+- kind: UnnecessaryListComprehensionDict
+  location:
+    row: 1
+    column: 5
+  end_location:
+    row: 1
+    column: 37
+  fix: ~
+