aicodestat 0.0.1__py3-none-any.whl

aicodestat-0.0.1.dist-info/METADATA

@@ -0,0 +1,110 @@
+Metadata-Version: 2.4
+Name: aicodestat
+Version: 0.0.1
+Summary: A local-first metrics tool that analyzes how you use AI coding assistants.
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: fastapi>=0.104.0
+Requires-Dist: uvicorn[standard]>=0.24.0
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: rich>=13.0.0
+Requires-Dist: questionary>=2.0.0
+Requires-Dist: httpx>=0.25.0
+Requires-Dist: mcp>=1.0.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.4.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
+
+## CodeStat · AI Code Metrics
+
+> Quantify how much AI actually contributes to your codebase.
+
+[![CI](https://img.shields.io/github/actions/workflow/status/2hangchen/CodeStat/ci.yml?style=flat-square&label=CI)](https://github.com/2hangchen/CodeStat/actions)
+[![License](https://img.shields.io/github/license/2hangchen/CodeStat?style=flat-square)](LICENSE)
+
+`CodeStat` is a local metrics tool that analyzes how you use AI coding assistants:
+how many lines are generated by AI, how many are kept, and how this evolves over time.
+
+> 中文文档见：[`README.zh-CN.md`](./README.zh-CN.md)
+
+---
+
+## Features
+
+- **Global dashboard for all data**  
+  - AI generated lines, adopted lines, adoption & generation rates  
+  - File count, session count, quick bar chart overview
+
+- **Multi‑dimension queries**  
+  - **By file**: see how much of a file comes from AI and how much you kept  
+  - **By session**: analyze one coding session with detailed diff lines  
+  - **By project**: aggregate metrics for an entire repository
+
+- **Agent / model comparison**  
+  - Compare multiple sessions (agents / models / settings) side‑by‑side  
+  - See which one actually produces more adopted code instead of just more tokens
+
+- **Local‑first & privacy‑friendly**  
+  - All metrics are computed locally from your own diffs  
+  - No source code or prompts are sent to any remote service
+
+- **Nice CLI UX**  
+  - Rich‑based tables & colors, arrow‑key navigation  
+  - Minimal but informative header (MCP status + repo info)
+
+---
+
+## Demo
+
+> TODO: add real screenshots / GIFs from your terminal
+
+- **Global dashboard**
+
+  *(insert GIF or screenshot here)*
+
+- **Session metrics with diff lines**
+
+  *(insert GIF or screenshot here)*
+
+---
+
+## Quickstart
+
+### Install
+
+```bash
+git clone https://github.com/2hangchen/CodeStat.git
+cd CodeStat
+pip install -r requirements.txt
+```
+
+> Once published to PyPI you can alternatively run:  
+> `pip install codestat-ai`
+
+### Start the CLI
+
+```bash
+python .\cli\main.py
+```
+
+Use `↑/↓` to move, `Enter` to confirm.  
+Choose **“📈 Global Dashboard (All Data)”** to see an overview of your local metrics.
+
+---
+
+## Typical Workflows
+
+- **Measure your own AI usage**  
+  - Record one or more coding sessions with your IDE + MCP server  
+  - Run `CodeStat` and inspect:
+    - AI generated vs adopted lines
+    - Which files receive the most AI help
+
+- **Compare agents / models / prompts**  
+  - Map different sessions to different agents / models  
+  - Use **Compare Agents** to get a per‑session comparison table
+
+- **Project‑level health check**  
+  - For a given repo, run project metrics to see:
+    - Where AI contributes the most
+    - Whether AI‑generated code is actually being kept

aicodestat-0.0.1.dist-info/RECORD

@@ -0,0 +1,34 @@
+config.py,sha256=EUE6AGyiO2-gaP2USqpO8MVMwDrrAwdqtaBTRz_HNZg,3829
+local_mcp_server.py,sha256=YTH4D_BH7BK7XWzhFrQgyxdPjlabNtQ0TPz9AcmcLIE,10901
+logging_config.py,sha256=_BQ8zNBM9jNDIASQZeX3pvTmitQbzXqgIvPIPiUy7II,1997
+main.py,sha256=LtsgvkCaSynwK4sGaDvCSd0ae2B0hBfQ3RJM9IT9ey0,5843
+service_manager.py,sha256=k1eEEioAlD44Z1zWJ5JHNv6sVkOdr92xsOgvkX0lYNg,6909
+cli/__init__.py,sha256=AqvbeRtWawDr_fC3owziHPI3LTZPEOMprtqhPUv6HzM,34
+cli/exporter.py,sha256=9eh3TKBY2VT1AScU2RTFEElYtEOtz_M7fmuR3LHr7Dc,3567
+cli/main.py,sha256=kpZxNMlb4fC1naY5PHj7zK3HfVFSrraRee3_zwItZLI,7072
+cli/menus.py,sha256=6JMBD0HT5DauB9eDsR5Hy6XOoo3nDAnibR-5Ki37lLs,19379
+cli/views.py,sha256=495tKFwW2PmKjDDCRweONKX_f3E3k6y-9WVBlJ1dXwg,9858
+compute/__init__.py,sha256=UZBI-hgsvGzb3BFH70Ce-QvWgxs5OvM4oQ1NzbDxGsY,25
+compute/cache.py,sha256=4U8bsFy0yBdq1bhsmjTnNTWFuuHK6h3_hrkGPmxxVC8,2417
+compute/diff_engine.py,sha256=ySYqGZfDDYfkTaKhA4BdYad4ACaeurWgWy3jwMbDE1I,2685
+compute/lcs_engine.py,sha256=zh6TJWZCzP8s-Tls2VHkVUfW2DJ63tZgSvatxtLCT78,2217
+compute/metrics_service.py,sha256=jHyQV5UfK6RpzCuAmq_C6UTknsynpKEc-aKFGQ-sLHQ,11128
+mcp/__init__.py,sha256=xoKlMDnbq-owd19TqBCnbAPudUyPHr3S2nGGrb-TghI,30
+mcp/agent_adapter.py,sha256=gBp4eaWZJ_z1XTynOsINrGLK6rR_AYC5IOWFYke6BKs,1777
+mcp/api_schemas.py,sha256=_ZkX5Q_yM-m447f2VcPqjsSAgR2cP4bn-m23oDyv1Ro,1304
+mcp/routes_after.py,sha256=ZuTWxgl7bmYxRqxX8VLvwiDRY_H7AtHh3R-6rhYxXNI,4374
+mcp/routes_before.py,sha256=NbNVVqZ5OnzsSY9I0gAYzDd-YFx9r01oAWrJVjtvEEo,2234
+mcp/routes_tools.py,sha256=1sgoaiVEw3LM6C_FE0WyzxLmssMXyH1k3NF_PPHcdSQ,4360
+storage/__init__.py,sha256=t8_StbWwdo4NOqdm0mDgisj3a1YNRET_qktZoKO_oDY,25
+storage/backup.py,sha256=UUV5-bazOV6Rs4-Vg39C4-dTdw4RQSqSvLYmVr3JxA0,5715
+storage/db.py,sha256=PF_ZkQcruYlMegdkCINOwjScS5FB_qVYKd_iilnOrqY,5522
+storage/models.py,sha256=-2x2loKCTVX4i56-9MWsU4NPDEUMx7llvmzseRarR-4,9842
+storage/scheduler.py,sha256=J9I2AHymduyaSG33bpZzxc4JfoIjRyYMNCkU2Sr0Wzc,3960
+utils/__init__.py,sha256=42ItHJEKhCz4PIrPFZnXNQV-hR0B5gJ9bb9dfW_uL-U,22
+utils/port_utils.py,sha256=2df9-gtSKT4yaQZapTZxkczvDDMBKL7gY5RPOTzQOxo,1907
+utils/time_utils.py,sha256=ruVPmc9pTIl_-HzG2uMaluoSJo5qRxExYRcJI84i02s,1126
+aicodestat-0.0.1.dist-info/METADATA,sha256=WJVZMlSmM2VohaqB3IyUXoxplqwvZ4DH6ai1GdxHzDQ,3383
+aicodestat-0.0.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+aicodestat-0.0.1.dist-info/entry_points.txt,sha256=TDZ9VLa7H2-5HXrc1HTEthOGdZL4B2HBt1cHfqzaJI4,149
+aicodestat-0.0.1.dist-info/top_level.txt,sha256=N8Qc1-jbxHDZ9jirkiVXwUSWxjMNySZxMaqd5_QeCx0,90
+aicodestat-0.0.1.dist-info/RECORD,,

aicodestat-0.0.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.10.2)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

aicodestat-0.0.1.dist-info/entry_points.txt

@@ -0,0 +1,5 @@
+[console_scripts]
+code-stat = main:main
+code-stat-cli = cli.main:main
+code-stat-mcp = mcp_stdio_server:main
+local-mcp-server = local_mcp_server:main

aicodestat-0.0.1.dist-info/top_level.txt

@@ -0,0 +1,10 @@
+cli
+compute
+config
+local_mcp_server
+logging_config
+main
+mcp
+service_manager
+storage
+utils

cli/__init__.py

@@ -0,0 +1,2 @@
+"""交互层（CLI）模块"""
+

cli/exporter.py

@@ -0,0 +1,111 @@
+"""Export query results to CSV/JSON"""
+import json
+import csv
+import logging
+from pathlib import Path
+from typing import Dict, Any, List
+from datetime import datetime
+from utils.time_utils import format_datetime
+
+logger = logging.getLogger(__name__)
+
+
+def export_to_json(data: Dict[str, Any], output_path: str) -> bool:
+    """
+    Export data as JSON format
+    
+    Args:
+        data: Data to export
+        output_path: Output file path
+    
+    Returns:
+        Whether successful
+    """
+    try:
+        output_file = Path(output_path)
+        output_file.parent.mkdir(parents=True, exist_ok=True)
+        
+        export_data = {
+            "export_time": format_datetime(datetime.now()),
+            "data": data
+        }
+        
+        with open(output_file, 'w', encoding='utf-8') as f:
+            json.dump(export_data, f, ensure_ascii=False, indent=2)
+        
+        logger.info(f"Data exported to JSON: {output_path}")
+        return True
+    except Exception as e:
+        logger.error(f"Failed to export to JSON: {e}")
+        return False
+
+
+def export_to_csv(metrics: Dict[str, Any], output_path: str) -> bool:
+    """
+    Export metrics data as CSV format
+    
+    Args:
+        metrics: Metrics data
+        output_path: Output file path
+    
+    Returns:
+        Whether successful
+    """
+    try:
+        output_file = Path(output_path)
+        output_file.parent.mkdir(parents=True, exist_ok=True)
+        
+        with open(output_file, 'w', newline='', encoding='utf-8') as f:
+            writer = csv.writer(f)
+            
+            # Write header row
+            writer.writerow(["Metric Name", "Value"])
+            
+            # Write metrics data
+            writer.writerow(["AI Generated Lines", metrics.get("ai_total_lines", 0)])
+            writer.writerow(["Adopted Lines", metrics.get("adopted_lines", 0)])
+            writer.writerow(["Code Adoption Rate (%)", metrics.get("adoption_rate", 0.0)])
+            writer.writerow(["Code Generation Rate (%)", metrics.get("generation_rate", 0.0)])
+            
+            if "file_count" in metrics:
+                writer.writerow(["Files Involved", metrics.get("file_count", 0)])
+            
+            if "session_count" in metrics:
+                writer.writerow(["Sessions", metrics.get("session_count", 0)])
+            
+            # If there are diff lines details, write them
+            diff_lines = metrics.get("diff_lines", [])
+            if diff_lines:
+                writer.writerow([])  # Empty row
+                writer.writerow(["Diff Lines Details"])
+                writer.writerow(["Diff Type", "Line Number", "Code Content"])
+                for diff_line in diff_lines:
+                    writer.writerow([
+                        diff_line.get("diff_type", ""),
+                        diff_line.get("line_number", ""),
+                        diff_line.get("line_content", "")
+                    ])
+        
+        logger.info(f"Data exported to CSV: {output_path}")
+        return True
+    except Exception as e:
+        logger.error(f"Failed to export to CSV: {e}")
+        return False
+
+
+def export_metrics(metrics: Dict[str, Any], output_path: str, format: str = "json") -> bool:
+    """
+    Export metrics data
+    
+    Args:
+        metrics: Metrics data
+        output_path: Output file path
+        format: Export format (json or csv)
+    
+    Returns:
+        Whether successful
+    """
+    if format.lower() == "csv":
+        return export_to_csv(metrics, output_path)
+    else:
+        return export_to_json(metrics, output_path)

cli/main.py

@@ -0,0 +1,213 @@
+"""CLI entry point, parses command line arguments"""
+import argparse
+import sys
+import logging
+from pathlib import Path
+
+# Add project root to Python path to ensure modules can be imported
+_project_root = Path(__file__).parent.parent
+if str(_project_root) not in sys.path:
+    sys.path.insert(0, str(_project_root))
+
+from config import get_cli_config
+from logging_config import setup_logging
+from storage.db import get_db
+from cli.menus import (
+    show_main_menu,
+    query_by_session,
+    query_by_file,
+    query_by_project,
+    compare_agents,
+    manage_data,
+    manage_service,
+    show_global_dashboard,
+)
+from cli.views import console
+from cli.exporter import export_metrics
+from compute.metrics_service import calculate_session_metrics, calculate_file_metrics, calculate_project_metrics
+import questionary
+
+logger = setup_logging(log_level="INFO", module_name="cli")
+
+
+def print_banner():
+    """Minimal banner (kept for future extension, currently no-op)."""
+    # Intentionally kept very light; main layout is handled in the menu.
+    console.print()
+
+
+def main():
+    """Main function"""
+    parser = argparse.ArgumentParser(description="Local MCP AI Code Statistics Tool CLI")
+    parser.add_argument(
+        "--session",
+        type=str,
+        help="Query by session ID (quick command)"
+    )
+    parser.add_argument(
+        "--file",
+        type=str,
+        help="Query by file path (quick command)"
+    )
+    parser.add_argument(
+        "--project",
+        type=str,
+        help="Query by project root directory (quick command)"
+    )
+    parser.add_argument(
+        "--export",
+        type=str,
+        help="Export format (json/csv), must be used with --session/--file/--project"
+    )
+    parser.add_argument(
+        "--output",
+        type=str,
+        help="Export file path, must be used with --export"
+    )
+    parser.add_argument(
+        "--start-service",
+        action="store_true",
+        help="Start MCP service (background)"
+    )
+    parser.add_argument(
+        "--stop-service",
+        action="store_true",
+        help="Stop MCP service"
+    )
+    parser.add_argument(
+        "--service-status",
+        action="store_true",
+        help="Check MCP service status"
+    )
+    
+    args = parser.parse_args()
+    
+    # Initialize database
+    try:
+        db = get_db()
+        db.initialize()
+    except Exception as e:
+        console.print(f"[red]❌ Database initialization failed: {e}[/red]")
+        sys.exit(1)
+    
+    # Handle service management commands
+    if args.start_service:
+        from service_manager import get_service_manager
+        manager = get_service_manager()
+        if manager.start(background=True):
+            console.print("[green]✅ MCP service started successfully[/green]")
+        else:
+            console.print("[red]❌ Failed to start MCP service[/red]")
+        return
+    
+    if args.stop_service:
+        from service_manager import get_service_manager
+        manager = get_service_manager()
+        if manager.stop():
+            console.print("[green]✅ MCP service stopped[/green]")
+        else:
+            console.print("[yellow]⚠️  MCP service is not running or failed to stop[/yellow]")
+        return
+    
+    if args.service_status:
+        from service_manager import get_service_manager
+        from cli.views import display_service_status
+        manager = get_service_manager()
+        status = manager.get_status()
+        display_service_status(status)
+        return
+    
+    # Quick command mode
+    if args.session:
+        try:
+            metrics = calculate_session_metrics(args.session)
+            from cli.views import display_metrics_table, display_session_info
+            if metrics.get("summaries"):
+                display_session_info(metrics["summaries"])
+            display_metrics_table(metrics)
+            
+            if args.export and args.output:
+                export_metrics(metrics, args.output, args.export)
+                console.print(f"[green]✅ Data exported to: {args.output}[/green]")
+        except Exception as e:
+            console.print(f"[red]❌ Query failed: {e}[/red]")
+            sys.exit(1)
+        return
+    
+    if args.file:
+        try:
+            metrics = calculate_file_metrics(args.file)
+            from cli.views import display_metrics_table
+            display_metrics_table(metrics)
+            
+            if args.export and args.output:
+                export_metrics(metrics, args.output, args.export)
+                console.print(f"[green]✅ Data exported to: {args.output}[/green]")
+        except Exception as e:
+            console.print(f"[red]❌ Query failed: {e}[/red]")
+            sys.exit(1)
+        return
+    
+    if args.project:
+        try:
+            metrics = calculate_project_metrics(args.project)
+            from cli.views import display_metrics_table
+            display_metrics_table(metrics)
+            
+            if args.export and args.output:
+                export_metrics(metrics, args.output, args.export)
+                console.print(f"[green]✅ Data exported to: {args.output}[/green]")
+        except Exception as e:
+            console.print(f"[red]❌ Query failed: {e}[/red]")
+            sys.exit(1)
+        return
+    
+    # Interactive menu mode
+    print_banner()
+    
+    while True:
+        try:
+            choice = show_main_menu()
+            
+            if choice == "exit":
+                console.print("\n[green]👋 Exiting tool, data is safely stored locally~[/green]")
+                break
+            elif choice == "overview":
+                show_global_dashboard()
+            elif choice == "file":
+                query_by_file()
+            elif choice == "session":
+                query_by_session()
+            elif choice == "project":
+                query_by_project()
+            elif choice == "compare":
+                compare_agents()
+            elif choice == "export":
+                # Export functionality is integrated in query functions
+                console.print("[yellow]Please select export option when querying[/yellow]")
+            elif choice == "service":
+                manage_service()
+            elif choice == "manage":
+                manage_data()
+            
+            # Ask if continue
+            if choice != "exit":
+                continue_choice = questionary.confirm("Continue querying?").ask()
+                if not continue_choice:
+                    console.print("\n[green]👋 Exiting tool, data is safely stored locally~[/green]")
+                    break
+                print()  # Empty line separator
+        
+        except KeyboardInterrupt:
+            console.print("\n[yellow]Operation cancelled[/yellow]")
+            break
+        except Exception as e:
+            console.print(f"[red]❌ Error occurred: {e}[/red]")
+            logger.error(f"CLI error: {e}", exc_info=True)
+            continue_choice = questionary.confirm("Continue?").ask()
+            if not continue_choice:
+                break
+
+
+if __name__ == "__main__":
+    main()