Split Constant to individual literal nodes (#8064)

## Summary

This PR splits the `Constant` enum as individual literal nodes. It
introduces the following new nodes for each variant:
* `ExprStringLiteral`
* `ExprBytesLiteral`
* `ExprNumberLiteral`
* `ExprBooleanLiteral`
* `ExprNoneLiteral`
* `ExprEllipsisLiteral`

The main motivation behind this refactor is to introduce the new AST
node for implicit string concatenation in the coming PR. The elements of
that node will be either a string literal, bytes literal or a f-string
which can be implemented using an enum. This means that a string or
bytes literal cannot be represented by `Constant::Str` /
`Constant::Bytes` which creates an inconsistency.

This PR avoids that inconsistency by splitting the constant nodes into
it's own literal nodes, literal being the more appropriate naming
convention from a static analysis tool perspective.

This also makes working with literals in the linter and formatter much
more ergonomic like, for example, if one would want to check if this is
a string literal, it can be done easily using
`Expr::is_string_literal_expr` or matching against `Expr::StringLiteral`
as oppose to matching against the `ExprConstant` and enum `Constant`. A
few AST helper methods can be simplified as well which will be done in a
follow-up PR.

This introduces a new `Expr::is_literal_expr` method which is the same
as `Expr::is_constant_expr`. There are also intermediary changes related
to implicit string concatenation which are quiet less. This is done so
as to avoid having a huge PR which this already is.

## Test Plan

1. Verify and update all of the existing snapshots (parser, visitor)
2. Verify that the ecosystem check output remains **unchanged** for both
the linter and formatter

### Formatter ecosystem check

#### `main`

| project | similarity index | total files | changed files |

|----------------|------------------:|------------------:|------------------:|
| cpython | 0.75803 | 1799 | 1647 |
| django | 0.99983 | 2772 | 34 |
| home-assistant | 0.99953 | 10596 | 186 |
| poetry | 0.99891 | 317 | 17 |
| transformers | 0.99966 | 2657 | 330 |
| twine | 1.00000 | 33 | 0 |
| typeshed | 0.99978 | 3669 | 20 |
| warehouse | 0.99977 | 654 | 13 |
| zulip | 0.99970 | 1459 | 22 |

#### `dhruv/constant-to-literal`

| project | similarity index | total files | changed files |

|----------------|------------------:|------------------:|------------------:|
| cpython | 0.75803 | 1799 | 1647 |
| django | 0.99983 | 2772 | 34 |
| home-assistant | 0.99953 | 10596 | 186 |
| poetry | 0.99891 | 317 | 17 |
| transformers | 0.99966 | 2657 | 330 |
| twine | 1.00000 | 33 | 0 |
| typeshed | 0.99978 | 3669 | 20 |
| warehouse | 0.99977 | 654 | 13 |
| zulip | 0.99970 | 1459 | 22 |
This commit is contained in:
Dhruv Manilawala 2023-10-30 12:13:23 +05:30 committed by GitHub
parent 78bbf6d403
commit 230c9ce236
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
268 changed files with 6663 additions and 6741 deletions

View file

@ -1606,38 +1606,218 @@ impl<'ast> IntoFormat<PyFormatContext<'ast>> for ast::ExprFString {
}
}
impl FormatRule<ast::ExprConstant, PyFormatContext<'_>>
for crate::expression::expr_constant::FormatExprConstant
impl FormatRule<ast::ExprStringLiteral, PyFormatContext<'_>>
for crate::expression::expr_string_literal::FormatExprStringLiteral
{
#[inline]
fn fmt(&self, node: &ast::ExprConstant, f: &mut PyFormatter) -> FormatResult<()> {
FormatNodeRule::<ast::ExprConstant>::fmt(self, node, f)
fn fmt(&self, node: &ast::ExprStringLiteral, f: &mut PyFormatter) -> FormatResult<()> {
FormatNodeRule::<ast::ExprStringLiteral>::fmt(self, node, f)
}
}
impl<'ast> AsFormat<PyFormatContext<'ast>> for ast::ExprConstant {
impl<'ast> AsFormat<PyFormatContext<'ast>> for ast::ExprStringLiteral {
type Format<'a> = FormatRefWithRule<
'a,
ast::ExprConstant,
crate::expression::expr_constant::FormatExprConstant,
ast::ExprStringLiteral,
crate::expression::expr_string_literal::FormatExprStringLiteral,
PyFormatContext<'ast>,
>;
fn format(&self) -> Self::Format<'_> {
FormatRefWithRule::new(
self,
crate::expression::expr_constant::FormatExprConstant::default(),
crate::expression::expr_string_literal::FormatExprStringLiteral::default(),
)
}
}
impl<'ast> IntoFormat<PyFormatContext<'ast>> for ast::ExprConstant {
impl<'ast> IntoFormat<PyFormatContext<'ast>> for ast::ExprStringLiteral {
type Format = FormatOwnedWithRule<
ast::ExprConstant,
crate::expression::expr_constant::FormatExprConstant,
ast::ExprStringLiteral,
crate::expression::expr_string_literal::FormatExprStringLiteral,
PyFormatContext<'ast>,
>;
fn into_format(self) -> Self::Format {
FormatOwnedWithRule::new(
self,
crate::expression::expr_constant::FormatExprConstant::default(),
crate::expression::expr_string_literal::FormatExprStringLiteral::default(),
)
}
}
impl FormatRule<ast::ExprBytesLiteral, PyFormatContext<'_>>
for crate::expression::expr_bytes_literal::FormatExprBytesLiteral
{
#[inline]
fn fmt(&self, node: &ast::ExprBytesLiteral, f: &mut PyFormatter) -> FormatResult<()> {
FormatNodeRule::<ast::ExprBytesLiteral>::fmt(self, node, f)
}
}
impl<'ast> AsFormat<PyFormatContext<'ast>> for ast::ExprBytesLiteral {
type Format<'a> = FormatRefWithRule<
'a,
ast::ExprBytesLiteral,
crate::expression::expr_bytes_literal::FormatExprBytesLiteral,
PyFormatContext<'ast>,
>;
fn format(&self) -> Self::Format<'_> {
FormatRefWithRule::new(
self,
crate::expression::expr_bytes_literal::FormatExprBytesLiteral::default(),
)
}
}
impl<'ast> IntoFormat<PyFormatContext<'ast>> for ast::ExprBytesLiteral {
type Format = FormatOwnedWithRule<
ast::ExprBytesLiteral,
crate::expression::expr_bytes_literal::FormatExprBytesLiteral,
PyFormatContext<'ast>,
>;
fn into_format(self) -> Self::Format {
FormatOwnedWithRule::new(
self,
crate::expression::expr_bytes_literal::FormatExprBytesLiteral::default(),
)
}
}
impl FormatRule<ast::ExprNumberLiteral, PyFormatContext<'_>>
for crate::expression::expr_number_literal::FormatExprNumberLiteral
{
#[inline]
fn fmt(&self, node: &ast::ExprNumberLiteral, f: &mut PyFormatter) -> FormatResult<()> {
FormatNodeRule::<ast::ExprNumberLiteral>::fmt(self, node, f)
}
}
impl<'ast> AsFormat<PyFormatContext<'ast>> for ast::ExprNumberLiteral {
type Format<'a> = FormatRefWithRule<
'a,
ast::ExprNumberLiteral,
crate::expression::expr_number_literal::FormatExprNumberLiteral,
PyFormatContext<'ast>,
>;
fn format(&self) -> Self::Format<'_> {
FormatRefWithRule::new(
self,
crate::expression::expr_number_literal::FormatExprNumberLiteral::default(),
)
}
}
impl<'ast> IntoFormat<PyFormatContext<'ast>> for ast::ExprNumberLiteral {
type Format = FormatOwnedWithRule<
ast::ExprNumberLiteral,
crate::expression::expr_number_literal::FormatExprNumberLiteral,
PyFormatContext<'ast>,
>;
fn into_format(self) -> Self::Format {
FormatOwnedWithRule::new(
self,
crate::expression::expr_number_literal::FormatExprNumberLiteral::default(),
)
}
}
impl FormatRule<ast::ExprBooleanLiteral, PyFormatContext<'_>>
for crate::expression::expr_boolean_literal::FormatExprBooleanLiteral
{
#[inline]
fn fmt(&self, node: &ast::ExprBooleanLiteral, f: &mut PyFormatter) -> FormatResult<()> {
FormatNodeRule::<ast::ExprBooleanLiteral>::fmt(self, node, f)
}
}
impl<'ast> AsFormat<PyFormatContext<'ast>> for ast::ExprBooleanLiteral {
type Format<'a> = FormatRefWithRule<
'a,
ast::ExprBooleanLiteral,
crate::expression::expr_boolean_literal::FormatExprBooleanLiteral,
PyFormatContext<'ast>,
>;
fn format(&self) -> Self::Format<'_> {
FormatRefWithRule::new(
self,
crate::expression::expr_boolean_literal::FormatExprBooleanLiteral::default(),
)
}
}
impl<'ast> IntoFormat<PyFormatContext<'ast>> for ast::ExprBooleanLiteral {
type Format = FormatOwnedWithRule<
ast::ExprBooleanLiteral,
crate::expression::expr_boolean_literal::FormatExprBooleanLiteral,
PyFormatContext<'ast>,
>;
fn into_format(self) -> Self::Format {
FormatOwnedWithRule::new(
self,
crate::expression::expr_boolean_literal::FormatExprBooleanLiteral::default(),
)
}
}
impl FormatRule<ast::ExprNoneLiteral, PyFormatContext<'_>>
for crate::expression::expr_none_literal::FormatExprNoneLiteral
{
#[inline]
fn fmt(&self, node: &ast::ExprNoneLiteral, f: &mut PyFormatter) -> FormatResult<()> {
FormatNodeRule::<ast::ExprNoneLiteral>::fmt(self, node, f)
}
}
impl<'ast> AsFormat<PyFormatContext<'ast>> for ast::ExprNoneLiteral {
type Format<'a> = FormatRefWithRule<
'a,
ast::ExprNoneLiteral,
crate::expression::expr_none_literal::FormatExprNoneLiteral,
PyFormatContext<'ast>,
>;
fn format(&self) -> Self::Format<'_> {
FormatRefWithRule::new(
self,
crate::expression::expr_none_literal::FormatExprNoneLiteral::default(),
)
}
}
impl<'ast> IntoFormat<PyFormatContext<'ast>> for ast::ExprNoneLiteral {
type Format = FormatOwnedWithRule<
ast::ExprNoneLiteral,
crate::expression::expr_none_literal::FormatExprNoneLiteral,
PyFormatContext<'ast>,
>;
fn into_format(self) -> Self::Format {
FormatOwnedWithRule::new(
self,
crate::expression::expr_none_literal::FormatExprNoneLiteral::default(),
)
}
}
impl FormatRule<ast::ExprEllipsisLiteral, PyFormatContext<'_>>
for crate::expression::expr_ellipsis_literal::FormatExprEllipsisLiteral
{
#[inline]
fn fmt(&self, node: &ast::ExprEllipsisLiteral, f: &mut PyFormatter) -> FormatResult<()> {
FormatNodeRule::<ast::ExprEllipsisLiteral>::fmt(self, node, f)
}
}
impl<'ast> AsFormat<PyFormatContext<'ast>> for ast::ExprEllipsisLiteral {
type Format<'a> = FormatRefWithRule<
'a,
ast::ExprEllipsisLiteral,
crate::expression::expr_ellipsis_literal::FormatExprEllipsisLiteral,
PyFormatContext<'ast>,
>;
fn format(&self) -> Self::Format<'_> {
FormatRefWithRule::new(
self,
crate::expression::expr_ellipsis_literal::FormatExprEllipsisLiteral::default(),
)
}
}
impl<'ast> IntoFormat<PyFormatContext<'ast>> for ast::ExprEllipsisLiteral {
type Format = FormatOwnedWithRule<
ast::ExprEllipsisLiteral,
crate::expression::expr_ellipsis_literal::FormatExprEllipsisLiteral,
PyFormatContext<'ast>,
>;
fn into_format(self) -> Self::Format {
FormatOwnedWithRule::new(
self,
crate::expression::expr_ellipsis_literal::FormatExprEllipsisLiteral::default(),
)
}
}