codeshift 0.4.0__py3-none-any.whl → 0.7.0__py3-none-any.whl

codeshift/health/models.py

@@ -0,0 +1,284 @@
+"""Data models for the health score feature."""
+
+from dataclasses import dataclass, field
+from datetime import datetime
+from enum import Enum
+from pathlib import Path
+from typing import Any
+
+
+class MetricCategory(Enum):
+    """Categories of health metrics."""
+
+    FRESHNESS = "freshness"
+    SECURITY = "security"
+    MIGRATION_READINESS = "migration_readiness"
+    TEST_COVERAGE = "test_coverage"
+    DOCUMENTATION = "documentation"
+
+
+class HealthGrade(Enum):
+    """Letter grade for overall health score."""
+
+    A = "A"  # 90-100
+    B = "B"  # 80-89
+    C = "C"  # 70-79
+    D = "D"  # 60-69
+    F = "F"  # Below 60
+
+    @classmethod
+    def from_score(cls, score: float) -> "HealthGrade":
+        """Convert a numeric score to a letter grade.
+
+        Args:
+            score: Numeric score from 0-100
+
+        Returns:
+            Corresponding letter grade
+        """
+        if score >= 90:
+            return cls.A
+        elif score >= 80:
+            return cls.B
+        elif score >= 70:
+            return cls.C
+        elif score >= 60:
+            return cls.D
+        else:
+            return cls.F
+
+    @property
+    def color(self) -> str:
+        """Get the display color for this grade."""
+        colors = {
+            HealthGrade.A: "green",
+            HealthGrade.B: "cyan",
+            HealthGrade.C: "yellow",
+            HealthGrade.D: "orange1",
+            HealthGrade.F: "red",
+        }
+        return colors.get(self, "white")
+
+    @property
+    def emoji(self) -> str:
+        """Get the emoji for this grade."""
+        emojis = {
+            HealthGrade.A: "🟢",
+            HealthGrade.B: "🔵",
+            HealthGrade.C: "🟡",
+            HealthGrade.D: "🟠",
+            HealthGrade.F: "🔴",
+        }
+        return emojis.get(self, "⚪")
+
+
+class VulnerabilitySeverity(Enum):
+    """Severity levels for security vulnerabilities."""
+
+    CRITICAL = "critical"
+    HIGH = "high"
+    MEDIUM = "medium"
+    LOW = "low"
+
+    @property
+    def penalty(self) -> int:
+        """Get the score penalty for this severity level."""
+        penalties = {
+            VulnerabilitySeverity.CRITICAL: 25,
+            VulnerabilitySeverity.HIGH: 15,
+            VulnerabilitySeverity.MEDIUM: 8,
+            VulnerabilitySeverity.LOW: 3,
+        }
+        return penalties.get(self, 0)
+
+
+@dataclass
+class SecurityVulnerability:
+    """Represents a security vulnerability in a dependency."""
+
+    package: str
+    vulnerability_id: str
+    severity: VulnerabilitySeverity
+    description: str
+    fixed_in: str | None = None
+    url: str | None = None
+
+    def to_dict(self) -> dict[str, Any]:
+        """Convert to dictionary."""
+        return {
+            "package": self.package,
+            "vulnerability_id": self.vulnerability_id,
+            "severity": self.severity.value,
+            "description": self.description,
+            "fixed_in": self.fixed_in,
+            "url": self.url,
+        }
+
+
+@dataclass
+class DependencyHealth:
+    """Health information for a single dependency."""
+
+    name: str
+    current_version: str | None
+    latest_version: str | None
+    is_outdated: bool
+    major_versions_behind: int = 0
+    minor_versions_behind: int = 0
+    has_tier1_support: bool = False
+    has_tier2_support: bool = False
+    vulnerabilities: list[SecurityVulnerability] = field(default_factory=list)
+
+    @property
+    def version_lag_penalty(self) -> int:
+        """Calculate the penalty for version lag."""
+        # Major version lag: -15 points each
+        # Minor version lag: -5 points each (max 3)
+        major_penalty = self.major_versions_behind * 15
+        minor_penalty = min(self.minor_versions_behind, 3) * 5
+        return major_penalty + minor_penalty
+
+    def to_dict(self) -> dict[str, Any]:
+        """Convert to dictionary."""
+        return {
+            "name": self.name,
+            "current_version": self.current_version,
+            "latest_version": self.latest_version,
+            "is_outdated": self.is_outdated,
+            "major_versions_behind": self.major_versions_behind,
+            "minor_versions_behind": self.minor_versions_behind,
+            "has_tier1_support": self.has_tier1_support,
+            "has_tier2_support": self.has_tier2_support,
+            "vulnerabilities": [v.to_dict() for v in self.vulnerabilities],
+        }
+
+
+@dataclass
+class MetricResult:
+    """Result from a single metric calculation."""
+
+    category: MetricCategory
+    score: float  # 0-100
+    weight: float  # 0.0-1.0
+    description: str
+    details: dict[str, Any] = field(default_factory=dict)
+    recommendations: list[str] = field(default_factory=list)
+
+    @property
+    def weighted_score(self) -> float:
+        """Calculate the weighted score contribution."""
+        return self.score * self.weight
+
+    def to_dict(self) -> dict[str, Any]:
+        """Convert to dictionary."""
+        return {
+            "category": self.category.value,
+            "score": self.score,
+            "weight": self.weight,
+            "weighted_score": self.weighted_score,
+            "description": self.description,
+            "details": self.details,
+            "recommendations": self.recommendations,
+        }
+
+
+@dataclass
+class HealthScore:
+    """Complete health score for a project."""
+
+    overall_score: float  # 0-100
+    grade: HealthGrade
+    metrics: list[MetricResult] = field(default_factory=list)
+    dependencies: list[DependencyHealth] = field(default_factory=list)
+    vulnerabilities: list[SecurityVulnerability] = field(default_factory=list)
+    calculated_at: datetime = field(default_factory=datetime.now)
+    project_path: Path = field(default_factory=lambda: Path("."))
+
+    @property
+    def summary(self) -> str:
+        """Get a summary string of the health score."""
+        return f"{self.grade.emoji} Grade {self.grade.value} ({self.overall_score:.1f}/100)"
+
+    @property
+    def top_recommendations(self) -> list[str]:
+        """Get the top 5 recommendations across all metrics."""
+        all_recs: list[tuple[float, str]] = []
+        for metric in self.metrics:
+            # Weight recommendations by how much improvement they could provide
+            improvement_potential = 100 - metric.score
+            for rec in metric.recommendations:
+                all_recs.append((improvement_potential * metric.weight, rec))
+
+        # Sort by improvement potential and return top 5
+        all_recs.sort(key=lambda x: x[0], reverse=True)
+        seen: set[str] = set()
+        unique_recs: list[str] = []
+        for _, rec in all_recs:
+            if rec not in seen:
+                seen.add(rec)
+                unique_recs.append(rec)
+                if len(unique_recs) >= 5:
+                    break
+        return unique_recs
+
+    def to_dict(self) -> dict[str, Any]:
+        """Convert to dictionary."""
+        return {
+            "overall_score": self.overall_score,
+            "grade": self.grade.value,
+            "metrics": [m.to_dict() for m in self.metrics],
+            "dependencies": [d.to_dict() for d in self.dependencies],
+            "vulnerabilities": [v.to_dict() for v in self.vulnerabilities],
+            "calculated_at": self.calculated_at.isoformat(),
+            "project_path": str(self.project_path),
+            "recommendations": self.top_recommendations,
+        }
+
+
+@dataclass
+class HealthReport:
+    """Health report comparing current score to previous."""
+
+    current: HealthScore
+    previous: HealthScore | None = None
+
+    @property
+    def trend(self) -> str:
+        """Get the trend direction."""
+        if self.previous is None:
+            return "new"
+
+        diff = self.current.overall_score - self.previous.overall_score
+        if diff > 2:
+            return "improving"
+        elif diff < -2:
+            return "declining"
+        else:
+            return "stable"
+
+    @property
+    def trend_emoji(self) -> str:
+        """Get the trend emoji."""
+        emojis = {
+            "improving": "📈",
+            "declining": "📉",
+            "stable": "➡️",
+            "new": "🆕",
+        }
+        return emojis.get(self.trend, "")
+
+    @property
+    def score_delta(self) -> float | None:
+        """Get the score change from previous."""
+        if self.previous is None:
+            return None
+        return self.current.overall_score - self.previous.overall_score
+
+    def to_dict(self) -> dict[str, Any]:
+        """Convert to dictionary."""
+        return {
+            "current": self.current.to_dict(),
+            "previous": self.previous.to_dict() if self.previous else None,
+            "trend": self.trend,
+            "score_delta": self.score_delta,
+        }

