feat: format

Author: yuvrajsinh5252

src/commands/__init__.py

@@ -1,4 +1,4 @@
-from .show import show
 from .profile import profile
+from .show import show
 
-__all__ = ["show", "profile"]
+__all__ = ["show", "profile"]

src/commands/daily.py

@@ -1,12 +1,18 @@
 import typer
-from .edit import edit
+
 from ..server.api import get_daily_question
+from .edit import edit
+
 
 def daily(
     lang: str = typer.Argument("py", help="Programming language to use."),
-    editor: str = typer.Option("vim", '--editor', '-e', help="Code editor to use."),
-    full: bool = typer.Option(False, "--full", "-f", help="Show full problem description"),
-    save: bool = typer.Option(False, "--save", "-s", help="Save problem description to a file"),
+    editor: str = typer.Option("vim", "--editor", "-e", help="Code editor to use."),
+    full: bool = typer.Option(
+        False, "--full", "-f", help="Show full problem description"
+    ),
+    save: bool = typer.Option(
+        False, "--save", "-s", help="Save problem description to a file"
+    ),
     no_editor: bool = typer.Option(False, "--no-editor", help="Skip opening editor"),
 ):
     """
@@ -17,25 +23,33 @@ def daily(
     """
     from .show import show
 
-    if editor not in ['code', 'vim', 'nano']:
+    if editor not in ["code", "vim", "nano"]:
         typer.echo(typer.style(f"❌ Unsupported editor: {editor}", fg=typer.colors.RED))
         raise typer.Exit(1)
 
-    typer.echo(typer.style("Fetching daily challenge...", fg=typer.colors.GREEN), nl=False)
+    typer.echo(
+        typer.style("Fetching daily challenge...", fg=typer.colors.GREEN), nl=False
+    )
     try:
         result = get_daily_question()
-        question = result['data']['activeDailyCodingChallengeQuestion']
+        question = result["data"]["activeDailyCodingChallengeQuestion"]
         typer.echo("\r" + " " * 30 + "\r", nl=False)
 
-        show(problem=question['question']['titleSlug'], save=save, compact=not full)
+        show(problem=question["question"]["titleSlug"], save=save, compact=not full)
     except Exception as e:
-        typer.echo("\n" + typer.style(f"❌ Failed to fetch daily question: {str(e)}", fg=typer.colors.RED))
+        typer.echo(
+            "\n"
+            + typer.style(
+                f"❌ Failed to fetch daily question: {str(e)}", fg=typer.colors.RED
+            )
+        )
         raise typer.Exit(1)
 
-
     if not no_editor and editor:
         try:
-            edit(problem=question['question']['titleSlug'], lang=lang, editor=editor)
+            edit(problem=question["question"]["titleSlug"], lang=lang, editor=editor)
         except Exception as e:
-            typer.echo(typer.style(f"❌ Failed to open editor: {str(e)}", fg=typer.colors.RED))
-            raise typer.Exit(1)
+            typer.echo(
+                typer.style(f"❌ Failed to open editor: {str(e)}", fg=typer.colors.RED)
+            )
+            raise typer.Exit(1)

src/commands/edit.py

@@ -1,37 +1,43 @@
 import typer
 
-from ..server.solution_manager import SolutionManager
 from ..server.auth import Auth
 from ..server.config import LANGUAGE_MAP
+from ..server.solution_manager import SolutionManager
 
 solution_manager = SolutionManager(Auth().get_session())
 
+
 def edit(
-  problem: str = typer.Argument(..., help="Problem name or id."),
-  lang: str = typer.Argument("cpp", help="Programming language to use."),
-  editor: str = typer.Option("vim", '-e', '--editor', help="Editor to use for code editing."),
+    problem: str = typer.Argument(..., help="Problem name or id."),
+    lang: str = typer.Argument("cpp", help="Programming language to use."),
+    editor: str = typer.Option(
+        "vim", "-e", "--editor", help="Editor to use for code editing."
+    ),
 ):
