Add --language script argument

`--language` argument allows to download only submissions in the specified programming languages

Author: NeverMendel

leetcode_export/__main__.py

@@ -6,10 +6,11 @@
 
 from leetcode_export._version import __version__
 from leetcode_export.leetcode import LeetCode
+from leetcode_export.utils import VALID_PROGRAMMING_LANGUAGES
 
 
 def parse_args():
-    parser = argparse.ArgumentParser(description='Export LeetCode solutions')
+    parser = argparse.ArgumentParser(description='Export LeetCode solutions', formatter_class=argparse.RawTextHelpFormatter)
     parser.add_argument('--cookies', type=str, help='set LeetCode cookies')
     parser.add_argument('--folder', type=str, default='.', help='set output folder')
     parser.add_argument('--problem-filename', type=str, default='${question_id} - ${title_slug}.html',
@@ -22,14 +23,30 @@ def parse_args():
                         default='${date_formatted} - ${status_display} - runtime ${runtime} - memory ${memory}.${extension}',
                         help='submission filename format')
     parser.add_argument('--only-accepted', dest='only_accepted', action='store_true', help='save accepted submissions only')
+    parser.add_argument('--language', dest='language_unprocessed', type=str,
+                        help="only save submissions for the specified programming languages (eg. '--language=python,python3,cpp,java,golang')\n" +
+                        "supported languages: 'python', 'python3', 'c', 'cpp', 'csharp', 'java', 'kotlin', 'mysql', 'mssql', 'oraclesql',\n"+
+                        "                     'javascript', 'html', 'php', 'golang', 'scala', 'pythonml', 'rust', 'ruby', 'bash', 'swift'")
     parser.add_argument('-v', '--verbose', dest='verbose', action='store_true', help='enable verbose logging details')
     parser.add_argument('-vv', '--extra-verbose', dest='extra_verbose', action='store_true',
                         help='enable more verbose logging details')
     parser.add_argument('-V', '--version', action='version',
                         version='%(prog)s {version}'.format(version=__version__))
     parser.set_defaults(verbose=False, extra_verbose=False)
 
-    return parser.parse_args()
+    args = parser.parse_args()
+
+    if args.language_unprocessed:
+        languages = args.language_unprocessed.split(',')
+        args.language = [lang.strip() for lang in languages]
+        for lang in languages:
+            if lang not in VALID_PROGRAMMING_LANGUAGES:
+                parser.error(f"Invalid language: '{lang}'")
+        args.language = languages
+    else:
+        args.language = None
+
+    return args
 
 
 def main():
@@ -79,6 +96,9 @@ def main():
         if args.only_accepted and submission.status_display != 'Accepted':
             continue
 
+        if args.language and submission.lang not in args.language:
+            continue
+
         if not os.path.exists(submission.title_slug):
             os.mkdir(submission.title_slug)
         os.chdir(submission.title_slug)

leetcode_export/leetcode.py

@@ -115,6 +115,6 @@ def get_submissions(self) -> Iterator[Submission]:
                     yield submission
 
             current += 20
-            sleep(1)
+            sleep(5)
         if 'detail' in response_json:
             logging.warning(response_json['detail'])

leetcode_export/utils.py

@@ -1,6 +1,8 @@
 import re
 from typing import Dict
 
+VALID_PROGRAMMING_LANGUAGES = ['python', 'python3', 'c', 'cpp', 'csharp', 'java', 'kotlin', 'mysql', 'mssql', 'oraclesql', 'javascript', 'html', 'php', 'golang', 'scala', 'pythonml', 'rust', 'ruby', 'bash', 'swift']
+
 FILE_EXTENSIONS = {
     "python": 'py',
     "python3": 'py',