language-servers/ruff
1
0
Fork 0
mirror of https://github.com/astral-sh/ruff.git synced 2025-08-09 13:18:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Include decorators in Function and Class definition ranges (#4467)

Browse source
This commit is contained in:
Micha Reiser 2023-05-22 17:50:42 +02:00 • committed by GitHub
parent 9308e939f4
commit daadd24bde
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
11 changed files with 198 additions and 136 deletions
Download patch file Download diff file Expand all files Collapse all files

15
crates/ruff_python_semantic/src/binding.rs
View file

@ -1,7 +1,10 @@
use std::num::TryFromIntError;
use std::ops::{Deref, Index, IndexMut};
use crate::model::SemanticModel;
use bitflags::bitflags;
use ruff_python_ast::helpers;
use ruff_python_ast::source_code::Locator;
use ruff_text_size::TextRange;
use crate::node::NodeId;
@ -108,6 +111,18 @@ impl<'a> Binding<'a> {
}
existing.is_definition()
}
/// Returns the appropriate visual range for highlighting this binding.
pub fn trimmed_range(&self, semantic_model: &SemanticModel, locator: &Locator) -> TextRange {
match self.kind {
BindingKind::ClassDefinition | BindingKind::FunctionDefinition => {
self.source.map_or(self.range, |source| {
helpers::identifier_range(semantic_model.stmts[source], locator)
})
}
_ => self.range,
}
}
}
/// ID uniquely identifying a [Binding] in a program.