Rename as_str to to_str (#8886)

This PR renames the method on `StringLiteralValue` from `as_str` to
`to_str`. The main motivation is to follow the naming convention as
described in the [Rust API
Guidelines](https://rust-lang.github.io/api-guidelines/naming.html#ad-hoc-conversions-follow-as_-to_-into_-conventions-c-conv).
This method can perform a string allocation in case the string is
implicitly concatenated.
This commit is contained in:
Dhruv Manilawala 2023-11-28 18:50:42 -06:00 committed by GitHub
parent b28556d739
commit ec7456bac0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
38 changed files with 68 additions and 68 deletions

View file

@ -380,13 +380,13 @@ pub(crate) fn expression(expr: &Expr, checker: &mut Checker) {
flynt::rules::static_join_to_fstring( flynt::rules::static_join_to_fstring(
checker, checker,
expr, expr,
string_value.as_str(), string_value.to_str(),
); );
} }
} else if attr == "format" { } else if attr == "format" {
// "...".format(...) call // "...".format(...) call
let location = expr.range(); let location = expr.range();
match pyflakes::format::FormatSummary::try_from(string_value.as_str()) { match pyflakes::format::FormatSummary::try_from(string_value.to_str()) {
Err(e) => { Err(e) => {
if checker.enabled(Rule::StringDotFormatInvalidFormat) { if checker.enabled(Rule::StringDotFormatInvalidFormat) {
checker.diagnostics.push(Diagnostic::new( checker.diagnostics.push(Diagnostic::new(
@ -432,7 +432,7 @@ pub(crate) fn expression(expr: &Expr, checker: &mut Checker) {
if checker.enabled(Rule::BadStringFormatCharacter) { if checker.enabled(Rule::BadStringFormatCharacter) {
pylint::rules::bad_string_format_character::call( pylint::rules::bad_string_format_character::call(
checker, checker,
string_value.as_str(), string_value.to_str(),
location, location,
); );
} }
@ -1045,7 +1045,7 @@ pub(crate) fn expression(expr: &Expr, checker: &mut Checker) {
Rule::PercentFormatUnsupportedFormatCharacter, Rule::PercentFormatUnsupportedFormatCharacter,
]) { ]) {
let location = expr.range(); let location = expr.range();
match pyflakes::cformat::CFormatSummary::try_from(value.as_str()) { match pyflakes::cformat::CFormatSummary::try_from(value.to_str()) {
Err(CFormatError { Err(CFormatError {
typ: CFormatErrorType::UnsupportedFormatChar(c), typ: CFormatErrorType::UnsupportedFormatChar(c),
.. ..

View file

@ -815,7 +815,7 @@ where
if let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = expr { if let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = expr {
self.deferred.string_type_definitions.push(( self.deferred.string_type_definitions.push((
expr.range(), expr.range(),
value.as_str(), value.to_str(),
self.semantic.snapshot(), self.semantic.snapshot(),
)); ));
} else { } else {
@ -1235,7 +1235,7 @@ where
{ {
self.deferred.string_type_definitions.push(( self.deferred.string_type_definitions.push((
expr.range(), expr.range(),
value.as_str(), value.to_str(),
self.semantic.snapshot(), self.semantic.snapshot(),
)); ));
} }

View file

@ -508,7 +508,7 @@ fn check_dynamically_typed<F>(
if let Expr::StringLiteral(ast::ExprStringLiteral { range, value }) = annotation { if let Expr::StringLiteral(ast::ExprStringLiteral { range, value }) = annotation {
// Quoted annotations // Quoted annotations
if let Ok((parsed_annotation, _)) = if let Ok((parsed_annotation, _)) =
parse_type_annotation(value.as_str(), *range, checker.locator().contents()) parse_type_annotation(value.to_str(), *range, checker.locator().contents())
{ {
if type_hint_resolves_to_any( if type_hint_resolves_to_any(
&parsed_annotation, &parsed_annotation,

View file

@ -10,7 +10,7 @@ static PASSWORD_CANDIDATE_REGEX: Lazy<Regex> = Lazy::new(|| {
pub(super) fn string_literal(expr: &Expr) -> Option<&str> { pub(super) fn string_literal(expr: &Expr) -> Option<&str> {
match expr { match expr {
Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) => Some(value.as_str()), Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) => Some(value.to_str()),
_ => None, _ => None,
} }
} }

View file

@ -35,7 +35,7 @@ impl Violation for HardcodedBindAllInterfaces {
/// S104 /// S104
pub(crate) fn hardcoded_bind_all_interfaces(string: &ExprStringLiteral) -> Option<Diagnostic> { pub(crate) fn hardcoded_bind_all_interfaces(string: &ExprStringLiteral) -> Option<Diagnostic> {
if string.value.as_str() == "0.0.0.0" { if string.value.to_str() == "0.0.0.0" {
Some(Diagnostic::new(HardcodedBindAllInterfaces, string.range)) Some(Diagnostic::new(HardcodedBindAllInterfaces, string.range))
} else { } else {
None None

View file

@ -55,7 +55,7 @@ fn password_target(target: &Expr) -> Option<&str> {
Expr::Name(ast::ExprName { id, .. }) => id.as_str(), Expr::Name(ast::ExprName { id, .. }) => id.as_str(),
// d["password"] = "s3cr3t" // d["password"] = "s3cr3t"
Expr::Subscript(ast::ExprSubscript { slice, .. }) => match slice.as_ref() { Expr::Subscript(ast::ExprSubscript { slice, .. }) => match slice.as_ref() {
Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) => value.as_str(), Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) => value.to_str(),
_ => return None, _ => return None,
}, },
// obj.password = "s3cr3t" // obj.password = "s3cr3t"

View file

@ -93,7 +93,7 @@ pub(crate) fn hardcoded_sql_expression(checker: &mut Checker, expr: &Expr) {
let Some(string) = left.as_string_literal_expr() else { let Some(string) = left.as_string_literal_expr() else {
return; return;
}; };
string.value.as_str().escape_default().to_string() string.value.to_str().escape_default().to_string()
} }
Expr::Call(ast::ExprCall { func, .. }) => { Expr::Call(ast::ExprCall { func, .. }) => {
let Expr::Attribute(ast::ExprAttribute { attr, value, .. }) = func.as_ref() else { let Expr::Attribute(ast::ExprAttribute { attr, value, .. }) = func.as_ref() else {
@ -106,7 +106,7 @@ pub(crate) fn hardcoded_sql_expression(checker: &mut Checker, expr: &Expr) {
let Some(string) = value.as_string_literal_expr() else { let Some(string) = value.as_string_literal_expr() else {
return; return;
}; };
string.value.as_str().escape_default().to_string() string.value.to_str().escape_default().to_string()
} }
// f"select * from table where val = {val}" // f"select * from table where val = {val}"
Expr::FString(f_string) => concatenated_f_string(f_string, checker.locator()), Expr::FString(f_string) => concatenated_f_string(f_string, checker.locator()),

View file

@ -57,7 +57,7 @@ pub(crate) fn hardcoded_tmp_directory(checker: &mut Checker, string: &ast::ExprS
.flake8_bandit .flake8_bandit
.hardcoded_tmp_directory .hardcoded_tmp_directory
.iter() .iter()
.any(|prefix| string.value.as_str().starts_with(prefix)) .any(|prefix| string.value.to_str().starts_with(prefix))
{ {
return; return;
} }

View file

@ -855,7 +855,7 @@ pub(crate) fn suspicious_function_call(checker: &mut Checker, call: &ExprCall) {
// If the `url` argument is a string literal, allow `http` and `https` schemes. // If the `url` argument is a string literal, allow `http` and `https` schemes.
if call.arguments.args.iter().all(|arg| !arg.is_starred_expr()) && call.arguments.keywords.iter().all(|keyword| keyword.arg.is_some()) { if call.arguments.args.iter().all(|arg| !arg.is_starred_expr()) && call.arguments.keywords.iter().all(|keyword| keyword.arg.is_some()) {
if let Some(Expr::StringLiteral(ast::ExprStringLiteral { value, .. })) = &call.arguments.find_argument("url", 0) { if let Some(Expr::StringLiteral(ast::ExprStringLiteral { value, .. })) = &call.arguments.find_argument("url", 0) {
let url = value.as_str().trim_start(); let url = value.to_str().trim_start();
if url.starts_with("http://") || url.starts_with("https://") { if url.starts_with("http://") || url.starts_with("https://") {
return None; return None;
} }

View file

@ -60,7 +60,7 @@ pub(crate) fn tarfile_unsafe_members(checker: &mut Checker, call: &ast::ExprCall
.arguments .arguments
.find_keyword("filter") .find_keyword("filter")
.and_then(|keyword| keyword.value.as_string_literal_expr()) .and_then(|keyword| keyword.value.as_string_literal_expr())
.is_some_and(|value| matches!(value.value.as_str(), "data" | "tar")) .is_some_and(|value| matches!(value.value.to_str(), "data" | "tar"))
{ {
return; return;
} }

View file

@ -69,10 +69,10 @@ pub(crate) fn getattr_with_constant(
let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = arg else { let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = arg else {
return; return;
}; };
if !is_identifier(value.as_str()) { if !is_identifier(value.to_str()) {
return; return;
} }
if is_mangled_private(value.as_str()) { if is_mangled_private(value.to_str()) {
return; return;
} }
if !checker.semantic().is_builtin("getattr") { if !checker.semantic().is_builtin("getattr") {

View file

@ -83,10 +83,10 @@ pub(crate) fn setattr_with_constant(
let Expr::StringLiteral(ast::ExprStringLiteral { value: name, .. }) = name else { let Expr::StringLiteral(ast::ExprStringLiteral { value: name, .. }) = name else {
return; return;
}; };
if !is_identifier(name.as_str()) { if !is_identifier(name.to_str()) {
return; return;
} }
if is_mangled_private(name.as_str()) { if is_mangled_private(name.to_str()) {
return; return;
} }
if !checker.semantic().is_builtin("setattr") { if !checker.semantic().is_builtin("setattr") {
@ -104,7 +104,7 @@ pub(crate) fn setattr_with_constant(
if expr == child.as_ref() { if expr == child.as_ref() {
let mut diagnostic = Diagnostic::new(SetAttrWithConstant, expr.range()); let mut diagnostic = Diagnostic::new(SetAttrWithConstant, expr.range());
diagnostic.set_fix(Fix::safe_edit(Edit::range_replacement( diagnostic.set_fix(Fix::safe_edit(Edit::range_replacement(
assignment(obj, name.as_str(), value, checker.generator()), assignment(obj, name.to_str(), value, checker.generator()),
expr.range(), expr.range(),
))); )));
checker.diagnostics.push(diagnostic); checker.diagnostics.push(diagnostic);

View file

@ -78,7 +78,7 @@ pub(crate) fn call_datetime_strptime_without_zone(checker: &mut Checker, call: &
if let Some(Expr::StringLiteral(ast::ExprStringLiteral { value: format, .. })) = if let Some(Expr::StringLiteral(ast::ExprStringLiteral { value: format, .. })) =
call.arguments.args.get(1).as_ref() call.arguments.args.get(1).as_ref()
{ {
if format.as_str().contains("%z") { if format.to_str().contains("%z") {
return; return;
} }
}; };

View file

@ -93,7 +93,7 @@ fn check_log_record_attr_clash(checker: &mut Checker, extra: &Keyword) {
for key in keys { for key in keys {
if let Some(key) = &key { if let Some(key) = &key {
if let Expr::StringLiteral(ast::ExprStringLiteral { value: attr, .. }) = key { if let Expr::StringLiteral(ast::ExprStringLiteral { value: attr, .. }) = key {
if is_reserved_attr(attr.as_str()) { if is_reserved_attr(attr.to_str()) {
checker.diagnostics.push(Diagnostic::new( checker.diagnostics.push(Diagnostic::new(
LoggingExtraAttrClash(attr.to_string()), LoggingExtraAttrClash(attr.to_string()),
key.range(), key.range(),

View file

@ -110,8 +110,8 @@ pub(crate) fn unnecessary_dict_kwargs(checker: &mut Checker, expr: &Expr, kwargs
/// Return `Some` if a key is a valid keyword argument name, or `None` otherwise. /// Return `Some` if a key is a valid keyword argument name, or `None` otherwise.
fn as_kwarg(key: &Expr) -> Option<&str> { fn as_kwarg(key: &Expr) -> Option<&str> {
if let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = key { if let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = key {
if is_identifier(value.as_str()) { if is_identifier(value.to_str()) {
return Some(value.as_str()); return Some(value.to_str());
} }
} }
None None

View file

@ -127,7 +127,7 @@ pub(crate) fn unrecognized_platform(checker: &mut Checker, test: &Expr) {
// Other values are possible but we don't need them right now. // Other values are possible but we don't need them right now.
// This protects against typos. // This protects against typos.
if checker.enabled(Rule::UnrecognizedPlatformName) { if checker.enabled(Rule::UnrecognizedPlatformName) {
if !matches!(value.as_str(), "linux" | "win32" | "cygwin" | "darwin") { if !matches!(value.to_str(), "linux" | "win32" | "cygwin" | "darwin") {
checker.diagnostics.push(Diagnostic::new( checker.diagnostics.push(Diagnostic::new(
UnrecognizedPlatformName { UnrecognizedPlatformName {
platform: value.to_string(), platform: value.to_string(),

View file

@ -258,7 +258,7 @@ fn elts_to_csv(elts: &[Expr], generator: Generator) -> Option<String> {
if !acc.is_empty() { if !acc.is_empty() {
acc.push(','); acc.push(',');
} }
acc.push_str(value.as_str()); acc.push_str(value.to_str());
} }
acc acc
}), }),
@ -301,7 +301,7 @@ fn check_names(checker: &mut Checker, decorator: &Decorator, expr: &Expr) {
match expr { match expr {
Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) => { Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) => {
let names = split_names(value.as_str()); let names = split_names(value.to_str());
if names.len() > 1 { if names.len() > 1 {
match names_type { match names_type {
types::ParametrizeNameType::Tuple => { types::ParametrizeNameType::Tuple => {
@ -476,7 +476,7 @@ fn check_values(checker: &mut Checker, names: &Expr, values: &Expr) {
.parametrize_values_row_type; .parametrize_values_row_type;
let is_multi_named = if let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = &names { let is_multi_named = if let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = &names {
split_names(value.as_str()).len() > 1 split_names(value.to_str()).len() > 1
} else { } else {
true true
}; };

View file

@ -161,12 +161,12 @@ pub(crate) fn use_capital_environment_variables(checker: &mut Checker, expr: &Ex
return; return;
} }
if is_lowercase_allowed(env_var.as_str()) { if is_lowercase_allowed(env_var.to_str()) {
return; return;
} }
let capital_env_var = env_var.as_str().to_ascii_uppercase(); let capital_env_var = env_var.to_str().to_ascii_uppercase();
if capital_env_var == env_var.as_str() { if capital_env_var == env_var.to_str() {
return; return;
} }
@ -201,12 +201,12 @@ fn check_os_environ_subscript(checker: &mut Checker, expr: &Expr) {
return; return;
}; };
if is_lowercase_allowed(env_var.as_str()) { if is_lowercase_allowed(env_var.to_str()) {
return; return;
} }
let capital_env_var = env_var.as_str().to_ascii_uppercase(); let capital_env_var = env_var.to_str().to_ascii_uppercase();
if capital_env_var == env_var.as_str() { if capital_env_var == env_var.to_str() {
return; return;
} }

View file

@ -69,7 +69,7 @@ pub(crate) fn path_constructor_current_directory(checker: &mut Checker, expr: &E
return; return;
}; };
if matches!(value.as_str(), "" | ".") { if matches!(value.to_str(), "" | ".") {
let mut diagnostic = Diagnostic::new(PathConstructorCurrentDirectory, *range); let mut diagnostic = Diagnostic::new(PathConstructorCurrentDirectory, *range);
diagnostic.set_fix(Fix::safe_edit(Edit::range_deletion(*range))); diagnostic.set_fix(Fix::safe_edit(Edit::range_deletion(*range)));
checker.diagnostics.push(diagnostic); checker.diagnostics.push(diagnostic);

View file

@ -67,7 +67,7 @@ fn build_fstring(joiner: &str, joinees: &[Expr]) -> Option<Expr> {
.iter() .iter()
.filter_map(|expr| { .filter_map(|expr| {
if let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = expr { if let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = expr {
Some(value.as_str()) Some(value.to_str())
} else { } else {
None None
} }

View file

@ -583,10 +583,10 @@ pub(crate) fn percent_format_extra_named_arguments(
.enumerate() .enumerate()
.filter_map(|(index, key)| match key { .filter_map(|(index, key)| match key {
Some(Expr::StringLiteral(ast::ExprStringLiteral { value, .. })) => { Some(Expr::StringLiteral(ast::ExprStringLiteral { value, .. })) => {
if summary.keywords.contains(value.as_str()) { if summary.keywords.contains(value.to_str()) {
None None
} else { } else {
Some((index, value.as_str())) Some((index, value.to_str()))
} }
} }
_ => None, _ => None,
@ -641,7 +641,7 @@ pub(crate) fn percent_format_missing_arguments(
for key in keys.iter().flatten() { for key in keys.iter().flatten() {
match key { match key {
Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) => { Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) => {
keywords.insert(value.as_str()); keywords.insert(value.to_str());
} }
_ => { _ => {
return; // Dynamic keys present return; // Dynamic keys present

View file

@ -12,7 +12,7 @@ pub(super) fn type_param_name(arguments: &Arguments) -> Option<&str> {
// Handle both `TypeVar("T")` and `TypeVar(name="T")`. // Handle both `TypeVar("T")` and `TypeVar(name="T")`.
let name_param = arguments.find_argument("name", 0)?; let name_param = arguments.find_argument("name", 0)?;
if let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = &name_param { if let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = &name_param {
Some(value.as_str()) Some(value.to_str())
} else { } else {
None None
} }

View file

@ -201,7 +201,7 @@ fn is_valid_dict(
value: mapping_key, .. value: mapping_key, ..
}) = key }) = key
{ {
let Some(format) = formats_hash.get(mapping_key.as_str()) else { let Some(format) = formats_hash.get(mapping_key.to_str()) else {
return true; return true;
}; };
if !equivalent(format, value) { if !equivalent(format, value) {

View file

@ -134,7 +134,7 @@ pub(crate) fn logging_call(checker: &mut Checker, call: &ast::ExprCall) {
return; return;
}; };
let Ok(summary) = CFormatSummary::try_from(value.as_str()) else { let Ok(summary) = CFormatSummary::try_from(value.to_str()) else {
return; return;
}; };

View file

@ -83,7 +83,7 @@ fn is_magic_value(literal_expr: LiteralExpressionRef, allowed_types: &[ConstantT
| LiteralExpressionRef::EllipsisLiteral(_) => false, | LiteralExpressionRef::EllipsisLiteral(_) => false,
// Special-case some common string and integer types. // Special-case some common string and integer types.
LiteralExpressionRef::StringLiteral(ast::ExprStringLiteral { value, .. }) => { LiteralExpressionRef::StringLiteral(ast::ExprStringLiteral { value, .. }) => {
!matches!(value.as_str(), "" | "__main__") !matches!(value.to_str(), "" | "__main__")
} }
LiteralExpressionRef::NumberLiteral(ast::ExprNumberLiteral { value, .. }) => match value { LiteralExpressionRef::NumberLiteral(ast::ExprNumberLiteral { value, .. }) => match value {
ast::Number::Int(value) => !matches!(*value, Int::ZERO | Int::ONE), ast::Number::Int(value) => !matches!(*value, Int::ZERO | Int::ONE),

View file

@ -60,7 +60,7 @@ pub(crate) fn repeated_keyword_argument(checker: &mut Checker, call: &ExprCall)
// Ex) `func(**{"a": 1, "a": 2})` // Ex) `func(**{"a": 1, "a": 2})`
for key in keys.iter().flatten() { for key in keys.iter().flatten() {
if let Expr::StringLiteral(ExprStringLiteral { value, .. }) = key { if let Expr::StringLiteral(ExprStringLiteral { value, .. }) = key {
if !seen.insert(value.as_str()) { if !seen.insert(value.to_str()) {
checker.diagnostics.push(Diagnostic::new( checker.diagnostics.push(Diagnostic::new(
RepeatedKeywordArgument { RepeatedKeywordArgument {
duplicate_keyword: value.to_string(), duplicate_keyword: value.to_string(),

View file

@ -185,13 +185,13 @@ fn create_fields_from_fields_arg(fields: &Expr) -> Option<Vec<Stmt>> {
return None; return None;
} }
let ast::ExprStringLiteral { value: field, .. } = field.as_string_literal_expr()?; let ast::ExprStringLiteral { value: field, .. } = field.as_string_literal_expr()?;
if !is_identifier(field.as_str()) { if !is_identifier(field.to_str()) {
return None; return None;
} }
if is_dunder(field.as_str()) { if is_dunder(field.to_str()) {
return None; return None;
} }
Some(create_field_assignment_stmt(field.as_str(), annotation)) Some(create_field_assignment_stmt(field.to_str(), annotation))
}) })
.collect() .collect()
} }

View file

@ -174,13 +174,13 @@ fn fields_from_dict_literal(keys: &[Option<Expr>], values: &[Expr]) -> Option<Ve
.zip(values.iter()) .zip(values.iter())
.map(|(key, value)| match key { .map(|(key, value)| match key {
Some(Expr::StringLiteral(ast::ExprStringLiteral { value: field, .. })) => { Some(Expr::StringLiteral(ast::ExprStringLiteral { value: field, .. })) => {
if !is_identifier(field.as_str()) { if !is_identifier(field.to_str()) {
return None; return None;
} }
if is_dunder(field.as_str()) { if is_dunder(field.to_str()) {
return None; return None;
} }
Some(create_field_assignment_stmt(field.as_str(), value)) Some(create_field_assignment_stmt(field.to_str(), value))
} }
_ => None, _ => None,
}) })

View file

@ -228,14 +228,14 @@ fn clean_params_dictionary(right: &Expr, locator: &Locator, stylist: &Stylist) -
}) = key }) = key
{ {
// If the dictionary key is not a valid variable name, abort. // If the dictionary key is not a valid variable name, abort.
if !is_identifier(key_string.as_str()) { if !is_identifier(key_string.to_str()) {
return None; return None;
} }
// If there are multiple entries of the same key, abort. // If there are multiple entries of the same key, abort.
if seen.contains(&key_string.as_str()) { if seen.contains(&key_string.to_str()) {
return None; return None;
} }
seen.push(key_string.as_str()); seen.push(key_string.to_str());
if is_multi_line { if is_multi_line {
if indent.is_none() { if indent.is_none() {
indent = indentation(locator, key); indent = indentation(locator, key);

View file

@ -76,7 +76,7 @@ pub(crate) fn redundant_open_modes(checker: &mut Checker, call: &ast::ExprCall)
.. ..
}) = &keyword.value }) = &keyword.value
{ {
if let Ok(mode) = OpenMode::from_str(mode_param_value.as_str()) { if let Ok(mode) = OpenMode::from_str(mode_param_value.to_str()) {
checker.diagnostics.push(create_check( checker.diagnostics.push(create_check(
call, call,
&keyword.value, &keyword.value,
@ -91,7 +91,7 @@ pub(crate) fn redundant_open_modes(checker: &mut Checker, call: &ast::ExprCall)
} }
Some(mode_param) => { Some(mode_param) => {
if let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = &mode_param { if let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = &mode_param {
if let Ok(mode) = OpenMode::from_str(value.as_str()) { if let Ok(mode) = OpenMode::from_str(value.to_str()) {
checker.diagnostics.push(create_check( checker.diagnostics.push(create_check(
call, call,
mode_param, mode_param,

View file

@ -74,7 +74,7 @@ fn match_encoded_variable(func: &Expr) -> Option<&Expr> {
fn is_utf8_encoding_arg(arg: &Expr) -> bool { fn is_utf8_encoding_arg(arg: &Expr) -> bool {
if let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = &arg { if let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = &arg {
UTF8_LITERALS.contains(&value.as_str().to_lowercase().as_str()) UTF8_LITERALS.contains(&value.to_str().to_lowercase().as_str())
} else { } else {
false false
} }
@ -161,7 +161,7 @@ pub(crate) fn unnecessary_encode_utf8(checker: &mut Checker, call: &ast::ExprCal
Expr::StringLiteral(ast::ExprStringLiteral { value: literal, .. }) => { Expr::StringLiteral(ast::ExprStringLiteral { value: literal, .. }) => {
// Ex) `"str".encode()`, `"str".encode("utf-8")` // Ex) `"str".encode()`, `"str".encode("utf-8")`
if let Some(encoding_arg) = match_encoding_arg(&call.arguments) { if let Some(encoding_arg) = match_encoding_arg(&call.arguments) {
if literal.as_str().is_ascii() { if literal.to_str().is_ascii() {
// Ex) Convert `"foo".encode()` to `b"foo"`. // Ex) Convert `"foo".encode()` to `b"foo"`.
let mut diagnostic = Diagnostic::new( let mut diagnostic = Diagnostic::new(
UnnecessaryEncodeUTF8 { UnnecessaryEncodeUTF8 {

View file

@ -66,7 +66,7 @@ pub(crate) fn no_implicit_cwd(checker: &mut Checker, call: &ExprCall) {
let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = arg else { let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = arg else {
return; return;
}; };
if !matches!(value.as_str(), "" | ".") { if !matches!(value.to_str(), "" | ".") {
return; return;
} }
} }

View file

@ -248,7 +248,7 @@ fn match_open_mode(mode: &Expr) -> Option<ReadMode> {
if value.is_implicit_concatenated() { if value.is_implicit_concatenated() {
return None; return None;
} }
match value.as_str() { match value.to_str() {
"r" => Some(ReadMode::Text), "r" => Some(ReadMode::Text),
"rb" => Some(ReadMode::Bytes), "rb" => Some(ReadMode::Bytes),
_ => None, _ => None,

View file

@ -184,7 +184,7 @@ pub(crate) fn implicit_optional(checker: &mut Checker, parameters: &Parameters)
if let Expr::StringLiteral(ast::ExprStringLiteral { range, value }) = annotation.as_ref() { if let Expr::StringLiteral(ast::ExprStringLiteral { range, value }) = annotation.as_ref() {
// Quoted annotation. // Quoted annotation.
if let Ok((annotation, kind)) = if let Ok((annotation, kind)) =
parse_type_annotation(value.as_str(), *range, checker.locator().contents()) parse_type_annotation(value.to_str(), *range, checker.locator().contents())
{ {
let Some(expr) = type_hint_explicitly_allows_none( let Some(expr) = type_hint_explicitly_allows_none(
&annotation, &annotation,

View file

@ -109,7 +109,7 @@ impl<'a> TypingTarget<'a> {
}) => Some(TypingTarget::PEP604Union(left, right)), }) => Some(TypingTarget::PEP604Union(left, right)),
Expr::NoneLiteral(_) => Some(TypingTarget::None), Expr::NoneLiteral(_) => Some(TypingTarget::None),
Expr::StringLiteral(ast::ExprStringLiteral { value, range }) => { Expr::StringLiteral(ast::ExprStringLiteral { value, range }) => {
parse_type_annotation(value.as_str(), *range, locator.contents()) parse_type_annotation(value.to_str(), *range, locator.contents())
.map_or(None, |(expr, _)| Some(TypingTarget::ForwardReference(expr))) .map_or(None, |(expr, _)| Some(TypingTarget::ForwardReference(expr)))
} }
_ => semantic.resolve_call_path(expr).map_or( _ => semantic.resolve_call_path(expr).map_or(

View file

@ -24,7 +24,7 @@ where
fn add_to_names<'a>(elts: &'a [Expr], names: &mut Vec<&'a str>, flags: &mut DunderAllFlags) { fn add_to_names<'a>(elts: &'a [Expr], names: &mut Vec<&'a str>, flags: &mut DunderAllFlags) {
for elt in elts { for elt in elts {
if let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = elt { if let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = elt {
names.push(value.as_str()); names.push(value.to_str());
} else { } else {
*flags |= DunderAllFlags::INVALID_OBJECT; *flags |= DunderAllFlags::INVALID_OBJECT;
} }

View file

@ -1233,10 +1233,10 @@ impl StringLiteralValue {
/// ///
/// Note that this will perform an allocation on the first invocation if the /// Note that this will perform an allocation on the first invocation if the
/// string value is implicitly concatenated. /// string value is implicitly concatenated.
pub fn as_str(&self) -> &str { pub fn to_str(&self) -> &str {
match &self.inner { match &self.inner {
StringLiteralValueInner::Single(value) => value.as_str(), StringLiteralValueInner::Single(value) => value.as_str(),
StringLiteralValueInner::Concatenated(value) => value.as_str(), StringLiteralValueInner::Concatenated(value) => value.to_str(),
} }
} }
} }
@ -1259,7 +1259,7 @@ impl PartialEq<String> for StringLiteralValue {
impl fmt::Display for StringLiteralValue { impl fmt::Display for StringLiteralValue {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.write_str(self.as_str()) f.write_str(self.to_str())
} }
} }
@ -1331,7 +1331,7 @@ struct ConcatenatedStringLiteral {
impl ConcatenatedStringLiteral { impl ConcatenatedStringLiteral {
/// Extracts a string slice containing the entire concatenated string. /// Extracts a string slice containing the entire concatenated string.
fn as_str(&self) -> &str { fn to_str(&self) -> &str {
self.value self.value
.get_or_init(|| self.strings.iter().map(StringLiteral::as_str).collect()) .get_or_init(|| self.strings.iter().map(StringLiteral::as_str).collect())
} }
@ -1354,7 +1354,7 @@ impl Debug for ConcatenatedStringLiteral {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("ConcatenatedStringLiteral") f.debug_struct("ConcatenatedStringLiteral")
.field("strings", &self.strings) .field("strings", &self.strings)
.field("value", &self.as_str()) .field("value", &self.to_str())
.finish() .finish()
} }
} }

View file

@ -1356,7 +1356,7 @@ impl<'a> Generator<'a> {
fn unparse_f_string_elem(&mut self, expr: &Expr, is_spec: bool) { fn unparse_f_string_elem(&mut self, expr: &Expr, is_spec: bool) {
match expr { match expr {
Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) => { Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) => {
self.unparse_f_string_literal(value.as_str()); self.unparse_f_string_literal(value.to_str());
} }
Expr::FString(ast::ExprFString { value, .. }) => { Expr::FString(ast::ExprFString { value, .. }) => {
self.unparse_f_string_value(value, is_spec); self.unparse_f_string_value(value, is_spec);