feat: problem stats

Author: yuvrajsinh5252

src/commands/show.py

@@ -8,8 +8,9 @@
 solution_manager = SolutionManager(auth_manager.get_session())
 console = Console()
 
-def show(problem: str = typer.Argument(..., help="Problem slug (e.g., 'two-sum')")):
+def show(problem: str = typer.Argument(..., help="Problem slug or number (e.g., 'two-sum' or '1')"), layout: bool = False):
     """Show problem details including description and test cases"""
+
     if not auth_manager.is_authenticated:
         typer.echo(typer.style("❌ Please login first using the login command", fg=typer.colors.RED))
         raise typer.Exit(1)
@@ -19,4 +20,7 @@ def show(problem: str = typer.Argument(..., help="Problem slug (e.g., 'two-sum')
         typer.echo(typer.style(f"❌ Problem '{problem}' not found", fg=typer.colors.RED))
         raise typer.Exit(1)
 
-    ProblemDetails(data.get('data', {}).get('question')).display()
+    if not layout:
+        ProblemDetails(data.get('data', {}).get('question')).display_full()
+    else:
+        ProblemDetails(data.get('data', {}).get('question')).display()

src/lib/problem_ui.py

@@ -1,19 +1,13 @@
 from rich.console import Console
-from rich.table import Table
 from rich.panel import Panel
 from rich.markdown import Markdown
 from rich.layout import Layout
 from rich.box import ROUNDED
-from rich.style import Style
-from rich.text import Text
-from rich.align import Align
 from bs4 import BeautifulSoup, Tag
 import json
 from dataclasses import dataclass
 from typing import List, Dict, Optional
 
-import typer
-
 console = Console()
 
 @dataclass
@@ -47,9 +41,13 @@ def __init__(self, problem_data: dict):
         except Exception:
             pass
 
-        self.total_accepted: int = problem_data.get('totalAccepted', 0)
-        self.total_submissions: int = problem_data.get('totalSubmissions', 0)
-        self.acceptance_rate: float = problem_data.get('acRate', 0)
+        # Parse stats from JSON string
+        self.stats = {}
+        try:
+            stats_str = problem_data.get('stats', '{}')
+            self.stats = json.loads(stats_str)
+        except Exception:
+            self.stats = {}
 
         self._parse_content()
         self.console_width = console.width
@@ -85,36 +83,6 @@ def _parse_content(self):
             else:
                 current_section.append(text)
 
-    @property
-    def available_languages(self) -> List[str]:
-        """Get list of available programming languages"""
-        return [snippet['lang'] for snippet in self.code_snippets]
-
-    @property
-    def formatted_function_signature(self) -> str:
-        """Get the formatted function signature"""
-        if not self.function_metadata:
-            return "[red]Error loading function signature[/]"
-
-        param_str = ', '.join(
-            f"{p['name']}: {p['type']}"
-            for p in self.function_metadata.params
-        )
-
-        return (
-            "[bold blue]Function Signature:[/]\n"
-            f"[bold cyan]def[/] [yellow]{self.function_metadata.name}[/](\n"
-            f"    [green]{param_str}[/]\n"
-            f") -> [green]{self.function_metadata.return_type}[/]"
-        )
-
-    def get_code_snippet(self, language: str) -> Optional[str]:
-        """Get code snippet for specific language"""
-        for snippet in self.code_snippets:
-            if snippet['lang'].lower() == language.lower():
-                return snippet['code']
-        return None
-
     def format_test_case(self, input_str: str, expected_str: str, case_num: int) -> str:
         return (
             f"[bold blue]Ex {case_num}:[/] "
@@ -220,15 +188,15 @@ def _format_test_cases(self):
 
     def _format_stats(self) -> str:
         """Format problem statistics"""
-        acceptance_rate = f"{self.acceptance_rate:.1f}%" if self.acceptance_rate else "N/A"
-        total_accepted = f"{self.total_accepted:,}" if self.total_accepted else "N/A"
-        total_submissions = f"{self.total_submissions:,}" if self.total_submissions else "N/A"
+        accepted = self.stats.get('totalAccepted', 'N/A')
+        submissions = self.stats.get('totalSubmission', 'N/A')
+        ac_rate = self.stats.get('acRate', 'N/A')
 
         return (
             "[bold blue]Problem Stats[/]\n\n"
-            f"[cyan]Acceptance Rate:[/] {acceptance_rate}\n"
-            f"[cyan]Total Accepted:[/] {total_accepted}\n"
-            f"[cyan]Total Submissions:[/] {total_submissions}"
+            f"[cyan]Acceptance Rate:[/] {ac_rate}\n"
+            f"[cyan]Total Accepted:[/] {accepted}\n"
+            f"[cyan]Total Submissions:[/] {submissions}"
         )
 
     def display(self):
@@ -281,4 +249,39 @@ def display(self):
             padding=(1, 2)
         ))
 
-        console.print(layout)
+        console.print(layout)
+
+    def display_only_description(self):
+        """Display only the problem description"""
+        console.clear()
+        console.print()
+
+        console.print(
+            Panel(
+                Markdown(self._format_description(self.content)),
+                box=ROUNDED,
+                title=str(self._create_header()),
+                border_style="blue",
+                padding=(1, 2)
+            )
+        )
+
+    def display_only_test_cases(self):
+        """Display only the test cases"""
+        console.clear()
+        console.print()
+
+        console.print(
+            Panel(
+                self._format_test_cases(),
+                box=ROUNDED,
+                title="[bold blue]Examples",
+                border_style="blue",
+                padding=(1, 2)
+            )
+        )
+
+    def display_full(self):
+        """Display the full problem details"""
+        self.display_only_description()
+        self.display_only_test_cases()

src/server/solution_manager.py

@@ -1,19 +1,27 @@
-import requests
 from typing import Dict, Any
-import json
 import time
-from bs4 import BeautifulSoup
-import html
-
-import typer
 
 class SolutionManager:
     def __init__(self, session):
         self.session = session
         self.BASE_URL = "https://leetcode.com"
 
-    def get_question_data(self, title_slug: str) -> Dict[str, Any]:
-        """Get question details using GraphQL"""
+    def get_question_data(self, question_identifier: str) -> Dict[str, Any]:
+        """Get question details using GraphQL
+        Args:
+            question_identifier: Can be either title slug (e.g. 'two-sum') or question number (e.g. '1')
+        """
+        if question_identifier.isdigit():
+            response = self.session.get(f"{self.BASE_URL}/api/problems/all/")
+            if response.status_code == 200:
+                problems = response.json().get('stat_status_pairs', [])
+                for problem in problems:
+                    if str(problem['stat']['frontend_question_id']) == question_identifier:
+                        question_identifier = problem['stat']['question__title_slug']
+                        break
+                else:
+                    return {"error": f"Question number {question_identifier} not found"}
+
         query = """
             query questionData($titleSlug: String!) {
                 question(titleSlug: $titleSlug) {
@@ -24,6 +32,7 @@ def get_question_data(self, title_slug: str) -> Dict[str, Any]:
                     difficulty
                     exampleTestcaseList
                     sampleTestCase
+                    stats
                     metaData
                     codeSnippets {
                         lang
@@ -38,38 +47,12 @@ def get_question_data(self, title_slug: str) -> Dict[str, Any]:
             f"{self.BASE_URL}/graphql",
             json={
                 "query": query,
-                "variables": {"titleSlug": title_slug}
+                "variables": {"titleSlug": question_identifier}
             }
         )
 
         return response.json()
 
-    def format_problem_details(self, data: Dict[str, Any]) -> Dict[str, Any]:
-        """Format problem details for display"""
-        question = data['data']['question']
-
-        # Parse HTML content
-        soup = BeautifulSoup(question['content'], 'html.parser')
-        content = soup.get_text('\n').strip()
-
-        # Parse metadata
-        metadata = json.loads(question.get('metaData', '{}'))
-
-        # Get example test cases
-        example_tests = question.get('exampleTestcases', '').split('\n')
-        example_tests = [test for test in example_tests if test]
-
-        return {
-            'id': question['questionId'],
-            'title': question['title'],
-            'difficulty': question['difficulty'],
-            'content': content,
-            'example_tests': example_tests,
-            'params': metadata.get('params', []),
-            'return_type': metadata.get('return', {}).get('type', 'Unknown'),
-            'code_snippets': question['codeSnippets']
-        }
-
     def submit_solution(self, title_slug: str, code: str, lang: str = "python3") -> Dict[str, Any]:
         """Submit a solution to LeetCode"""
         # First get the question ID