style: run ruff format on new rule files

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: themohitkhare

src/pycodegate/rules/celery.py

@@ -157,7 +157,9 @@ def _check_broad_autoretry(self, func_node, call: ast.Call, filename: str) -> li
     # ------------------------------------------------------------------
     # Rule: celery-direct-call
     # ------------------------------------------------------------------
-    def _check_direct_call(self, node: ast.Call, task_names: set[str], filename: str) -> list[Diagnostic]:
+    def _check_direct_call(
+        self, node: ast.Call, task_names: set[str], filename: str
+    ) -> list[Diagnostic]:
         if isinstance(node.func, ast.Name) and node.func.id in task_names:
             return [
                 Diagnostic(

src/pycodegate/rules/numpy_rules.py

@@ -53,9 +53,7 @@ def _compare_has_array(self, node: ast.Compare) -> bool:
                     return True
         return False
 
-    def _check_array_equality(
-        self, node: ast.If | ast.While, filename: str
-    ) -> list[Diagnostic]:
+    def _check_array_equality(self, node: ast.If | ast.While, filename: str) -> list[Diagnostic]:
         test = node.test
         flagged: list[ast.AST] = []
 
@@ -160,7 +158,11 @@ def _is_np_array_literal(node: ast.Call) -> bool:
 
     @staticmethod
     def _is_int_literal(elt: ast.expr) -> bool:
-        return isinstance(elt, ast.Constant) and isinstance(elt.value, int) and not isinstance(elt.value, bool)
+        return (
+            isinstance(elt, ast.Constant)
+            and isinstance(elt.value, int)
+            and not isinstance(elt.value, bool)
+        )
 
     @staticmethod
     def _is_none_or_nan(elt: ast.expr) -> bool:

src/pycodegate/rules/pandas_rules.py

@@ -25,18 +25,13 @@ def check(self, source: str, filename: str) -> list[Diagnostic]:
     # ------------------------------------------------------------------
     # pandas-chained-indexing
     # ------------------------------------------------------------------
-    def _check_chained_indexing(
-        self, tree: ast.Module, filename: str
-    ) -> list[Diagnostic]:
+    def _check_chained_indexing(self, tree: ast.Module, filename: str) -> list[Diagnostic]:
         results: list[Diagnostic] = []
         for node in ast.walk(tree):
             if not isinstance(node, ast.Assign):
                 continue
             for target in node.targets:
-                if (
-                    isinstance(target, ast.Subscript)
-                    and isinstance(target.value, ast.Subscript)
-                ):
+                if isinstance(target, ast.Subscript) and isinstance(target.value, ast.Subscript):
                     results.append(
                         Diagnostic(
                             file_path=filename,
@@ -55,9 +50,7 @@ def _check_chained_indexing(
     # ------------------------------------------------------------------
     # pandas-inplace-assignment
     # ------------------------------------------------------------------
-    def _check_inplace_assignment(
-        self, tree: ast.Module, filename: str
-    ) -> list[Diagnostic]:
+    def _check_inplace_assignment(self, tree: ast.Module, filename: str) -> list[Diagnostic]:
         results: list[Diagnostic] = []
         for node in ast.walk(tree):
             if not isinstance(node, ast.Assign):
@@ -90,9 +83,7 @@ def _check_inplace_assignment(
     # ------------------------------------------------------------------
     # pandas-nan-comparison
     # ------------------------------------------------------------------
-    def _check_nan_comparison(
-        self, tree: ast.Module, filename: str
-    ) -> list[Diagnostic]:
+    def _check_nan_comparison(self, tree: ast.Module, filename: str) -> list[Diagnostic]:
         results: list[Diagnostic] = []
         for node in ast.walk(tree):
             if not isinstance(node, ast.Compare):

src/pycodegate/rules/pydantic.py

@@ -10,9 +10,7 @@
 
 def _is_basemodel_subclass(node: ast.ClassDef) -> bool:
     """Check if any base in node.bases is a Name with id 'BaseModel'."""
-    return any(
-        isinstance(base, ast.Name) and base.id == "BaseModel" for base in node.bases
-    )
+    return any(isinstance(base, ast.Name) and base.id == "BaseModel" for base in node.bases)
 
 
 class PydanticRules(BaseRules):
@@ -112,9 +110,17 @@ def _check_validator_no_return(self, cls: ast.ClassDef, filename: str) -> list[D
     @staticmethod
     def _has_validator_decorator(node: ast.FunctionDef | ast.AsyncFunctionDef) -> bool:
         for dec in node.decorator_list:
-            if isinstance(dec, ast.Name) and dec.id in ("field_validator", "model_validator", "validator"):
+            if isinstance(dec, ast.Name) and dec.id in (
+                "field_validator",
+                "model_validator",
+                "validator",
+            ):
                 return True
-            if isinstance(dec, ast.Call) and isinstance(dec.func, ast.Name) and dec.func.id in ("field_validator", "model_validator", "validator"):
+            if (
+                isinstance(dec, ast.Call)
+                and isinstance(dec.func, ast.Name)
+                and dec.func.id in ("field_validator", "model_validator", "validator")
+            ):
                 return True
         return False
 
@@ -277,7 +283,11 @@ def _has_field_validator_decorator(node: ast.FunctionDef | ast.AsyncFunctionDef)
         for dec in node.decorator_list:
             if isinstance(dec, ast.Name) and dec.id == "field_validator":
                 return True
-            if isinstance(dec, ast.Call) and isinstance(dec.func, ast.Name) and dec.func.id == "field_validator":
+            if (
+                isinstance(dec, ast.Call)
+                and isinstance(dec.func, ast.Name)
+                and dec.func.id == "field_validator"
+            ):
                 return True
         return False
 

src/pycodegate/rules/pytest_rules.py

@@ -50,9 +50,7 @@ def _check_assert_tuple(self, tree: ast.Module, filename: str) -> list[Diagnosti
     # Rule 2: pytest-raises-instead-of-try
     # ------------------------------------------------------------------
 
-    def _check_raises_instead_of_try(
-        self, tree: ast.Module, filename: str
-    ) -> list[Diagnostic]:
+    def _check_raises_instead_of_try(self, tree: ast.Module, filename: str) -> list[Diagnostic]:
         """Detect try/except anti-patterns in test functions."""
         diags: list[Diagnostic] = []
         for node in ast.walk(tree):
@@ -147,6 +145,10 @@ def _has_float_eq(node: ast.expr | None) -> bool:
         if not isinstance(node, ast.Compare):
             return False
         for op, comparator in zip(node.ops, node.comparators):  # noqa: B905
-            if isinstance(op, ast.Eq) and isinstance(comparator, ast.Constant) and isinstance(comparator.value, float):
+            if (
+                isinstance(op, ast.Eq)
+                and isinstance(comparator, ast.Constant)
+                and isinstance(comparator.value, float)
+            ):
                 return True
         return False

src/pycodegate/rules/requests_rules.py

@@ -147,7 +147,11 @@ def _is_attr_call(node: ast.Call, attr: str) -> bool:
 
     def _check_verify_disabled(self, node: ast.Call, filename: str) -> list[Diagnostic]:
         for kw in node.keywords:
-            if kw.arg == "verify" and isinstance(kw.value, ast.Constant) and kw.value.value is False:
+            if (
+                kw.arg == "verify"
+                and isinstance(kw.value, ast.Constant)
+                and kw.value.value is False
+            ):
                 return [
                     Diagnostic(
                         file_path=filename,

tests/rules/test_celery.py

@@ -7,6 +7,7 @@ def _run(source: str) -> list:
 
 # -- celery-missing-bind --------------------------------------------------
 
+
 def test_missing_bind():
     source = """
 from celery import shared_task
@@ -33,6 +34,7 @@ def add(self, x, y):
 
 # -- celery-retry-no-exc --------------------------------------------------
 
+
 def test_retry_no_exc():
     source = """
 @app.task(bind=True)
@@ -61,6 +63,7 @@ def fetch(self, url):
 
 # -- celery-broad-autoretry ------------------------------------------------
 
+
 def test_broad_autoretry():
     source = """
 @app.task(autoretry_for=(Exception,))
@@ -83,6 +86,7 @@ def process(data):
 
 # -- celery-direct-call ----------------------------------------------------
 
+
 def test_direct_call():
     source = """
 @app.task

tests/rules/test_logging_rules.py

@@ -8,6 +8,7 @@ def _run(source: str, filename: str = "app.py") -> list:
 
 # --- logging-fstring ---
 
+
 def test_logging_fstring_detected():
     source = "logger.info(f'User {user_id} logged in')"
     diags = _run(source)
@@ -24,6 +25,7 @@ def test_logging_percent_style_ok():
 
 # --- logging-root-logger ---
 
+
 def test_logging_root_logger_detected():
     source = "import logging\nlogging.info('starting up')"
     diags = _run(source)
@@ -38,13 +40,9 @@ def test_logging_named_logger_ok():
 
 # --- logging-error-no-exc-info ---
 
+
 def test_logging_error_no_exc_info_detected():
-    source = (
-        "try:\n"
-        "    do_something()\n"
-        "except Exception as e:\n"
-        "    logger.error('failed')\n"
-    )
+    source = "try:\n    do_something()\nexcept Exception as e:\n    logger.error('failed')\n"
     diags = _run(source)
     assert any(d.rule == "logging-error-no-exc-info" for d in diags)
 

tests/rules/test_sqlalchemy.py

@@ -7,6 +7,7 @@ def _run(source: str) -> list:
 
 # -- sqla-sql-injection ------------------------------------------------
 
+
 def test_sql_injection_fstring():
     source = """
 from sqlalchemy import text
@@ -27,6 +28,7 @@ def test_sql_injection_parameterized_ok():
 
 # -- sqla-identity-compare ---------------------------------------------
 
+
 def test_identity_compare_is_none():
     source = """
 session.query(User).filter(User.name is None)
@@ -45,6 +47,7 @@ def test_identity_compare_eq_none_ok():
 
 # -- sqla-mutable-default ----------------------------------------------
 
+
 def test_mutable_default_list_literal():
     source = """
 from sqlalchemy import Column, JSON
@@ -65,6 +68,7 @@ def test_mutable_default_callable_ok():
 
 # -- sqla-len-all -------------------------------------------------------
 
+
 def test_len_all_flagged():
     source = """
 count = len(session.query(User).all())