Merge pull request #148 from ghinks/feat/cache-clearing

feat: add cache management CLI options and remove unused functions

Author: ghinks

reviewtally/cache/sqlite_cache.py

@@ -224,36 +224,6 @@ def set_pr_review(  # noqa: PLR0913
 
         conn.commit()
 
-    def delete_pr_review(
-        self,
-        owner: str,
-        repo: str,
-        pull_number: int,
-    ) -> bool:
-        """
-        Delete a PR review cache entry.
-
-        Args:
-            owner: Repository owner
-            repo: Repository name
-            pull_number: Pull request number
-
-        Returns:
-            True if entry was deleted, False if not found
-
-        """
-        conn = self._get_connection()
-
-        cursor = conn.execute(
-            """
-            DELETE FROM pr_reviews_cache
-            WHERE owner = ? AND repo = ? AND pull_number = ?
-        """,
-            (owner, repo, pull_number),
-        )
-        conn.commit()
-        return cursor.rowcount > 0
-
     # PR Metadata cache methods
 
     def get_pr_metadata(
@@ -344,36 +314,6 @@ def set_pr_metadata(  # noqa: PLR0913
 
         conn.commit()
 
-    def delete_pr_metadata(
-        self,
-        owner: str,
-        repo: str,
-        pr_number: int,
-    ) -> bool:
-        """
-        Delete a PR metadata cache entry.
-
-        Args:
-            owner: Repository owner
-            repo: Repository name
-            pr_number: Pull request number
-
-        Returns:
-            True if entry was deleted, False if not found
-
-        """
-        conn = self._get_connection()
-
-        cursor = conn.execute(
-            """
-            DELETE FROM pr_metadata_cache
-            WHERE owner = ? AND repo = ? AND pr_number = ?
-        """,
-            (owner, repo, pr_number),
-        )
-        conn.commit()
-        return cursor.rowcount > 0
-
     def get_pr_metadata_date_range(
         self,
         owner: str,

reviewtally/cli/parse_cmd_line.py

@@ -30,6 +30,9 @@ class CommandLineArgs(TypedDict):
     plot_individual: bool
     individual_chart_metric: str
     use_cache: bool
+    clear_cache: bool
+    clear_expired_cache: bool
+    show_cache_stats: bool
 
 
 def print_toml_version() -> None:
@@ -164,6 +167,21 @@ def parse_cmd_line() -> CommandLineArgs:  # noqa: C901, PLR0912, PLR0915
         action="store_true",
         help="Disable PR review caching (always fetch fresh data from API)",
     )
+    parser.add_argument(
+        "--clear-cache",
+        action="store_true",
+        help="Clear all cached data and exit",
+    )
+    parser.add_argument(
+        "--clear-expired-cache",
+        action="store_true",
+        help="Clear only expired cached data and exit",
+    )
+    parser.add_argument(
+        "--cache-stats",
+        action="store_true",
+        help="Show cache statistics and exit",
+    )
 
     args = parser.parse_args()
     # catch ValueError if the date format is not correct
@@ -236,4 +254,7 @@ def parse_cmd_line() -> CommandLineArgs:  # noqa: C901, PLR0912, PLR0915
         plot_individual=args.plot_individual,
         individual_chart_metric=args.individual_chart_metric,
         use_cache=not args.no_cache,
+        clear_cache=args.clear_cache,
+        clear_expired_cache=args.clear_expired_cache,
+        show_cache_stats=args.cache_stats,
     )

reviewtally/main.py

@@ -1,12 +1,14 @@
 from __future__ import annotations
 
+import sys
 import time
 from typing import Any
 
 from tqdm import tqdm
 
 from reviewtally.analysis.sprint_periods import calculate_sprint_periods
 from reviewtally.analysis.team_metrics import calculate_sprint_team_metrics
+from reviewtally.cache.cache_manager import CacheManager
 from reviewtally.cli.parse_cmd_line import CommandLineArgs, parse_cmd_line
 from reviewtally.data_collection import (
     ProcessRepositoriesContext,
@@ -32,6 +34,15 @@ def main() -> None:
     # Parse command line arguments
     args = parse_cmd_line()
 
+    # Handle cache management operations
+    if (
+        args["clear_cache"]
+        or args["clear_expired_cache"]
+        or args["show_cache_stats"]
+    ):
+        _handle_cache_operations(args)
+        sys.exit(0)
+
     # Get repositories
     timestamped_print(
         f"Calling get_repos_by_language {time.time() - start_time:.2f} "
@@ -191,5 +202,45 @@ def _handle_sprint_plotting(context: SprintPlottingContext) -> None:
         print(f"Plotting failed: {e}")  # noqa: T201
 
 
+def _handle_cache_operations(args: CommandLineArgs) -> None:
+    """Handle cache management operations."""
+    cache_manager = CacheManager()
+
+    if args["show_cache_stats"]:
+        if cache_manager.cache:
+            stats = cache_manager.cache.get_stats()
+            print("Cache Statistics:")  # noqa: T201
+            print(f"  Database path: {stats['db_path']}")  # noqa: T201
+            print(f"  Total entries: {stats['total_entries']}")  # noqa: T201
+            print(f"  Valid entries: {stats['valid_entries']}")  # noqa: T201
+            print(f"  Expired entries: {stats['expired_entries']}")  # noqa: T201
+            print(f"  Cache size: {stats['cache_size_mb']} MB")  # noqa: T201
+            print(f"  Database size: {stats['db_size_mb']} MB")  # noqa: T201
+            print("\nBy Table:")  # noqa: T201
+            for table_name, table_stats in stats["by_table"].items():
+                print(f"  {table_name}:")  # noqa: T201
+                print(f"    Total: {table_stats['total']}")  # noqa: T201
+                print(f"    Valid: {table_stats['valid']}")  # noqa: T201
+                print(f"    Expired: {table_stats['expired']}")  # noqa: T201
+                size_mb = table_stats["size_bytes"] / (1024 * 1024)
+                print(f"    Size: {size_mb:.2f} MB")  # noqa: T201
+        else:
+            print("Cache is disabled")  # noqa: T201
+
+    if args["clear_expired_cache"]:
+        if cache_manager.cache:
+            removed = cache_manager.cache.cleanup_expired()
+            print(f"Cleared {removed} expired cache entries")  # noqa: T201
+        else:
+            print("Cache is disabled")  # noqa: T201
+
+    if args["clear_cache"]:
+        if cache_manager.cache:
+            removed = cache_manager.cache.clear_all()
+            print(f"Cleared all {removed} cache entries")  # noqa: T201
+        else:
+            print("Cache is disabled")  # noqa: T201
+
+
 if __name__ == "__main__":
     main()

tests/cli/test_parse_cmd_line.py

@@ -212,8 +212,8 @@ def test_valid_dates_no_exit(self, mock_argv: Any, mock_exit: Any) -> None:
         # Assert
         mock_exit.assert_not_called()
         self.assertIsInstance(result, dict)
-        # Check it's the right type
-        self.assertEqual(len(result), 14)
+        # Check it's the right type (14 original + 3 cache options = 17)
+        self.assertEqual(len(result), 17)
 
         # Verify the parsed dates
         self.assertEqual(result["org_name"], "test-org")