style: Formatting to match ruff

Author: AdiKsOnDev

src/linear_cli/api/client/client.py

@@ -1284,8 +1284,10 @@ async def resolve_milestone_id(
         if project_id:
             # First pass: look for exact match in the specified project
             for milestone in nodes:
-                if (milestone.get("name", "").lower() == milestone_identifier.lower() and
-                    milestone.get("project", {}).get("id") == project_id):
+                if (
+                    milestone.get("name", "").lower() == milestone_identifier.lower()
+                    and milestone.get("project", {}).get("id") == project_id
+                ):
                     return milestone.get("id")
 
         # Fallback: search all milestones without project constraint

src/linear_cli/cli/commands/project.py

@@ -318,13 +318,11 @@ async def fetch_updates() -> dict[str, Any]:
 
 @project.command("milestones")
 @click.argument("project_id")
-@click.option("--limit", "-l", type=int, default=50, help="Maximum number of milestones to show")
+@click.option(
+    "--limit", "-l", type=int, default=50, help="Maximum number of milestones to show"
+)
 @click.pass_context
-def list_milestones(
-    ctx: click.Context,
-    project_id: str,
-    limit: int
-) -> None:
+def list_milestones(ctx: click.Context, project_id: str, limit: int) -> None:
     """
     List milestones for a project.
 
@@ -360,10 +358,7 @@ async def fetch_milestones() -> dict[str, Any]:
                 if milestone.get("project", {}).get("id") == project_data["id"]:
                     project_milestones.append(milestone)
 
-            return {
-                "nodes": project_milestones,
-                "pageInfo": result.get("pageInfo", {})
-            }
+            return {"nodes": project_milestones, "pageInfo": result.get("pageInfo", {})}
 
         return {}
 
@@ -429,7 +424,11 @@ async def fetch_milestone() -> dict[str, Any] | None:
 @click.option("--target-date", help="Target completion date (YYYY-MM-DD)")
 @click.pass_context
 def create_milestone(
-    ctx: click.Context, project_id: str, name: str, description: str | None, target_date: str | None
+    ctx: click.Context,
+    project_id: str,
+    name: str,
+    description: str | None,
+    target_date: str | None,
 ) -> None:
     """
     Create a new milestone for a project.
@@ -465,7 +464,9 @@ async def create_milestone_async() -> dict[str, Any]:
                 parsed_date = datetime.strptime(target_date, "%Y-%m-%d")
                 formatted_date = f"{parsed_date.strftime('%Y-%m-%d')}T00:00:00Z"
             except ValueError:
-                print_error(f"Invalid date format: {target_date}. Use YYYY-MM-DD format.")
+                print_error(
+                    f"Invalid date format: {target_date}. Use YYYY-MM-DD format."
+                )
                 return {}
 
         result = await client.create_milestone(
@@ -541,7 +542,9 @@ async def update_milestone_async() -> dict[str, Any]:
                 parsed_date = datetime.strptime(target_date, "%Y-%m-%d")
                 formatted_date = f"{parsed_date.strftime('%Y-%m-%d')}T00:00:00Z"
             except ValueError:
-                print_error(f"Invalid date format: {target_date}. Use YYYY-MM-DD format.")
+                print_error(
+                    f"Invalid date format: {target_date}. Use YYYY-MM-DD format."
+                )
                 return {}
 
         result = await client.update_milestone(
@@ -575,7 +578,9 @@ async def update_milestone_async() -> dict[str, Any]:
 @click.argument("milestone_id")
 @click.option("--yes", "-y", is_flag=True, help="Skip confirmation prompt")
 @click.pass_context
-def delete_milestone(ctx: click.Context, project_id: str, milestone_id: str, yes: bool) -> None:
+def delete_milestone(
+    ctx: click.Context, project_id: str, milestone_id: str, yes: bool
+) -> None:
     """
     Delete a milestone from a project.
 
@@ -600,7 +605,10 @@ async def delete_milestone_async() -> bool:
 
     try:
         if not yes:
-            click.confirm(f"Are you sure you want to delete milestone '{milestone_id}'?", abort=True)
+            click.confirm(
+                f"Are you sure you want to delete milestone '{milestone_id}'?",
+                abort=True,
+            )
 
         success = asyncio.run(delete_milestone_async())
         if success:
@@ -616,9 +624,13 @@ async def delete_milestone_async() -> bool:
 @project.command("milestone-issues")
 @click.argument("project_id")
 @click.argument("milestone_id")
-@click.option("--limit", "-l", type=int, default=50, help="Maximum number of issues to show")
+@click.option(
+    "--limit", "-l", type=int, default=50, help="Maximum number of issues to show"
+)
 @click.pass_context
-def list_milestone_issues(ctx: click.Context, project_id: str, milestone_id: str, limit: int) -> None:
+def list_milestone_issues(
+    ctx: click.Context, project_id: str, milestone_id: str, limit: int
+) -> None:
     """
     List issues in a project milestone.
 
@@ -652,7 +664,9 @@ async def fetch_milestone_issues() -> dict[str, Any] | None:
         if not milestone_data:
             raise click.Abort()
 
-        console.print(f"[bold]Issues in milestone:[/bold] {milestone_data.get('name', milestone_id)}")
+        console.print(
+            f"[bold]Issues in milestone:[/bold] {milestone_data.get('name', milestone_id)}"
+        )
 
         issues = milestone_data.get("issues", {}).get("nodes", [])
         if not issues:
@@ -670,9 +684,18 @@ async def fetch_milestone_issues() -> dict[str, Any] | None:
 
 @project.command("create-test-data")
 @click.option("--team", "-t", required=True, help="Team key or ID to use for test data")
-@click.option("--projects", type=int, default=1, help="Number of test projects to create")
-@click.option("--milestones-per-project", type=int, default=3, help="Number of milestones per project")
-@click.option("--issues-per-milestone", type=int, default=5, help="Number of issues per milestone")
+@click.option(
+    "--projects", type=int, default=1, help="Number of test projects to create"
+)
+@click.option(
+    "--milestones-per-project",
+    type=int,
+    default=3,
+    help="Number of milestones per project",
+)
+@click.option(
+    "--issues-per-milestone", type=int, default=5, help="Number of issues per milestone"
+)
 @click.pass_context
 def create_test_data(
     ctx: click.Context,
@@ -730,7 +753,9 @@ async def create_test_data_async() -> dict[str, int]:
 
                 # Create milestones for this project
                 for milestone_num in range(1, milestones_per_project + 1):
-                    status.update(f"Creating milestone {milestone_num} for project {project_num}...")
+                    status.update(
+                        f"Creating milestone {milestone_num} for project {project_num}..."
+                    )
 
                     milestone_name = f"Milestone {milestone_num}"
                     milestone_result = await client.create_milestone(
@@ -740,15 +765,19 @@ async def create_test_data_async() -> dict[str, int]:
                     )
 
                     if not milestone_result.get("success"):
-                        console.print(f"[red]Failed to create milestone {milestone_name}[/red]")
+                        console.print(
+                            f"[red]Failed to create milestone {milestone_name}[/red]"
+                        )
                         continue
 
                     milestone_id = milestone_result["projectMilestone"]["id"]
                     counts["milestones"] += 1
 
                     # Create issues for this milestone
                     for issue_num in range(1, issues_per_milestone + 1):
-                        status.update(f"Creating issue {issue_num} for milestone {milestone_num}...")
+                        status.update(
+                            f"Creating issue {issue_num} for milestone {milestone_num}..."
+                        )
 
                         # WHY: Use secure random choice for realistic test data variety
                         # Different priorities and states make testing more realistic

tests/unit/test_project_milestone_commands.py

@@ -97,7 +97,9 @@ def test_list_project_milestones(self, runner, mock_cli_context):
         )
 
         assert result.exit_code == 0
-        mock_cli_context.get_client().get_project.assert_called_once_with("Test Project")
+        mock_cli_context.get_client().get_project.assert_called_once_with(
+            "Test Project"
+        )
         mock_cli_context.get_client().get_milestones.assert_called_once_with(limit=50)
 
     def test_list_milestones_project_not_found(self, runner, mock_cli_context):
@@ -128,7 +130,9 @@ def test_show_project_milestone(self, runner, mock_cli_context):
         }
 
         # Mock milestone resolution first
-        mock_cli_context.get_client().resolve_milestone_id.return_value = "milestone_123"
+        mock_cli_context.get_client().resolve_milestone_id.return_value = (
+            "milestone_123"
+        )
         mock_cli_context.get_client().get_milestone.return_value = milestone_data
 
         result = runner.invoke(
@@ -138,18 +142,21 @@ def test_show_project_milestone(self, runner, mock_cli_context):
         )
 
         assert result.exit_code == 0
-        mock_cli_context.get_client().resolve_milestone_id.assert_called_once_with("milestone_123", "Test Project")
-        mock_cli_context.get_client().get_milestone.assert_called_once_with("milestone_123")
+        mock_cli_context.get_client().resolve_milestone_id.assert_called_once_with(
+            "milestone_123", "Test Project"
+        )
+        mock_cli_context.get_client().get_milestone.assert_called_once_with(
+            "milestone_123"
+        )
 
     def test_show_milestone_by_name_with_resolution(self, runner, mock_cli_context):
         """Test showing milestone by name with resolution."""
-        milestone_data = {
-            "id": "milestone_123",
-            "name": "Sprint 1"
-        }
+        milestone_data = {"id": "milestone_123", "name": "Sprint 1"}
 
         # Mock name resolution
-        mock_cli_context.get_client().resolve_milestone_id.return_value = "milestone_123"
+        mock_cli_context.get_client().resolve_milestone_id.return_value = (
+            "milestone_123"
+        )
         mock_cli_context.get_client().get_milestone.return_value = milestone_data
 
         result = runner.invoke(
@@ -159,8 +166,12 @@ def test_show_milestone_by_name_with_resolution(self, runner, mock_cli_context):
         )
 
         assert result.exit_code == 0
-        mock_cli_context.get_client().resolve_milestone_id.assert_called_once_with("Sprint 1", "Test Project")
-        mock_cli_context.get_client().get_milestone.assert_called_once_with("milestone_123")
+        mock_cli_context.get_client().resolve_milestone_id.assert_called_once_with(
+            "Sprint 1", "Test Project"
+        )
+        mock_cli_context.get_client().get_milestone.assert_called_once_with(
+            "milestone_123"
+        )
 
 
 class TestCreateMilestone:
@@ -259,7 +270,9 @@ class TestUpdateMilestone:
     def test_update_project_milestone(self, runner, mock_cli_context):
         """Test updating milestone in a project."""
         # Mock milestone resolution
-        mock_cli_context.get_client().resolve_milestone_id.return_value = "milestone_123"
+        mock_cli_context.get_client().resolve_milestone_id.return_value = (
+            "milestone_123"
+        )
 
         # Mock update result
         update_result = {
@@ -273,7 +286,13 @@ def test_update_project_milestone(self, runner, mock_cli_context):
 
         result = runner.invoke(
             project,
-            ["update-milestone", "Test Project", "Sprint 1", "--name", "Sprint 1 Updated"],
+            [
+                "update-milestone",
+                "Test Project",
+                "Sprint 1",
+                "--name",
+                "Sprint 1 Updated",
+            ],
             obj={"cli_context": mock_cli_context},
         )
 
@@ -294,7 +313,9 @@ class TestDeleteMilestone:
     def test_delete_project_milestone(self, runner, mock_cli_context):
         """Test deleting milestone from a project."""
         # Mock milestone resolution
-        mock_cli_context.get_client().resolve_milestone_id.return_value = "milestone_123"
+        mock_cli_context.get_client().resolve_milestone_id.return_value = (
+            "milestone_123"
+        )
         mock_cli_context.get_client().delete_milestone.return_value = True
 
         result = runner.invoke(
@@ -307,11 +328,15 @@ def test_delete_project_milestone(self, runner, mock_cli_context):
         assert result.exit_code == 0
         assert "Deleted milestone: Sprint 1" in result.output
 
-        mock_cli_context.get_client().delete_milestone.assert_called_once_with("milestone_123")
+        mock_cli_context.get_client().delete_milestone.assert_called_once_with(
+            "milestone_123"
+        )
 
     def test_delete_milestone_with_yes_flag(self, runner, mock_cli_context):
         """Test deleting milestone with --yes flag."""
-        mock_cli_context.get_client().resolve_milestone_id.return_value = "milestone_123"
+        mock_cli_context.get_client().resolve_milestone_id.return_value = (
+            "milestone_123"
+        )
         mock_cli_context.get_client().delete_milestone.return_value = True
 
         result = runner.invoke(
@@ -330,7 +355,9 @@ class TestMilestoneIssues:
     def test_list_milestone_issues(self, runner, mock_cli_context):
         """Test listing issues in a project milestone."""
         # Mock milestone resolution
-        mock_cli_context.get_client().resolve_milestone_id.return_value = "milestone_123"
+        mock_cli_context.get_client().resolve_milestone_id.return_value = (
+            "milestone_123"
+        )
 
         # Mock milestone data with issues
         milestone_data = {
@@ -431,11 +458,13 @@ def test_project_milestone_command_hierarchy(self):
             "update-milestone",
             "delete-milestone",
             "milestone-issues",
-            "create-test-data"
+            "create-test-data",
         ]
 
         for expected in expected_milestone_commands:
-            assert expected in command_names, f"Command {expected} not found in project commands"
+            assert expected in command_names, (
+                f"Command {expected} not found in project commands"
+            )
 
     def test_milestone_commands_require_project_context(self):
         """Test that milestone commands properly require project context."""
@@ -455,14 +484,16 @@ def test_milestone_commands_require_project_context(self):
             create_milestone,
             update_milestone,
             delete_milestone,
-            list_milestone_issues
+            list_milestone_issues,
         ]
 
         for cmd in commands_with_project_arg:
             # Check that first parameter is project_id
-            params = [p for p in cmd.params if hasattr(p, 'name')]
-            project_params = [p for p in params if 'project' in p.name.lower()]
-            assert len(project_params) > 0, f"Command {cmd.name} missing project parameter"
+            params = [p for p in cmd.params if hasattr(p, "name")]
+            project_params = [p for p in params if "project" in p.name.lower()]
+            assert len(project_params) > 0, (
+                f"Command {cmd.name} missing project parameter"
+            )
 
 
 if __name__ == "__main__":