-  """Solves a problem by passing lang param and open it with your code editor."""
-  question_data = solution_manager.get_question_data(problem).get('data', {}).get('question')
-
-  if not question_data:
-    typer.echo(f"Problem {problem} not found.")
-    return
-
-  filename_prefix = question_data.get('questionFrontendId') or (
-    problem if problem.isdigit() else question_data.get('questionId')
-  )
-
-  def create_file_with_template(lang: str):
-    filename = f"{filename_prefix}.{lang}"
-    with open(filename, "w") as f:
-      for snippet in question_data.get('codeSnippets', []):
-          if snippet.get('langSlug').lower() == LANGUAGE_MAP.get(lang):
-            f.write(snippet.get('code'))
-            return
-      typer.echo(f"No template found for language {lang}")
-
-  create_file_with_template(lang)
-
-  import subprocess
-  subprocess.run([editor, f"{filename_prefix}.{lang}"])
+    """Solves a problem by passing lang param and open it with your code editor."""
+    question_data = (
+        solution_manager.get_question_data(problem).get("data", {}).get("question")
+    )
+
+    if not question_data:
+        typer.echo(f"Problem {problem} not found.")
+        return
+
+    filename_prefix = question_data.get("questionFrontendId") or (
+        problem if problem.isdigit() else question_data.get("questionId")
+    )
+
+    def create_file_with_template(lang: str):
+        filename = f"{filename_prefix}.{lang}"
+        with open(filename, "w") as f:
+            for snippet in question_data.get("codeSnippets", []):
+                if snippet.get("langSlug").lower() == LANGUAGE_MAP.get(lang):
+                    f.write(snippet.get("code"))
+                    return
+            typer.echo(f"No template found for language {lang}")
+
+    create_file_with_template(lang)
+
+    import subprocess
+
+    subprocess.run([editor, f"{filename_prefix}.{lang}"])

src/commands/list_problems.py

@@ -1,57 +1,73 @@
-
-from rich.spinner import Spinner
-from rich.live import Live
 from typing import Optional
-from ..lib.ui import display_problem_list
-from ..server.auth import Auth
-from rich.progress import Progress, SpinnerColumn, TextColumn
 
 import typer
+from rich.progress import Progress, SpinnerColumn, TextColumn
 
+from ..lib.ui import display_problem_list
 from ..server.api import fetch_problem_list
+from ..server.auth import Auth
 
 status_map = {
-  "attempted": "TRIED",
-  "solved": "AC",
-  "todo": "NOT_STARTED",
+    "attempted": "TRIED",
+    "solved": "AC",
+    "todo": "NOT_STARTED",
 }
 
 AuthManager = Auth()
 
+
 def list_problems(
-  difficulty: Optional[str] = typer.Option(None, "--difficulty", "-d", help="Filter by difficulty (easy/medium/hard)"),
-  status: Optional[str] = typer.Option(None, "--status", "-s", help="Filter by status (todo/attempted/solved)"),
-  tag: Optional[str] = typer.Option(None, "--tag", "-t", help="Filter by tags (comma-separated)"),
-  category_slug: Optional[str] = typer.Option("all-code-essentials", "--category-slug", "-c", help="Filter by category slug")
+    difficulty: Optional[str] = typer.Option(
+        None, "--difficulty", "-d", help="Filter by difficulty (easy/medium/hard)"
+    ),
+    status: Optional[str] = typer.Option(
+        None, "--status", "-s", help="Filter by status (todo/attempted/solved)"
+    ),
+    tag: Optional[str] = typer.Option(
+        None, "--tag", "-t", help="Filter by tags (comma-separated)"
+    ),
+    category_slug: Optional[str] = typer.Option(
+        "all-code-essentials", "--category-slug", "-c", help="Filter by category slug"
+    ),
 ):
-  """List available LeetCode problems with optional filters."""
-  if (difficulty is not None or status is not None or tag is not None):
-    if not AuthManager.is_authenticated:
-      typer.echo(typer.style("❌ Please login first to enable problem listing with filters", fg=typer.colors.RED))
-      raise typer.Exit(1)
+    """List available LeetCode problems with optional filters."""
+    if difficulty is not None or status is not None or tag is not None:
+        if not AuthManager.is_authenticated:
+            typer.echo(
+                typer.style(
+                    "❌ Please login first to enable problem listing with filters",
+                    fg=typer.colors.RED,
+                )
+            )
+            raise typer.Exit(1)
 
-  tags = tag.split(',') if tag else []
+    tags = tag.split(",") if tag else []
 