codeshift/health/report.py

@@ -0,0 +1,310 @@
+"""Report generation for health scores (JSON and HTML)."""
+
+import html
+import json
+from datetime import datetime
+from pathlib import Path
+from typing import Any
+
+from codeshift.health.models import HealthReport, HealthScore
+
+
+def generate_json_report(report: HealthReport | HealthScore, pretty: bool = True) -> str:
+    """Generate a JSON report.
+
+    Args:
+        report: HealthReport or HealthScore to serialize
+        pretty: Whether to pretty-print the JSON
+
+    Returns:
+        JSON string
+    """
+    if isinstance(report, HealthScore):
+        data = report.to_dict()
+    else:
+        data = report.to_dict()
+
+    if pretty:
+        return json.dumps(data, indent=2, default=_json_serializer)
+    return json.dumps(data, default=_json_serializer)
+
+
+def save_json_report(report: HealthReport | HealthScore, output_path: Path) -> None:
+    """Save a JSON report to a file.
+
+    Args:
+        report: HealthReport or HealthScore to serialize
+        output_path: Path to save the report
+    """
+    json_content = generate_json_report(report)
+    output_path.write_text(json_content)
+
+
+def generate_html_report(report: HealthReport | HealthScore) -> str:
+    """Generate an HTML report.
+
+    Args:
+        report: HealthReport or HealthScore to render
+
+    Returns:
+        HTML string
+    """
+    if isinstance(report, HealthScore):
+        score = report
+        trend_info = ""
+    else:
+        score = report.current
+        if report.previous:
+            delta = report.score_delta or 0
+            sign = "+" if delta >= 0 else ""
+            trend_info = (
+                f'<span class="trend {report.trend}">{report.trend_emoji} {sign}{delta:.1f}</span>'
+            )
+        else:
+            trend_info = '<span class="trend new">New baseline</span>'
+
+    # Build metrics rows
+    metrics_rows = ""
+    for metric in score.metrics:
+        grade_class = _get_score_class(metric.score)
+        metrics_rows += f"""
+        <tr>
+            <td>{_format_category(metric.category.value)}</td>
+            <td class="{grade_class}">{metric.score:.1f}</td>
+            <td>{metric.weight * 100:.0f}%</td>
+            <td>{html.escape(metric.description)}</td>
+        </tr>
+        """
+
+    # Build recommendations list
+    recs_html = ""
+    for rec in score.top_recommendations:
+        recs_html += f"<li>{html.escape(rec)}</li>\n"
+
+    # Build dependencies table
+    deps_rows = ""
+    for dep in score.dependencies:
+        status = "✓" if not dep.is_outdated else "↑"
+        status_class = "up-to-date" if not dep.is_outdated else "outdated"
+        tier = "Tier 1" if dep.has_tier1_support else ("Tier 2" if dep.has_tier2_support else "-")
+        vuln_count = len(dep.vulnerabilities)
+        vuln_class = "vuln-none" if vuln_count == 0 else "vuln-some"
+
+        deps_rows += f"""
+        <tr>
+            <td>{html.escape(dep.name)}</td>
+            <td>{html.escape(dep.current_version or 'unknown')}</td>
+            <td>{html.escape(dep.latest_version or 'unknown')}</td>
+            <td class="{status_class}">{status}</td>
+            <td>{tier}</td>
+            <td class="{vuln_class}">{vuln_count}</td>
+        </tr>
+        """
+
+    # Build vulnerabilities section
+    vulns_html = ""
+    if score.vulnerabilities:
+        vulns_rows = ""
+        for vuln in score.vulnerabilities:
+            vulns_rows += f"""
+            <tr class="severity-{vuln.severity.value}">
+                <td>{html.escape(vuln.package)}</td>
+                <td>{html.escape(vuln.vulnerability_id)}</td>
+                <td>{vuln.severity.value.upper()}</td>
+                <td>{html.escape(vuln.description[:100])}...</td>
+                <td>{html.escape(vuln.fixed_in or '-')}</td>
+            </tr>
+            """
+
+        vulns_html = f"""
+        <section class="vulnerabilities">
+            <h2>Security Vulnerabilities</h2>
+            <table>
+                <thead>
+                    <tr>
+                        <th>Package</th>
+                        <th>ID</th>
+                        <th>Severity</th>
+                        <th>Description</th>
+                        <th>Fixed In</th>
+                    </tr>
+                </thead>
+                <tbody>
+                    {vulns_rows}
+                </tbody>
+            </table>
+        </section>
+        """
+
+    grade_class = score.grade.value.lower()
+
+    return f"""<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Codeshift Health Report</title>
+    <style>
+        :root {{
+            --color-a: #22c55e;
+            --color-b: #06b6d4;
+            --color-c: #eab308;
+            --color-d: #f97316;
+            --color-f: #ef4444;
+        }}
+        * {{ box-sizing: border-box; margin: 0; padding: 0; }}
+        body {{
+            font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif;
+            line-height: 1.6;
+            color: #1f2937;
+            background: #f9fafb;
+            padding: 2rem;
+        }}
+        .container {{ max-width: 1200px; margin: 0 auto; }}
+        h1 {{ margin-bottom: 0.5rem; }}
+        h2 {{ margin: 2rem 0 1rem; border-bottom: 2px solid #e5e7eb; padding-bottom: 0.5rem; }}
+        .header {{ display: flex; justify-content: space-between; align-items: center; margin-bottom: 2rem; }}
+        .score-card {{
+            background: white;
+            border-radius: 12px;
+            padding: 2rem;
+            box-shadow: 0 1px 3px rgba(0,0,0,0.1);
+            text-align: center;
+        }}
+        .grade {{ font-size: 4rem; font-weight: bold; }}
+        .grade.a {{ color: var(--color-a); }}
+        .grade.b {{ color: var(--color-b); }}
+        .grade.c {{ color: var(--color-c); }}
+        .grade.d {{ color: var(--color-d); }}
+        .grade.f {{ color: var(--color-f); }}
+        .overall-score {{ font-size: 1.5rem; color: #6b7280; }}
+        .trend {{ font-size: 1rem; margin-top: 0.5rem; display: block; }}
+        .trend.improving {{ color: var(--color-a); }}
+        .trend.declining {{ color: var(--color-f); }}
+        .trend.stable {{ color: #6b7280; }}
+        table {{ width: 100%; border-collapse: collapse; background: white; border-radius: 8px; overflow: hidden; box-shadow: 0 1px 3px rgba(0,0,0,0.1); }}
+        th, td {{ padding: 0.75rem 1rem; text-align: left; border-bottom: 1px solid #e5e7eb; }}
+        th {{ background: #f3f4f6; font-weight: 600; }}
+        tr:last-child td {{ border-bottom: none; }}
+        .excellent {{ color: var(--color-a); font-weight: 600; }}
+        .good {{ color: var(--color-b); font-weight: 600; }}
+        .fair {{ color: var(--color-c); font-weight: 600; }}
+        .poor {{ color: var(--color-d); font-weight: 600; }}
+        .critical {{ color: var(--color-f); font-weight: 600; }}
+        .up-to-date {{ color: var(--color-a); }}
+        .outdated {{ color: var(--color-d); }}
+        .vuln-none {{ color: var(--color-a); }}
+        .vuln-some {{ color: var(--color-f); font-weight: 600; }}
+        .severity-critical td {{ background: #fef2f2; }}
+        .severity-high td {{ background: #fff7ed; }}
+        .recommendations {{ background: white; border-radius: 8px; padding: 1.5rem; box-shadow: 0 1px 3px rgba(0,0,0,0.1); }}
+        .recommendations ul {{ padding-left: 1.5rem; }}
+        .recommendations li {{ margin: 0.5rem 0; }}
+        .meta {{ color: #6b7280; font-size: 0.875rem; margin-top: 2rem; }}
+    </style>
+</head>
+<body>
+    <div class="container">
+        <div class="header">
+            <div>
+                <h1>Codeshift Health Report</h1>
+                <p>{html.escape(str(score.project_path))}</p>
+            </div>
+            <div class="score-card">
+                <div class="grade {grade_class}">{score.grade.value}</div>
+                <div class="overall-score">{score.overall_score:.1f}/100</div>
+                {trend_info}
+            </div>
+        </div>
+
+        <section>
+            <h2>Metrics Breakdown</h2>
+            <table>
+                <thead>
+                    <tr>
+                        <th>Category</th>
+                        <th>Score</th>
+                        <th>Weight</th>
+                        <th>Details</th>
+                    </tr>
+                </thead>
+                <tbody>
+                    {metrics_rows}
+                </tbody>
+            </table>
+        </section>
+
+        <section class="recommendations">
+            <h2>Recommendations</h2>
+            <ul>
+                {recs_html if recs_html else "<li>No recommendations - your project is in great shape!</li>"}
+            </ul>
+        </section>
+
+        <section>
+            <h2>Dependencies ({len(score.dependencies)})</h2>
+            <table>
+                <thead>
+                    <tr>
+                        <th>Package</th>
+                        <th>Current</th>
+                        <th>Latest</th>
+                        <th>Status</th>
+                        <th>Migration Support</th>
+                        <th>Vulnerabilities</th>
+                    </tr>
+                </thead>
+                <tbody>
+                    {deps_rows if deps_rows else "<tr><td colspan='6'>No dependencies found</td></tr>"}
+                </tbody>
+            </table>
+        </section>
+
+        {vulns_html}
+
+        <p class="meta">
+            Generated by Codeshift on {score.calculated_at.strftime('%Y-%m-%d %H:%M:%S')}
+        </p>
+    </div>
+</body>
+</html>
+"""
+
+
+def save_html_report(report: HealthReport | HealthScore, output_path: Path) -> None:
+    """Save an HTML report to a file.
+
+    Args:
+        report: HealthReport or HealthScore to render
+        output_path: Path to save the report
+    """
+    html_content = generate_html_report(report)
+    output_path.write_text(html_content)
+
+
+def _json_serializer(obj: Any) -> Any:
+    """Custom JSON serializer for non-standard types."""
+    if isinstance(obj, datetime):
+        return obj.isoformat()
+    if isinstance(obj, Path):
+        return str(obj)
+    raise TypeError(f"Object of type {type(obj)} is not JSON serializable")
+
+
+def _get_score_class(score: float) -> str:
+    """Get CSS class based on score."""
+    if score >= 90:
+        return "excellent"
+    elif score >= 80:
+        return "good"
+    elif score >= 70:
+        return "fair"
+    elif score >= 60:
+        return "poor"
+    else:
+        return "critical"
+
+
+def _format_category(category: str) -> str:
+    """Format category name for display."""
+    return category.replace("_", " ").title()

