fix: Update add_plugin.py to account for b78b6f27

This commit is contained in:
Martin Fischer 2023-01-20 08:18:02 +01:00 committed by Charlie Marsh
parent c00df647e1
commit 3aec1100f5

View file

@ -27,6 +27,7 @@ def main(*, plugin: str, url: str) -> None:
with open(rust_module / "rules.rs", "w+") as fp:
fp.write("use crate::checkers::ast::Checker;\n")
with open(rust_module / "mod.rs", "w+") as fp:
fp.write(f"//! Rules from [{plugin}]({url}).\n")
fp.write("pub(crate) mod rules;\n")
fp.write("\n")
fp.write(
@ -77,10 +78,6 @@ mod tests {
fp.write(f"{indent}{pascal_case(plugin)},")
fp.write("\n")
elif line.strip() == 'RuleOrigin::Ruff => "Ruff-specific rules",':
fp.write(f'{indent}RuleOrigin::{pascal_case(plugin)} => "{plugin}",')
fp.write("\n")
elif line.strip() == "RuleOrigin::Ruff => vec![RuleCodePrefix::RUF],":
fp.write(
f"{indent}RuleOrigin::{pascal_case(plugin)} => vec![\n"
@ -89,10 +86,6 @@ mod tests {
)
fp.write("\n")
elif line.strip() == "RuleOrigin::Ruff => None,":
fp.write(f"{indent}RuleOrigin::{pascal_case(plugin)} => " f'Some(("{url}", &Platform::PyPI)),')
fp.write("\n")
fp.write(line)
fp.write("\n")