Merge branch 'master' into dev/list_id

Author: fspv

README.md

@@ -33,19 +33,38 @@ cd leetcode-anki
 ```
 
 After that initialize and activate python virtualenv somewhere
+
+Linux/MacOS
 ```
 virtualenv -p python leetcode-anki
 . leetcode-anki/bin/activate
 ```
 
+Windows
+```
+python -m venv leetcode-anki
+.\leetcode-anki\Scripts\activate.bat
+```
+
 Then initialize session id variable. You can get it directly from your browser (if you're using chrome, cookies can be found here chrome://settings/cookies/detail?site=leetcode.com)
+
+Linux/Macos
 ```
 export LEETCODE_SESSION_ID="yyy"
 ```
 
-And finally run
+Windows
+```
+set LEETCODE_SESSION_ID="yyy"
+```
+
+And finally run for Linux/MacOS
 ```
 make generate
 ```
+Or for Windows
+```
+python generate.py
+```
 
 You'll get `leetcode.apkg` file, which you can import directly to your anki app.

generate.py

@@ -7,6 +7,7 @@
 import argparse
 import asyncio
 import logging
+from pathlib import Path
 from typing import Any, Coroutine, List
 
 # https://github.com/kerrickstaley/genanki
@@ -47,6 +48,12 @@ def parse_args() -> argparse.Namespace:
         help="Get all questions from a specific list id (https://leetcode.com/list?selectedList=<list_id>",
         default="",
     )
+    parser.add_argument(
+        "--output-file",
+        type=str,
+        help="Output filename",
+        default=OUTPUT_FILE,
+    )
 
     args = parser.parse_args()
 
@@ -102,7 +109,7 @@ async def generate_anki_note(
     )
 
 
-async def generate(start: int, stop: int, page_size: int, list_id: str) -> None:
+async def generate(start: int, stop: int, page_size: int, list_id: str, output_file: str) -> None:
     """
     Generate an Anki deck
     """
@@ -167,7 +174,7 @@ async def generate(start: int, stop: int, page_size: int, list_id: str) -> None:
             },
         ],
     )
-    leetcode_deck = genanki.Deck(LEETCODE_ANKI_DECK_ID, "leetcode")
+    leetcode_deck = genanki.Deck(LEETCODE_ANKI_DECK_ID, Path(output_file).stem)
 
     leetcode_data = leetcode_anki.helpers.leetcode.LeetcodeData(start, stop, page_size, list_id)
 
@@ -188,7 +195,7 @@ async def generate(start: int, stop: int, page_size: int, list_id: str) -> None:
     for leetcode_note in tqdm(note_generators, unit="flashcard"):
         leetcode_deck.add_note(await leetcode_note)
 
-    genanki.Package(leetcode_deck).write_to_file(OUTPUT_FILE)
+    genanki.Package(leetcode_deck).write_to_file(output_file)
 
 
 async def main() -> None:
@@ -197,8 +204,8 @@ async def main() -> None:
     """
     args = parse_args()
 
-    start, stop, page_size, list_id = args.start, args.stop, args.page_size, args.list_id
-    await generate(start, stop, page_size, list_id)
+    start, stop, page_size, list_id, output_file = args.start, args.stop, args.page_size, args.list_id, args.output_file
+    await generate(start, stop, page_size, list_id, output_file)
 
 
 if __name__ == "__main__":

leetcode_anki/helpers/leetcode.py

@@ -353,7 +353,9 @@ async def tags(self, problem_slug: str) -> List[str]:
         List of the tags for this problem (string slugs)
         """
         data = self._get_problem_data(problem_slug)
-        return list(map(lambda x: x.slug, data.topic_tags))
+        tags = list(map(lambda x: x.slug, data.topic_tags))
+        tags.append(data.difficulty)
+        return tags
 
     async def freq_bar(self, problem_slug: str) -> float:
         """