diff --git a/src/api/api.py b/src/api/api.py index 62154f5..e021f18 100644 --- a/src/api/api.py +++ b/src/api/api.py @@ -40,7 +40,11 @@ async def _fetch_all_questions(self): questionId questionFrontendId title - titleSlug + titleSlug + difficulty + paidOnly: isPaidOnly + hasSolution + hasVideoSolution } } }""" @@ -58,7 +62,7 @@ async def _fetch_all_questions(self): for q in questions: self.questions[q["questionId"]] = q self.slug_to_id[q["titleSlug"]] = q["questionId"] - self.frontend_id_to_slug[q["questionFrontendId"]] = q["titleSlug"] + self.frontend_id_to_slug[q["questionFrontendId"]] = q["titleSlug"] except Exception as e: print(f"Error updating questions: {e}") @@ -90,7 +94,11 @@ async def get_all_problems(): "frontend_id": q["questionFrontendId"], "title": q["title"], "title_slug": q["titleSlug"], - "url": f"https://leetcode.com/problems/{q['titleSlug']}/" + "url": f"https://leetcode.com/problems/{q['titleSlug']}/", + "difficulty": q["difficulty"], + "paid_only": q["paidOnly"], + "has_solution": q["hasSolution"], + "has_video_solution": q["hasVideoSolution"], } for q in cache.questions.values()] @app.get("/problem/{id_or_slug}", tags=["Problems"])