Add frequency to cards

Author: fspv

generate.py

@@ -91,6 +91,7 @@ async def _get_problem_data(self, problem_slug: str) -> Dict[str, str]:
             query="""
                 query getQuestionDetail($titleSlug: String!) {
                   question(titleSlug: $titleSlug) {
+                    freqBar
                     questionId
                     questionFrontendId
                     boundTopicId
@@ -222,6 +223,10 @@ async def tags(self, problem_slug: str) -> List[str]:
         data = await self._get_problem_data(problem_slug)
         return list(map(lambda x: x.slug, data.topic_tags))
 
+    async def freq_bar(self, problem_slug: str) -> float:
+        data = await self._get_problem_data(problem_slug)
+        return data.freq_bar or 0
+
 
 class LeetcodeNote(genanki.Note):
     @property
@@ -265,7 +270,7 @@ async def generate_anki_note(
     leetcode_task_title: str,
     topic: str,
 ) -> LeetcodeNote:
-    return LeetcodeNote(
+    note = LeetcodeNote(
         model=leetcode_model,
         fields=[
             leetcode_task_handle,
@@ -286,10 +291,15 @@ async def generate_anki_note(
                     * 100
                 )
             ),
+            str(await leetcode_data.freq_bar(leetcode_task_handle)),
         ],
         tags=await leetcode_data.tags(leetcode_task_handle),
+        # FIXME: sort field doesn't work doesn't work
+        sort_field=str(await leetcode_data.freq_bar(leetcode_task_handle)).zfill(3),
     )
 
+    return note
+
 
 async def generate(start: int, stop: int) -> None:
     leetcode_model = genanki.Model(
@@ -308,6 +318,7 @@ async def generate(start: int, stop: int) -> None:
             {"name": "SubmissionsTotal"},
             {"name": "SubmissionsAccepted"},
             {"name": "SumissionAcceptRate"},
+            {"name": "Frequency"},
             # TODO: add hints
         ],
         templates=[
@@ -322,6 +333,11 @@ async def generate(start: int, stop: int) -> None:
                 ({{SumissionAcceptRate}}%)
                 <br/>
                 <b>Topic:</b> {{Topic}}<br/>
+                <b>Frequency:</b>
+                <progress value="{{Frequency}}" max="100">
+                {{Frequency}}%
+                </progress>
+                <br/>
                 <b>URL:</b>
                 <a href='https://leetcode.com/problems/{{Slug}}/'>
                     https://leetcode.com/problems/{{Slug}}/
@@ -352,7 +368,7 @@ async def generate(start: int, stop: int) -> None:
 
     note_generators: List[Coroutine[Any, Any, LeetcodeNote]] = []
 
-    for topic, leetcode_task_title, leetcode_task_handle in list(
+    for (topic, leetcode_task_title, leetcode_task_handle) in list(
         get_leetcode_task_handles()
     )[start:stop]:
         note_generators.append(

requirements.txt

@@ -1,4 +1,4 @@
-python-leetcode==1.0.9
+python-leetcode==1.0.10
 setuptools==57.5.0
 diskcache
 genanki