Implement a common trait for the string flags (#11564)

This commit is contained in:
Alex Waygood 2024-05-27 16:02:01 +01:00 committed by GitHub
parent 6be00d5775
commit 246a3388ee
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
20 changed files with 184 additions and 153 deletions

View file

@ -1,4 +1,4 @@
use ruff_python_ast::AnyStringFlags;
use ruff_python_ast::{AnyStringFlags, StringFlags};
/// The context representing the current f-string that the lexer is in.
#[derive(Debug)]
@ -36,13 +36,13 @@ impl FStringContext {
}
/// Returns the quote character for the current f-string.
pub(crate) const fn quote_char(&self) -> char {
pub(crate) fn quote_char(&self) -> char {
self.flags.quote_style().as_char()
}
/// Returns the triple quotes for the current f-string if it is a triple-quoted
/// f-string, `None` otherwise.
pub(crate) const fn triple_quotes(&self) -> Option<&'static str> {
pub(crate) fn triple_quotes(&self) -> Option<&'static str> {
if self.is_triple_quoted() {
Some(self.flags.quote_str())
} else {
@ -56,7 +56,7 @@ impl FStringContext {
}
/// Returns `true` if the current f-string is a triple-quoted f-string.
pub(crate) const fn is_triple_quoted(&self) -> bool {
pub(crate) fn is_triple_quoted(&self) -> bool {
self.flags.is_triple_quoted()
}