-  with Progress(
+    with Progress(
         SpinnerColumn(),
         TextColumn("[progress.description]{task.description}"),
         transient=True,
     ) as progress:
-    progress.add_task("Fetching problems...", total=1)
-
-    session = AuthManager.session_manager.load_session()
-    if not session:
-        typer.echo(typer.style("❌ No valid session found. Please login first.", fg=typer.colors.RED))
-        raise typer.Exit(1)
-
-    data = fetch_problem_list(
-      csrf_token=session["csrftoken"],
-      session_id=session["session_token"],
-      categorySlug=category_slug or "all-code-essentials",
-      filters={
-        "difficulty": difficulty,
-        "status":  status_map.get(status) if status is not None else None,
-        "tags": tags
-      }
-    )
-  display_problem_list(data)
+        progress.add_task("Fetching problems...", total=1)
+
+        session = AuthManager.session_manager.load_session()
+        if not session:
+            typer.echo(
+                typer.style(
+                    "❌ No valid session found. Please login first.",
+                    fg=typer.colors.RED,
+                )
+            )
+            raise typer.Exit(1)
+
+        data = fetch_problem_list(
+            csrf_token=session["csrftoken"],
+            session_id=session["session_token"],
+            categorySlug=category_slug or "all-code-essentials",
+            filters={
+                "difficulty": difficulty,
+                "status": status_map.get(status) if status is not None else None,
+                "tags": tags,
+            },
+        )
+    display_problem_list(data)

src/commands/login.py

@@ -1,17 +1,24 @@
 import typer
+
 from ..server.auth import Auth
 
 auth_manager = Auth()
 
+
 def login():
     """Login to LeetCode"""
     if auth_manager.is_authenticated:
         saved_session = auth_manager.session_manager.load_session()
         if saved_session:
-            typer.echo(typer.style(f"Already logged in as {saved_session['user_name']}!", fg=typer.colors.GREEN))
+            typer.echo(
+                typer.style(
+                    f"Already logged in as {saved_session['user_name']}!",
+                    fg=typer.colors.GREEN,
+                )
+            )
             return
 
-    instruction = f"""
+    instruction = """
                     📝 To get your login tokens:
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
     1️  Open LeetCode in your browser
@@ -28,25 +35,40 @@ def login():
         csrf_result = auth_manager.verify_csrf_token(csrf_token)
 
         if not csrf_result["success"]:
-            typer.echo(typer.style(f"\n✗ CSRF verification failed: {csrf_result['message']}", fg=typer.colors.RED))
+            typer.echo(
+                typer.style(
+                    f"\n✗ CSRF verification failed: {csrf_result['message']}",
+                    fg=typer.colors.RED,
+                )
+            )
             return
 
         leetcode_session = typer.prompt("Please enter your LEETCODE_SESSION token")
         result = auth_manager.login_with_session(csrf_token, leetcode_session)
 
         if result["success"]:
-            typer.echo(typer.style(f"\n✓ Successfully logged in as {result['user_name']}!", fg=typer.colors.GREEN))
+            typer.echo(
+                typer.style(
+                    f"\n✓ Successfully logged in as {result['user_name']}!",
+                    fg=typer.colors.GREEN,
+                )
+            )
             break
         else:
-            typer.echo(typer.style(f"\n✗ Login failed: {result['message']}", fg=typer.colors.RED))
+            typer.echo(
+                typer.style(
+                    f"\n✗ Login failed: {result['message']}", fg=typer.colors.RED
+                )
+            )
             if not typer.confirm("Do you want to try again?", abort=True):
                 break
 
+
 def logout():
     """Logout from LeetCode"""
     if not auth_manager.is_authenticated:
         typer.echo(typer.style("❌ You are not logged in", fg=typer.colors.RED))
         return
 
     auth_manager.session_manager.clear_session()
-    typer.echo(typer.style("✓ Successfully logged out", fg=typer.colors.GREEN))
+    typer.echo(typer.style("✓ Successfully logged out", fg=typer.colors.GREEN))

src/commands/profile.py

@@ -1,11 +1,13 @@
-from ..server.api import fetch_user_profile
+from rich.live import Live
 from rich.spinner import Spinner
+
 from ..lib.ui import display_user_stats
-from rich.live import Live
+from ..server.api import fetch_user_profile
+
 
 def profile():
     """List all available LeetCode problems."""
-    spinner = Spinner('dots')
+    spinner = Spinner("dots")
     with Live(spinner, refresh_per_second=10, transient=True) as live:
         live.console.print("[cyan]Fetching user profile...")
         data = fetch_user_profile()