from __future__ import annotations import ast from dataclasses import dataclass from typing import Iterable from flake8_pie.base import Body, Error, Flake8Error from flake8_pie.pie781_assign_and_return import pie781_assign_and_return from flake8_pie.pie783_celery_explicit_names import pie783_celery_explicit_names from flake8_pie.pie784_celery_crontab_args import pie784_celery_crontab_args from flake8_pie.pie785_celery_require_tasks_expire import ( pie785_celery_require_tasks_expire, ) from flake8_pie.pie786_precise_exception_handler import pie786_precise_exception_handler from flake8_pie.pie787_no_len_condition import pie787_no_len_condition from flake8_pie.pie788_no_bool_condition import pie788_no_bool_condition from flake8_pie.pie789_prefer_isinstance_type_compare import ( pie789_prefer_isinstance_type_compare, ) from flake8_pie.pie790_no_unnecessary_pass import pie790_no_unnecessary_pass from flake8_pie.pie791_no_pointless_statements import pie791_no_pointless_statements from flake8_pie.pie792_no_inherit_object import pie792_no_inherit_object from flake8_pie.pie793_prefer_dataclass import pie793_prefer_dataclass from flake8_pie.pie794_dupe_class_field_definitions import ( pie794_dupe_class_field_definition, ) from flake8_pie.pie795_prefer_stdlib_enums import pie795_prefer_stdlib_enums from flake8_pie.pie796_prefer_unique_enums import pie786_prefer_unique_enum from flake8_pie.pie797_no_unnecessary_if_expr import pie797_no_unnecessary_if_expr from flake8_pie.pie798_no_unnecessary_class import pie798_no_unnecessary_class from flake8_pie.pie799_prefer_col_init import pie799_prefer_col_init from flake8_pie.pie800_no_unnecessary_spread import pie800_no_unnecessary_spread from flake8_pie.pie801_prefer_simple_return import pie801_prefer_simple_return from flake8_pie.pie802_prefer_simple_any_all import pie802_prefer_simple_any_all from flake8_pie.pie803_prefer_logging_interpolation import ( pie803_prefer_logging_interpolation, ) from flake8_pie.pie804_no_unnecessary_dict_kwargs import pie804_no_dict_kwargs from flake8_pie.pie805_prefer_literal import pie805_prefer_literal from flake8_pie.pie806_no_assert_except import pie806_no_assert_except from flake8_pie.pie807_pefer_list_builtin import pie807_prefer_list_builtin from flake8_pie.pie808_prefer_simple_range import pie808_prefer_simple_range from flake8_pie.pie809_django_prefer_bulk import pie809_django_prefer_bulk @dataclass(frozen=True) class BodyNode: body: list[ast.stmt] class Flake8PieVisitor(ast.NodeVisitor): def __init__(self, filename: str) -> None: self.errors: list[Error] = [] self.filename = filename self.inside_inheriting_cls_stack: list[bool] = [] def visit_FunctionDef(self, node: ast.FunctionDef) -> None: pie783_celery_explicit_names(node, self.errors) self._visit_body(node) self.generic_visit(node) def visit_GeneratorExp(self, node: ast.GeneratorExp) -> None: pie809_django_prefer_bulk(node, self.errors) self.generic_visit(node) def visit_AsyncFunctionDef(self, node: ast.AsyncFunctionDef) -> None: self._visit_body(node) self.generic_visit(node) def visit_For(self, node: ast.For) -> None: self._visit_body(node) self._visit_body(BodyNode(node.orelse)) self.generic_visit(node) def visit_AsyncFor(self, node: ast.AsyncFor) -> None: self._visit_body(node) self._visit_body(BodyNode(node.orelse)) self.generic_visit(node) def visit_While(self, node: ast.While) -> None: self._visit_body(node) self._visit_body(BodyNode(node.orelse)) self.generic_visit(node) def visit_ClassDef(self, node: ast.ClassDef) -> None: pie794_dupe_class_field_definition(node, self.errors) pie792_no_inherit_object(node, self.errors) pie793_prefer_dataclass(node, self.errors, self.inside_inheriting_cls_stack) pie795_prefer_stdlib_enums(node, self.errors, self.inside_inheriting_cls_stack) pie786_prefer_unique_enum(node, self.errors) pie798_no_unnecessary_class(node, self.errors) self._visit_body(node) is_inheriting_cls = len(node.bases) > 0 self.inside_inheriting_cls_stack.append(is_inheriting_cls) self.generic_visit(node) self.inside_inheriting_cls_stack.pop() def visit_Call(self, node: ast.Call) -> None: pie784_celery_crontab_args(node, self.errors) pie785_celery_require_tasks_expire(node, self.errors) pie802_prefer_simple_any_all(node, self.errors) pie803_prefer_logging_interpolation(node, self.errors) pie804_no_dict_kwargs(node, self.errors) pie805_prefer_literal(node, self.errors) pie808_prefer_simple_range(node, self.errors) self.generic_visit(node) def visit_Dict(self, node: ast.Dict) -> None: pie785_celery_require_tasks_expire(node, self.errors) pie800_no_unnecessary_spread(node, self.errors) self.generic_visit(node) def visit_ExceptHandler(self, node: ast.ExceptHandler) -> None: pie786_precise_exception_handler(node, self.errors) self._visit_body(node) self.generic_visit(node) def visit_Expr(self, node: ast.Expr) -> None: pie791_no_pointless_statements(node, self.errors) self.generic_visit(node) def visit_If(self, node: ast.If) -> None: pie787_no_len_condition(node, self.errors) pie789_prefer_isinstance_type_compare(node, self.errors) pie788_no_bool_condition(node, self.errors) self._visit_body(node) self._visit_body(BodyNode(node.orelse)) self.generic_visit(node) def visit_With(self, node: ast.With) -> None: self._visit_body(node) self.generic_visit(node) def visit_AsyncWith(self, node: ast.AsyncWith) -> None: self._visit_body(node) self.generic_visit(node) def visit_Try(self, node: ast.Try) -> None: self._visit_body(node) pie806_no_assert_except(node, self.errors) self.generic_visit(node) def visit_IfExp(self, node: ast.IfExp) -> None: pie787_no_len_condition(node, self.errors) pie788_no_bool_condition(node, self.errors) pie789_prefer_isinstance_type_compare(node, self.errors) pie797_no_unnecessary_if_expr(node, self.errors) self.generic_visit(node) def visit_Lambda(self, node: ast.Lambda) -> None: pie807_prefer_list_builtin(node, self.errors) self.generic_visit(node) def visit_ListComp(self, node: ast.ListComp) -> None: pie809_django_prefer_bulk(node, self.errors) self.generic_visit(node) def _visit_body(self, node: Body) -> None: pie781_assign_and_return(node, self.errors) pie790_no_unnecessary_pass(node, self.errors) pie799_prefer_col_init(node, self.errors) pie801_prefer_simple_return(node, self.errors) def visit_Module(self, node: ast.Module) -> None: self._visit_body(node) self.generic_visit(node) def __repr__(self) -> str: return f"<{self.__class__.__name__}: errors={self.errors}>" class Flake8PieCheck: name = "flake8-pie" version = "0.15.0" def __init__( self, tree: ast.Module, filename: str, *args: object, **kwargs: object ) -> None: self.filename = filename self.tree = tree def run(self) -> Iterable[Flake8Error]: # When using flake8-pyi, skip the stub files. if self.filename.endswith(".pyi"): return visitor = Flake8PieVisitor(self.filename) visitor.visit(self.tree) for err in visitor.errors: yield Flake8Error( message=err.message, type=Flake8PieCheck, lineno=err.lineno, col_offset=err.col_offset, )