language-servers/ruff
1
0
Fork 0
mirror of https://github.com/astral-sh/ruff.git synced 2025-11-01 20:31:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

refactor: Introduce get_indent helper for scripts

Browse source
This commit is contained in:
Martin Fischer 2023-01-20 08:00:59 +01:00 • committed by Charlie Marsh
parent 67ca50e9f2
commit ff6defc988
3 changed files with 12 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

6
scripts/add_rule.py
View file

@ -11,7 +11,7 @@ Example usage:
import argparse
from _utils import ROOT_DIR, dir_name
from _utils import ROOT_DIR, dir_name, get_indent
def snake_case(name: str) -> str:
@ -34,7 +34,7 @@ def main(*, name: str, code: str, origin: str) -> None:
with open(mod_rs, "w") as fp:
for line in content.splitlines():
if line.strip() == "fn rules(rule_code: Rule, path: &Path) -> Result<()> {":
indent = line.split("fn rules(rule_code: Rule, path: &Path) -> Result<()> {")[0]
indent = get_indent(line)
fp.write(f'{indent}#[test_case(Rule::{code}, Path::new("{code}.py"); "{code}")]')
fp.write("\n")
@ -99,7 +99,7 @@ impl Violation for %s {
continue
if line.strip() == f"// {origin}":
indent = line.split("//")[0]
indent = get_indent(line)
fp.write(f"{indent}{code} => violations::{name},")
fp.write("\n")
has_written = True