Add tryceratops to flake8-to-ruff

astral-sh · Jan 24, 2023 · a204829 · a204829
1 parent d97c078
commit a204829
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/flake8_to_ruff/plugin.rs b/src/flake8_to_ruff/plugin.rs
@@ -40,6 +40,7 @@ pub enum Plugin {
     PEP8Naming,
     PandasVet,
     Pyupgrade,
+    Tryceratops,
 }
 
 impl FromStr for Plugin {
@@ -79,6 +80,7 @@ impl FromStr for Plugin {
             "pep8-naming" => Ok(Plugin::PEP8Naming),
             "pandas-vet" => Ok(Plugin::PandasVet),
             "pyupgrade" => Ok(Plugin::Pyupgrade),
+            "tryceratops" => Ok(Plugin::Tryceratops),
             _ => Err(anyhow!("Unknown plugin: {string}")),
         }
     }
@@ -122,6 +124,7 @@ impl fmt::Debug for Plugin {
                 Plugin::PEP8Naming => "pep8-naming",
                 Plugin::PandasVet => "pandas-vet",
                 Plugin::Pyupgrade => "pyupgrade",
+                Plugin::Tryceratops => "tryceratops",
             }
         )
     }
@@ -154,14 +157,15 @@ impl Plugin {
             Plugin::Flake8Quotes => RuleSelector::Q,
             Plugin::Flake8Return => RuleSelector::RET,
             Plugin::Flake8Simplify => RuleSelector::SIM,
-            Plugin::Flake8TidyImports => RuleSelector::TID25,
+            Plugin::Flake8TidyImports => RuleSelector::TID,
             Plugin::Flake8TypeChecking => RuleSelector::TYP,
             Plugin::Flake8UnusedArguments => RuleSelector::ARG,
             Plugin::Flake8UsePathlib => RuleSelector::PTH,
             Plugin::McCabe => RuleSelector::C9,
             Plugin::PEP8Naming => RuleSelector::N,
             Plugin::PandasVet => RuleSelector::PD,
             Plugin::Pyupgrade => RuleSelector::UP,
+            Plugin::Tryceratops => RuleSelector::TRY,
         }
     }
 }
@@ -314,13 +318,14 @@ pub fn infer_plugins_from_codes(selectors: &BTreeSet<RuleSelector>) -> Vec<Plugi
         Plugin::Flake8Quotes,
         Plugin::Flake8Return,
         Plugin::Flake8Simplify,
-        Plugin::Flake8TidyImports,
+        // Plugin::Flake8TidyImports,
         // Plugin::Flake8TypeChecking,
         Plugin::Flake8UnusedArguments,
         // Plugin::Flake8UsePathlib,
         Plugin::McCabe,
         Plugin::PEP8Naming,
         Plugin::PandasVet,
+        Plugin::Tryceratops,
     ]
     .into_iter()
     .filter(|plugin| {