feat(api): fetch and parse LeetCode submission details.

Author: arnav-on-code

leetcode/api.py

@@ -1,22 +1,29 @@
 from config.settings import Config
-from leetcode.parser import parse_submissions
+
+from leetcode.parser import (
+    parse_submissions,
+    parse_submission_detail,
+)
+
 from leetcode.queries import (
     PROFILE_QUERY,
     RECENT_SUBMISSIONS_QUERY,
+    SUBMISSION_DETAILS_QUERY,
 )
 
 
 class LeetCodeAPI:
-
     def __init__(self, client):
         self.client = client
 
     def get_profile(self):
+        """Fetch the user's profile and solved statistics."""
+
         response = self.client.post(
             PROFILE_QUERY,
             {
                 "username": Config.LEETCODE_USERNAME
-            }
+            },
         )
 
         if "errors" in response:
@@ -25,6 +32,7 @@ def get_profile(self):
         return response["data"]["matchedUser"]
 
     def get_recent_submissions(self, limit=15):
+        """Fetch the user's recent accepted submissions."""
 
         response = self.client.post(
             RECENT_SUBMISSIONS_QUERY,
@@ -39,4 +47,22 @@ def get_recent_submissions(self, limit=15):
 
         return parse_submissions(
             response["data"]["recentAcSubmissionList"]
+        )
+
+    def get_submission_detail(self, submission_id):
+        """Fetch detailed information about a submission."""
+
+        response = self.client.post(
+            SUBMISSION_DETAILS_QUERY,
+            {
+                "submissionId": int(submission_id)
+            },
+        )
+
+        if "errors" in response:
+            raise Exception(response["errors"])
+
+        return parse_submission_detail(
+            submission_id,
+            response["data"]["submissionDetails"],
         )

leetcode/models.py

@@ -9,6 +9,33 @@ class Submission:
     slug: str
     timestamp: int
 
+    @property
+    def date(self):
+        return datetime.fromtimestamp(self.timestamp)
+
+
+@dataclass(slots=True)
+class SubmissionDetail:
+    submission_id: str
+
+    question_id: str
+    title_slug: str
+
+    language: str
+    language_verbose: str
+
+    runtime: int
+    runtime_display: str
+
+    memory: int
+    memory_display: str
+
+    status_code: int
+
+    code: str
+
+    timestamp: int
+
     @property
     def date(self):
         return datetime.fromtimestamp(self.timestamp)

leetcode/parser.py

@@ -1,7 +1,9 @@
-from leetcode.models import Submission
+from leetcode.models import Submission, SubmissionDetail
 
 
 def parse_submissions(data):
+    """Parse recent submissions."""
+
     submissions = []
 
     for item in data:
@@ -10,8 +12,34 @@ def parse_submissions(data):
                 id=item["id"],
                 title=item["title"],
                 slug=item["titleSlug"],
-                timestamp=int(item["timestamp"])
+                timestamp=int(item["timestamp"]),
             )
         )
 
-    return submissions
+    return submissions
+
+
+def parse_submission_detail(submission_id, data):
+    """Parse GraphQL response into a SubmissionDetail object."""
+
+    return SubmissionDetail(
+        submission_id=str(submission_id),
+
+        question_id=data["question"]["questionId"],
+        title_slug=data["question"]["titleSlug"],
+
+        language=data["lang"]["name"],
+        language_verbose=data["lang"]["verboseName"],
+
+        runtime=data["runtime"],
+        runtime_display=data["runtimeDisplay"],
+
+        memory=data["memory"],
+        memory_display=data["memoryDisplay"],
+
+        status_code=data["statusCode"],
+
+        code=data["code"],
+
+        timestamp=int(data["timestamp"]),
+    )

leetcode/queries.py

@@ -14,6 +14,7 @@
 }
 """
 
+
 RECENT_SUBMISSIONS_QUERY = """
 query recentAcSubmissions($username: String!, $limit: Int!) {
     recentAcSubmissionList(
@@ -26,4 +27,33 @@
         timestamp
     }
 }
+"""
+
+
+SUBMISSION_DETAILS_QUERY = """
+query submissionDetails($submissionId: Int!) {
+    submissionDetails(submissionId: $submissionId) {
+
+        runtime
+        runtimeDisplay
+
+        memory
+        memoryDisplay
+
+        code
+        timestamp
+
+        statusCode
+
+        lang {
+            name
+            verboseName
+        }
+
+        question {
+            questionId
+            titleSlug
+        }
+    }
+}
 """

main.py

@@ -109,6 +109,29 @@ def main():
     if new_submissions:
         detector.update(new_submissions[0])
 
+    detail = api.get_submission_detail(
+    submissions[0].id
+    )
 
+    console.print("\n[bold cyan]Latest Submission Details[/bold cyan]\n")
+
+    console.print(f"Question ID : {detail.question_id}")
+    console.print(f"Slug        : {detail.title_slug}")
+    console.print(f"Language    : {detail.language}")
+    console.print(f"Runtime     : {detail.runtime_display}")
+    console.print(f"Memory      : {detail.memory_display}")
+
+
+    STATUS_MAP = {
+    10: "Accepted",
+    11: "Wrong Answer",
+    12: "Memory Limit Exceeded",
+    13: "Output Limit Exceeded",
+    14: "Time Limit Exceeded",
+    15: "Runtime Error",
+    16: "Internal Error",
+    20: "Compile Error",
+}
+    console.print(f"Status      : {STATUS_MAP.get(detail.status_code, 'Unknown')}")
 if __name__ == "__main__":
     main()