保存工具代码

Author: diguage

tools/1-add-title-and-url.py

@@ -11,6 +11,7 @@
 import random
 import time
 from re import search
+from datetime import datetime
 
 
 def get_problems(json_data):
@@ -26,7 +27,7 @@ def get_data(url):
     r = requests.get(url)
     if r.status_code == 200:
         result = r.text
-        f = open('./leetcode.json', 'w')
+        f = open('./leetcode-%s.json'%(datetime.today().strftime('%Y%m%d')), 'w')
         f.write(result)
         f.close
         return result
@@ -40,15 +41,12 @@ def get_data(url):
 for problem in problems:
     id = problem["stat"]["frontend_question_id"]
 
-    if id < 3251:
-        continue
-
     title = "= %d. %s" %(id, problem["stat"]["question__title"])
     url = "https://leetcode.com/problems/%s/[LeetCode - %d. %s ^]" % (problem["stat"]["question__title_slug"], id, problem["stat"]["question__title"])
-    filename = "%04d-%s.adoc"%(problem["stat"]["frontend_question_id"], problem["stat"]["question__title_slug"])
+    filename = "%04d-%s.adoc"%(id, problem["stat"]["question__title_slug"])
 
     f = open("../docs/"+filename, "a")
-    f.write("[#%04d-%s]\n"%(problem["stat"]["frontend_question_id"], problem["stat"]["question__title_slug"]))
+    f.write("[#%04d-%s]\n"%(id, problem["stat"]["question__title_slug"]))
     f.write(title)
     f.write("\n\n")
     f.write(url)

tools/2-fetch-problems.py

@@ -48,12 +48,23 @@ def fetch_problem(title_slug):
     else:
         return None
 
+# 如果是锁定题目，则 data.question.content 为 null；
+def fetch_translations(title_slug):
+    url = 'https://leetcode.cn/graphql/'
+    headers = {'content-type': 'application/json'}
+    data = '{"query":"\n    query questionTranslations($titleSlug: String!) {\n  question(titleSlug: $titleSlug) {\n    translatedTitle\n    translatedContent\n  }\n}\n    ","variables":{"titleSlug":"%s"},"operationName":"questionTranslations"}' % title_slug
+    response = requests.post(url, headers=headers, data=data)
+    if response.status_code == 200:
+        return response.text
+    else:
+        return None
+
 problem_list = get_problems()
 
 for problem in problem_list:
     id = problem["stat"]["frontend_question_id"]
     titleSlug = problem["stat"]["question__title_slug"]
-    filename = "%04d-%s.adoc"%(problem["stat"]["frontend_question_id"], titleSlug)
+    filename = "%04d-%s.adoc"%(id, titleSlug)
 
     filepath = "../docs/"+filename
     problem_file = filepath+".json"

tools/7-download-images.js

@@ -1,5 +1,5 @@
 a = [];
-document.querySelectorAll('div.w-full > div.break-words > div > div > div > div > img').forEach(e => a.push(e.src));
+document.querySelectorAll('div.h-full.w-full > div  div > div.w-full > div.break-words > div > div  img').forEach(e => a.push(e.src));
 console.log("count=0\nfor f in '" + a.join("'  '") + "'\ndo\n  ((count = count + 1))\n  cnt=$(printf \"%02d\" count)\n  wget $f -O \"-$cnt.png\"\ndone");
 
 // 从 LeetCode 下载图片