feat: problem filter, add edit command

Author: yuvrajsinh5252

src/commands/edit.py

@@ -0,0 +1,32 @@
+import typer
+
+from src.server.solution_manager import SolutionManager
+from src.server.auth import Auth
+from commands.test import map_lang
+
+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("code", '-e', help="Code editor to use."),
+):
+  """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
+
+  def create_file_with_template(lang: str):
+    with open(f"{question_data.get('titleSlug')}.{lang}", "w") as f:
+      for snippet in question_data.get('codeSnippets', []):
+          if snippet.get('langSlug').lower() == map_lang.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"{question_data.get('titleSlug')}.{lang}"])

src/commands/list_problems.py

@@ -19,7 +19,7 @@
 
 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/in-progress/done)"),
+  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")
 ):

src/main.py

@@ -1,6 +1,7 @@
 import profile
 import typer
 from src import show, profile
+from src.commands.edit import edit
 from src.commands.list_problems import list_problems
 from src.commands.login import login, logout
 from src.commands.submit import submit
@@ -15,6 +16,7 @@
 app.command(name="logout")(logout)
 app.command(name="submit")(submit)
 app.command(name="test")(test)
+app.command(name="edit")(edit)
 
 if __name__ == "__main__":
     app()

src/server/api.py

@@ -60,6 +60,9 @@ def fetch_problem_list(
     skip: int = 0,
     filters: dict = {}
 ):
+    if filters and 'difficulty' in filters:
+        filters['difficulty'] = filters['difficulty'].upper()
+
     client = create_leetcode_client(csrf_token, session_id)
     variables = {
         "categorySlug": categorySlug,