Python
/
ruff
mirror of https://github.com/astral-sh/ruff
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

refactor: Avoid unnecessary Map indexing

This commit is contained in:
Martin Fischer 2023-01-21 05:49:06 +01:00 committed by Charlie Marsh
parent aedee7294e
commit 16d2ceba79
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ruff_macros/src/rule_code_prefix.rs
View File

@ -93,6 +93,8 @@ pub fn expand<'a>(
// Build up a map from prefix to matching RuleCodes.
let mut prefix_to_codes: BTreeMap<String, BTreeSet<String>> = BTreeMap::default();
let mut all_codes = BTreeSet::new();
for variant in variants {
let code_str = variant.to_string();
let code_prefix_len = code_str
@ -107,12 +109,11 @@ pub fn expand<'a>(
.or_default()
.insert(code_str.clone());
}
prefix_to_codes
.entry(ALL.to_string())
.or_default()
.insert(code_str.clone());
all_codes.insert(code_str);
}
prefix_to_codes.insert(ALL.to_string(), all_codes);
// Add any prefix aliases (e.g., "U" to "UP").
for (alias, rule_code) in PREFIX_REDIRECTS.iter() {
prefix_to_codes.insert(