Track applicability in fix, add unsafeFixes

Author: jhossbach

pylsp_ruff/plugin.py

@@ -214,6 +214,8 @@ def pylsp_code_actions(
     code_actions = []
     has_organize_imports = False
 
+    settings = load_settings(workspace=workspace, document_path=document.path)
+
     for diagnostic in diagnostics:
         code_actions.append(
             create_disable_code_action(document=document, diagnostic=diagnostic)
@@ -222,6 +224,10 @@ def pylsp_code_actions(
         if diagnostic.data:  # Has fix
             fix = converter.structure(diagnostic.data, RuffFix)
 
+            # Ignore fix if marked as unsafe and unsafe_fixes are disabled
+            if fix.applicability != "safe" and not settings.unsafe_fixes:
+                continue
+
             if diagnostic.code == "I001":
                 code_actions.append(
                     create_organize_imports_code_action(
@@ -236,7 +242,6 @@ def pylsp_code_actions(
                     ),
                 )
 
-    settings = load_settings(workspace=workspace, document_path=document.path)
     checks = run_ruff_check(document=document, settings=settings)
     checks_with_fixes = [c for c in checks if c.fix]
     checks_organize_imports = [c for c in checks_with_fixes if c.code == "I001"]

pylsp_ruff/ruff.py

@@ -19,6 +19,7 @@ class Edit:
 class Fix:
     edits: List[Edit]
     message: str
+    applicability: str
 
 
 @dataclass

tests/test_code_actions.py

@@ -115,6 +115,12 @@ def f():
             pass
         """
     )
+    expected_str_safe = dedent(
+        """
+        def f():
+            a = 2
+        """
+    )
     workspace._config.update(
         {
             "plugins": {
@@ -129,6 +135,20 @@ def f():
     fixed_str = ruff_lint.run_ruff_fix(doc, settings)
     assert fixed_str == expected_str
 
+    workspace._config.update(
+        {
+            "plugins": {
+                "ruff": {
+                    "unsafeFixes": False,
+                }
+            }
+        }
+    )
+    _, doc = temp_document(codeaction_str, workspace)
+    settings = ruff_lint.load_settings(workspace, doc.path)
+    fixed_str = ruff_lint.run_ruff_fix(doc, settings)
+    assert fixed_str == expected_str_safe
+
 
 def test_format_document_default_settings(workspace):
     _, doc = temp_document(import_str, workspace)