feat: Implement comprehensive project management system with issue lifecycle support
- Add ProjectManager with milestone and label-based project organization - Support project states (Todo, Active, Review, Done, Blocked) via labels - Add priority management (Low, Medium, High, Critical) with label integration - Implement milestone creation and management for project tracking - Enhance IssueWriter with project management methods (assign_to_milestone, add/remove_labels) - Add 8 new CLI commands for complete project management workflow - Support automatic project management setup with ensure_project_labels() - Enable issue state transitions with automatic closing for completed issues - Integrate with existing Gitea API authentication and error handling patterns 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
322
tddai/project_manager.py
Normal file
322
tddai/project_manager.py
Normal file
@@ -0,0 +1,322 @@
|
||||
"""
|
||||
Project management functionality for Gitea using milestones and labels.
|
||||
|
||||
Since Gitea project boards may not be available in all instances, this module
|
||||
provides project management using milestones (for projects) and labels (for states).
|
||||
"""
|
||||
|
||||
import json
|
||||
import os
|
||||
import subprocess
|
||||
from subprocess import PIPE
|
||||
from typing import Dict, Any, List, Optional
|
||||
from dataclasses import dataclass
|
||||
from enum import Enum
|
||||
|
||||
from .config import get_config
|
||||
from .exceptions import IssueError
|
||||
|
||||
|
||||
class ProjectState(Enum):
|
||||
"""Standard project states using labels."""
|
||||
TODO = "status:todo"
|
||||
ACTIVE = "status:active"
|
||||
REVIEW = "status:review"
|
||||
DONE = "status:done"
|
||||
BLOCKED = "status:blocked"
|
||||
|
||||
|
||||
class Priority(Enum):
|
||||
"""Priority levels using labels."""
|
||||
LOW = "priority:low"
|
||||
MEDIUM = "priority:medium"
|
||||
HIGH = "priority:high"
|
||||
CRITICAL = "priority:critical"
|
||||
|
||||
|
||||
@dataclass
|
||||
class Milestone:
|
||||
"""Represents a project milestone."""
|
||||
id: int
|
||||
title: str
|
||||
description: str
|
||||
state: str
|
||||
open_issues: int
|
||||
closed_issues: int
|
||||
due_on: Optional[str] = None
|
||||
|
||||
|
||||
@dataclass
|
||||
class Label:
|
||||
"""Represents an issue label."""
|
||||
id: int
|
||||
name: str
|
||||
color: str
|
||||
description: str
|
||||
|
||||
|
||||
class ProjectManager:
|
||||
"""Manages project organization using milestones and labels."""
|
||||
|
||||
def __init__(self, config=None, auth_token=None):
|
||||
self.config = config or get_config()
|
||||
self.auth_token = auth_token or os.getenv('GITEA_API_TOKEN')
|
||||
|
||||
def _make_api_call(self, method: str, url: str, data: Dict[str, Any] = None) -> Dict[str, Any]:
|
||||
"""Make authenticated API call to Gitea."""
|
||||
if not self.auth_token:
|
||||
raise IssueError("Authentication token required for project operations")
|
||||
|
||||
cmd = [
|
||||
'curl', '-s', '-X', method,
|
||||
'-H', 'Content-Type: application/json',
|
||||
'-H', f'Authorization: token {self.auth_token}',
|
||||
]
|
||||
|
||||
if data:
|
||||
cmd.extend(['-d', json.dumps(data)])
|
||||
|
||||
cmd.append(url)
|
||||
|
||||
try:
|
||||
result = subprocess.run(cmd, stdout=PIPE, stderr=PIPE, universal_newlines=True, check=True)
|
||||
|
||||
if result.returncode != 0:
|
||||
raise IssueError(f"API call failed: {result.stderr}")
|
||||
|
||||
if result.stdout.strip():
|
||||
response_data = json.loads(result.stdout)
|
||||
|
||||
# Check for API error responses
|
||||
if isinstance(response_data, dict) and 'message' in response_data and 'id' not in response_data:
|
||||
raise IssueError(f"API error: {response_data['message']}")
|
||||
|
||||
return response_data
|
||||
else:
|
||||
return {}
|
||||
|
||||
except subprocess.CalledProcessError as e:
|
||||
raise IssueError(f"API call failed: {e}")
|
||||
except json.JSONDecodeError as e:
|
||||
raise IssueError(f"Failed to parse API response: {e}")
|
||||
|
||||
# Milestone Management (Projects)
|
||||
|
||||
def create_milestone(self, title: str, description: str = "", due_date: str = None) -> Milestone:
|
||||
"""Create a new milestone (project)."""
|
||||
url = f"{self.config.gitea_url}/api/v1/repos/{self.config.repo_owner}/{self.config.repo_name}/milestones"
|
||||
|
||||
data = {
|
||||
'title': title,
|
||||
'description': description,
|
||||
}
|
||||
|
||||
if due_date:
|
||||
data['due_on'] = due_date
|
||||
|
||||
response = self._make_api_call('POST', url, data)
|
||||
|
||||
return Milestone(
|
||||
id=response['id'],
|
||||
title=response['title'],
|
||||
description=response.get('description', ''),
|
||||
state=response['state'],
|
||||
open_issues=response['open_issues'],
|
||||
closed_issues=response['closed_issues'],
|
||||
due_on=response.get('due_on')
|
||||
)
|
||||
|
||||
def list_milestones(self, state: str = "open") -> List[Milestone]:
|
||||
"""List all milestones (projects)."""
|
||||
url = f"{self.config.gitea_url}/api/v1/repos/{self.config.repo_owner}/{self.config.repo_name}/milestones"
|
||||
|
||||
params = f"?state={state}" if state else ""
|
||||
response = self._make_api_call('GET', url + params)
|
||||
|
||||
return [
|
||||
Milestone(
|
||||
id=m['id'],
|
||||
title=m['title'],
|
||||
description=m.get('description', ''),
|
||||
state=m['state'],
|
||||
open_issues=m['open_issues'],
|
||||
closed_issues=m['closed_issues'],
|
||||
due_on=m.get('due_on')
|
||||
)
|
||||
for m in response
|
||||
]
|
||||
|
||||
def update_milestone(self, milestone_id: int, **kwargs) -> Milestone:
|
||||
"""Update milestone details."""
|
||||
url = f"{self.config.gitea_url}/api/v1/repos/{self.config.repo_owner}/{self.config.repo_name}/milestones/{milestone_id}"
|
||||
|
||||
# Only include fields that can be updated
|
||||
valid_fields = ['title', 'description', 'state', 'due_on']
|
||||
data = {k: v for k, v in kwargs.items() if k in valid_fields}
|
||||
|
||||
response = self._make_api_call('PATCH', url, data)
|
||||
|
||||
return Milestone(
|
||||
id=response['id'],
|
||||
title=response['title'],
|
||||
description=response.get('description', ''),
|
||||
state=response['state'],
|
||||
open_issues=response['open_issues'],
|
||||
closed_issues=response['closed_issues'],
|
||||
due_on=response.get('due_on')
|
||||
)
|
||||
|
||||
def close_milestone(self, milestone_id: int) -> Milestone:
|
||||
"""Close a milestone (complete project)."""
|
||||
return self.update_milestone(milestone_id, state='closed')
|
||||
|
||||
# Label Management (States & Priority)
|
||||
|
||||
def create_label(self, name: str, color: str, description: str = "") -> Label:
|
||||
"""Create a new label."""
|
||||
url = f"{self.config.gitea_url}/api/v1/repos/{self.config.repo_owner}/{self.config.repo_name}/labels"
|
||||
|
||||
data = {
|
||||
'name': name,
|
||||
'color': color,
|
||||
'description': description
|
||||
}
|
||||
|
||||
response = self._make_api_call('POST', url, data)
|
||||
|
||||
return Label(
|
||||
id=response['id'],
|
||||
name=response['name'],
|
||||
color=response['color'],
|
||||
description=response.get('description', '')
|
||||
)
|
||||
|
||||
def list_labels(self) -> List[Label]:
|
||||
"""List all repository labels."""
|
||||
url = f"{self.config.gitea_url}/api/v1/repos/{self.config.repo_owner}/{self.config.repo_name}/labels"
|
||||
|
||||
response = self._make_api_call('GET', url)
|
||||
|
||||
return [
|
||||
Label(
|
||||
id=l['id'],
|
||||
name=l['name'],
|
||||
color=l['color'],
|
||||
description=l.get('description', '')
|
||||
)
|
||||
for l in response
|
||||
]
|
||||
|
||||
def ensure_project_labels(self) -> None:
|
||||
"""Ensure all required project management labels exist."""
|
||||
existing_labels = {label.name for label in self.list_labels()}
|
||||
|
||||
# Standard state labels
|
||||
required_labels = [
|
||||
('status:todo', 'e6e6e6', 'Issues ready to be worked on'),
|
||||
('status:active', '0052cc', 'Issues currently being worked on'),
|
||||
('status:review', 'fbca04', 'Issues under review'),
|
||||
('status:done', '0e8a16', 'Completed issues'),
|
||||
('status:blocked', 'd93f0b', 'Issues blocked by dependencies'),
|
||||
|
||||
# Priority labels
|
||||
('priority:low', 'c2e0c6', 'Low priority issue'),
|
||||
('priority:medium', 'fef2c0', 'Medium priority issue'),
|
||||
('priority:high', 'f9d0c4', 'High priority issue'),
|
||||
('priority:critical', 'f4c2c2', 'Critical priority issue'),
|
||||
|
||||
# Type labels
|
||||
('type:bug', 'fc2929', 'Bug report'),
|
||||
('type:feature', '84b6eb', 'New feature request'),
|
||||
('type:enhancement', '7057ff', 'Enhancement to existing feature'),
|
||||
('type:documentation', '0075ca', 'Documentation update'),
|
||||
]
|
||||
|
||||
for name, color, description in required_labels:
|
||||
if name not in existing_labels:
|
||||
try:
|
||||
self.create_label(name, color, description)
|
||||
print(f"✅ Created label: {name}")
|
||||
except IssueError as e:
|
||||
print(f"⚠️ Failed to create label {name}: {e}")
|
||||
|
||||
# Project Management Operations
|
||||
|
||||
def assign_issue_to_milestone(self, issue_number: int, milestone_id: int) -> Dict[str, Any]:
|
||||
"""Assign issue to a milestone (project)."""
|
||||
url = f"{self.config.issues_api_url}/{issue_number}"
|
||||
|
||||
data = {'milestone': milestone_id}
|
||||
return self._make_api_call('PATCH', url, data)
|
||||
|
||||
def set_issue_state(self, issue_number: int, state: ProjectState) -> Dict[str, Any]:
|
||||
"""Set issue project state using labels."""
|
||||
# First remove any existing state labels
|
||||
issue_url = f"{self.config.issues_api_url}/{issue_number}"
|
||||
issue_data = self._make_api_call('GET', issue_url)
|
||||
|
||||
current_labels = [label['name'] for label in issue_data.get('labels', [])]
|
||||
state_labels = [label for label in current_labels if label.startswith('status:')]
|
||||
|
||||
# Remove old state labels
|
||||
for old_state in state_labels:
|
||||
if old_state in current_labels:
|
||||
current_labels.remove(old_state)
|
||||
|
||||
# Add new state label
|
||||
current_labels.append(state.value)
|
||||
|
||||
# Update issue with new labels
|
||||
data = {'labels': current_labels}
|
||||
return self._make_api_call('PATCH', issue_url, data)
|
||||
|
||||
def set_issue_priority(self, issue_number: int, priority: Priority) -> Dict[str, Any]:
|
||||
"""Set issue priority using labels."""
|
||||
issue_url = f"{self.config.issues_api_url}/{issue_number}"
|
||||
issue_data = self._make_api_call('GET', issue_url)
|
||||
|
||||
current_labels = [label['name'] for label in issue_data.get('labels', [])]
|
||||
priority_labels = [label for label in current_labels if label.startswith('priority:')]
|
||||
|
||||
# Remove old priority labels
|
||||
for old_priority in priority_labels:
|
||||
if old_priority in current_labels:
|
||||
current_labels.remove(old_priority)
|
||||
|
||||
# Add new priority label
|
||||
current_labels.append(priority.value)
|
||||
|
||||
# Update issue with new labels
|
||||
data = {'labels': current_labels}
|
||||
return self._make_api_call('PATCH', issue_url, data)
|
||||
|
||||
def move_issue_to_done(self, issue_number: int) -> Dict[str, Any]:
|
||||
"""Move issue to done state and close it."""
|
||||
# Set state to done
|
||||
self.set_issue_state(issue_number, ProjectState.DONE)
|
||||
|
||||
# Close the issue
|
||||
url = f"{self.config.issues_api_url}/{issue_number}"
|
||||
data = {'state': 'closed'}
|
||||
return self._make_api_call('PATCH', url, data)
|
||||
|
||||
def get_project_overview(self) -> Dict[str, Any]:
|
||||
"""Get overview of project status."""
|
||||
milestones = self.list_milestones("all")
|
||||
labels = self.list_labels()
|
||||
|
||||
# Count issues by state
|
||||
state_counts = {}
|
||||
for state in ProjectState:
|
||||
state_counts[state.value] = 0
|
||||
|
||||
# This would require fetching all issues to count by labels
|
||||
# For now, return milestone overview
|
||||
|
||||
return {
|
||||
'milestones': len(milestones),
|
||||
'active_projects': len([m for m in milestones if m.state == 'open']),
|
||||
'completed_projects': len([m for m in milestones if m.state == 'closed']),
|
||||
'total_labels': len(labels),
|
||||
'project_management_ready': len([l for l in labels if l.name.startswith('status:')]) > 0
|
||||
}
|
||||
Reference in New Issue
Block a user