codeshift/knowledge/generator.py

@@ -172,6 +172,12 @@ TIER_1_LIBRARIES = {
     "pytest",
     "marshmallow",
     "flask",
+    "celery",
+    "httpx",
+    "aiohttp",
+    "click",
+    "attrs",
+    "django",
 }
 
 

codeshift/knowledge_base/libraries/aiohttp.yaml

@@ -69,7 +69,7 @@ breaking_changes:
 
   # connector_owner default change
   - symbol: "ClientSession(connector=..., connector_owner=None)"
-    change_type: behavior_change
+    change_type: behavior_changed
     severity: medium
     from_version: "3.7"
     to_version: "3.9"
@@ -133,7 +133,7 @@ breaking_changes:
 
   # Middleware signature changes (old-style to new-style)
   - symbol: "@middleware"
-    change_type: signature_change
+    change_type: signature_changed
     severity: high
     from_version: "3.7"
     to_version: "3.9"
@@ -154,7 +154,7 @@ breaking_changes:
     transform_name: ws_connect_timeout_rename
 
   - symbol: "ws_connect(receive_timeout=...)"
-    change_type: behavior_change
+    change_type: behavior_changed
     severity: low
     from_version: "3.7"
     to_version: "3.9"

codeshift/knowledge_base/libraries/httpx.yaml

