Use chrome headers when querying leetcode APIs

Author: NeverMendel

README.md

@@ -25,7 +25,7 @@ This script uses LeetCode REST and GraphQL APIs to download all your LeetCode su
 
 Before running the script, make sure that python3 is installed in your system.
 
-If you prefer, you can use the Docker image to download LeetCode submissions. For further instructions read the
+If you prefer, you can use the Docker image to download LeetCode submissions. For more information read the
 section [Docker Image](#docker-image).
 
 ## 🏁 Getting started <a name="getting-started"></a>

leetcode_export/__main__.py

@@ -157,7 +157,7 @@ def main():
 
     if not leetcode.set_cookies(cookies):
         logging.error(
-            "Cookies not valid. Copy them from the Network tab of your browser by clicking on any leetcode.com request and going in Request Headers > cookie. Check README.md file for further information"
+            "Cookies not valid. Copy them from the Network tab of your browser by clicking on any leetcode.com request and going in Request Headers > cookie. Check README.md file for more information"
         )
         exit(1)
 

leetcode_export/leetcode.py

@@ -11,22 +11,24 @@
 
 from leetcode_export.leetcode_graphql import GRAPHQL_URL, question_detail_json, Problem
 from leetcode_export.leetcode_rest import (
+    BASE_URL,
     LOGIN_URL,
     SUBMISSIONS_API_URL,
     Submission,
-    BASE_URL,
 )
 from leetcode_export.utils import (
+    dict_camelcase_to_snakecase,
     language_to_extension,
     remove_special_characters,
-    dict_camelcase_to_snakecase,
+    REQUEST_HEADERS,
 )
 
 
 class LeetCode(object):
     def __init__(self):
         logging.debug("LeetCode class instantiated")
         self.session = requests.Session()
+        self.session.headers.update(REQUEST_HEADERS)
         self.user_logged = False
         self.user_logged_expiration = datetime.datetime.now()
 
@@ -104,6 +106,7 @@ def is_user_logged(self) -> bool:
         cookie_dict = self.session.cookies.get_dict()
         if "csrftoken" in cookie_dict and "LEETCODE_SESSION" in cookie_dict:
             get_request = self.session.get(SUBMISSIONS_API_URL.format(0, 1))
+            logging.debug(get_request.text)
             sleep(1)  # cooldown time for get request
             if "detail" not in get_request.json():
                 logging.debug("User is logged in")
@@ -145,9 +148,9 @@ def get_submissions(self) -> Iterator[Submission]:
             and response_json["has_next"]
         ):
             logging.debug(f"Exporting submissions from {current} to {current + 20}")
-            response_json = self.session.get(
-                SUBMISSIONS_API_URL.format(current, 20)
-            ).json()
+            response = self.session.get(SUBMISSIONS_API_URL.format(current, 20))
+            logging.debug(response.content)
+            response_json = response.json()
             if "submissions_dump" in response_json:
                 for submission_dict in response_json["submissions_dump"]:
                     submission_dict["runtime"] = submission_dict["runtime"].replace(

leetcode_export/utils.py

@@ -63,6 +63,18 @@
 _regex_camelcase_to_snakecase1 = re.compile(r"(.)([A-Z][a-z]+)")
 _regex_camelcase_to_snakecase2 = re.compile("([a-z0-9])([A-Z])")
 
+REQUEST_HEADERS = {
+    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/122.0.0.0 Safari/537.36",
+    "Sec-Ch-Ua": '"Chromium";v="122", "Not(A:Brand";v="24", "Google Chrome";v="122"',
+    "Sec-Ch-Ua-Mobile": "?0",
+    "Sec-Ch-Ua-Platform": '"Windows"',
+    "Sec-Fetch-Dest": "document",
+    "Sec-Fetch-Mode": "navigate",
+    "Sec-Fetch-Site": "cross-site",
+    "Sec-Fetch-User": "?1",
+    "Upgrade-Insecure-Requests": "1",
+}
+
 
 def language_to_extension(language: str) -> str:
     """