@@ -119,7 +119,7 @@ breaking_changes:
 
   # Response.iter_lines() behavior change
   - symbol: "Response.iter_lines()"
-    change_type: behavior_change
+    change_type: behavior_changed
     severity: medium
     from_version: "0.23"
     to_version: "0.24"
@@ -130,7 +130,7 @@ breaking_changes:
 
   # NetRC authentication change
   - symbol: "trust_env=True"
-    change_type: behavior_change
+    change_type: behavior_changed
     severity: medium
     from_version: "0.23"
     to_version: "0.24"
@@ -141,7 +141,7 @@ breaking_changes:
 
   # Query parameter encoding change
   - symbol: "params encoding"
-    change_type: behavior_change
+    change_type: behavior_changed
     severity: low
     from_version: "0.23"
     to_version: "0.24"
@@ -173,7 +173,7 @@ breaking_changes:
 
   # Follow redirects default change
   - symbol: "follow_redirects"
-    change_type: behavior_change
+    change_type: behavior_changed
     severity: high
     from_version: "0.19"
     to_version: "0.20"

codeshift/knowledge_base/libraries/pytest.yaml

@@ -182,7 +182,7 @@ breaking_changes:
 
   # pytest.importorskip exc_type
   - symbol: "pytest.importorskip()"
-    change_type: behavior_change
+    change_type: behavior_changed
     severity: low
     from_version: "8.0"
     to_version: "8.2"

codeshift/knowledge_base/models.py

@@ -9,6 +9,7 @@ class ChangeType(Enum):
 
     RENAMED = "renamed"
     REMOVED = "removed"
+    MOVED = "moved"
     SIGNATURE_CHANGED = "signature_changed"
     BEHAVIOR_CHANGED = "behavior_changed"
     DEPRECATED = "deprecated"