chore: update main script

Author: yanglbme

solution/main.py

@@ -5,28 +5,6 @@
 
 import requests
 
-new_content = """# [{}. {}]({})
-
-[English Version]({})
-
-## 题目描述
-
-<!-- 这里写题目描述 -->
-
-{}
-
-"""
-
-new_content_en = """# [{}. {}]({})
-
-[中文文档]({})
-
-## Description
-
-{}
-
-"""
-
 
 class LCSpider:
     graph_url = 'https://leetcode-cn.com/graphql'
@@ -66,7 +44,7 @@ def get_question_detail(self, question_title_slug):
             'Connection': 'keep-alive',
             'Content-Type': 'application/json',
             'Referer': 'https://leetcode-cn.com/problems/' + question_title_slug,
-            # leetcode-cn.com cookie here
+            # lc-cn cookie here
             # 'cookie': ''
         }
         self.session.post(url=LCSpider.graph_url,
@@ -79,7 +57,21 @@ def get_question_detail(self, question_title_slug):
             'variables': {
                 'titleSlug': question_title_slug
             },
-            'query': """query questionData($titleSlug: String!) {\n  question(titleSlug: $titleSlug) {\n    questionId\n    questionFrontendId\n    categoryTitle\n    boundTopicId\n    title\n    titleSlug\n    content\n    translatedTitle\n    translatedContent\n    isPaidOnly\n    difficulty\n    likes\n    dislikes\n    isLiked\n    similarQuestions\n    contributors {\n      username\n      profileUrl\n      avatarUrl\n      __typename\n    }\n    langToValidPlayground\n    topicTags {\n      name\n      slug\n      translatedName\n      __typename\n    }\n    companyTagStats\n    codeSnippets {\n      lang\n      langSlug\n      code\n      __typename\n    }\n    stats\n    hints\n    solution {\n      id\n      canSeeDetail\n      __typename\n    }\n    status\n    sampleTestCase\n    metaData\n    judgerAvailable\n    judgeType\n    mysqlSchemas\n    enableRunCode\n    envInfo\n    book {\n      id\n      bookName\n      pressName\n      source\n      shortDescription\n      fullDescription\n      bookImgUrl\n      pressImgUrl\n      productUrl\n      __typename\n    }\n    isSubscribed\n    isDailyQuestion\n    dailyRecordStatus\n    editorType\n    ugcQuestionId\n    style\n    exampleTestcases\n    __typename\n  }\n}\n"""
+            'query': 'query questionData($titleSlug: String!) {\n  question(titleSlug: $titleSlug) {\n    '
+                     'questionId\n    questionFrontendId\n    categoryTitle\n    boundTopicId\n    title\n    '
+                     'titleSlug\n    content\n    translatedTitle\n    translatedContent\n    isPaidOnly\n    '
+                     'difficulty\n    likes\n    dislikes\n    isLiked\n    similarQuestions\n    '
+                     'contributors {\n      username\n      profileUrl\n      avatarUrl\n      __typename\n    '
+                     '}\n    langToValidPlayground\n    topicTags {\n      name\n      slug\n      '
+                     'translatedName\n      __typename\n    }\n    companyTagStats\n    codeSnippets {\n      '
+                     'lang\n      langSlug\n      code\n      __typename\n    }\n    stats\n    hints\n    '
+                     'solution {\n      id\n      canSeeDetail\n      __typename\n    }\n    status\n    '
+                     'sampleTestCase\n    metaData\n    judgerAvailable\n    judgeType\n    mysqlSchemas\n    '
+                     'enableRunCode\n    envInfo\n    book {\n      id\n      bookName\n      pressName\n      '
+                     'source\n      shortDescription\n      fullDescription\n      bookImgUrl\n      '
+                     'pressImgUrl\n      productUrl\n      __typename\n    }\n    isSubscribed\n    '
+                     'isDailyQuestion\n    dailyRecordStatus\n    editorType\n    ugcQuestionId\n    style\n    '
+                     'exampleTestcases\n    __typename\n  }\n}\n'
         }
 
         # get question detail
@@ -97,7 +89,7 @@ def get_all_questions(self):
             'content-type': 'application/json',
             'user-agent': LCSpider.user_agent,
             'x-requested-with': 'XMLHttpRequest',
-            # leetcode.com cookie here
+            # lc cookie here
             # 'cookie': ''
         }
         resp = self.session.get(url='https://leetcode.com/api/problems/all/',
@@ -127,7 +119,7 @@ def get_all_questions(self):
 
             print(f'{frontend_question_id}. {question_title_en}')
             topic_tags = question_detail.get('topicTags')
-            
+
             item = {
                 'question_id': str(question['stat']['question_id']).zfill(4),
                 'frontend_question_id': frontend_question_id,
@@ -186,10 +178,10 @@ def generate_readme():
 
         # generate README.md
         items = []
-        table_cn = "\n|  题号  |  题解  |  标签  |  难度  | 备注 |\n| --- | --- | --- | --- | --- |"
+        table_cn = '\n|  题号  |  题解  |  标签  |  难度  | 备注 |\n| --- | --- | --- | --- | --- |'
         for item in sorted(md_table_cn, key=lambda x: x[0]):
-            items.append("\n|  {}  |  {}  |  {}  |  {}  |  {}  |".format(item[0], item[1], item[2], item[3], item[4]))
-        table_cn += "".join(items)
+            items.append(f'\n|  {item[0]}  |  {item[1]}  |  {item[2]}  |  {item[3]}  |  {item[4]}  |')
+        table_cn += ''.join(items)
 
         with open('./readme_template.md', 'r', encoding='utf-8') as f:
             readme_cn = f.read()
@@ -198,11 +190,11 @@ def generate_readme():
 
         # generate README_EN.md
         items = []
-        table_en = "\n|  #  |  Solution  |  Tags  |  Difficulty  |  Remark |\n| --- | --- | --- | --- | --- |"
+        table_en = '\n|  #  |  Solution  |  Tags  |  Difficulty  |  Remark |\n| --- | --- | --- | --- | --- |'
         for item in sorted(md_table_en, key=lambda x: x[0]):
-            items.append(
-                "\n|  {}  |  {}  |  {}  |  {}  |  {}  |".format(item[0], item[1], item[2], item[3], item[4]))
-        table_en += "".join(items)
+            items.append(f'\n|  {item[0]}  |  {item[1]}  |  {item[2]}  |  {item[3]}  |  {item[4]}  |')
+        table_en += ''.join(items)
+
         with open('./readme_template_en.md', 'r', encoding='utf-8') as f:
             readme_en = f.read()
         with open('./README_EN.md', 'w', encoding='utf-8') as f:
@@ -246,32 +238,32 @@ def generate_question_readme(self):
                     readme_template_cn = readme_cn
                     readme_template_en = readme_en
 
-                # generate lc problem readme cn
+                # generate lc-cn problem readme
                 with open(f'{path}/README.md', 'w', encoding='utf-8') as f1:
                     f1.write(readme_template_cn.format(int(item['frontend_question_id']),
-                                              item["title_cn"],
-                                              item['url_cn'],
-                                              item['relative_path_en'],
-                                              item['content_cn']))
+                                                       item["title_cn"],
+                                                       item['url_cn'],
+                                                       item['relative_path_en'],
+                                                       item['content_cn']))
 
-                # generate lc problem readme en
+                # generate lc-en problem readme
                 with open(f'{path}/README_EN.md', 'w', encoding='utf-8') as f2:
                     f2.write(readme_template_en.format(int(item['frontend_question_id']),
-                                              item["title_en"],
-                                              item['url_en'],
-                                              item['relative_path_cn'],
-                                              item['content_en']))
+                                                       item["title_en"],
+                                                       item['url_en'],
+                                                       item['relative_path_cn'],
+                                                       item['content_en']))
 
     @staticmethod
     def generate_summary():
         """generate summary files"""
-        summary_cn = ""
-        summary_en = ""
-        for file in os.listdir("./"):
+        summary_cn = ''
+        summary_en = ''
+        for file in os.listdir('./'):
             if os.path.isdir("./" + file) and file != '__pycache__':
-                summary_cn += ("\n- " + file + "\n")
-                summary_en += ("\n- " + file + "\n")
-                for sub in os.listdir("./" + file):
+                summary_cn += f'\n- {file}\n'
+                summary_en += f'\n- {file}\n'
+                for sub in os.listdir('./' + file):
                     sub = sub.replace('`', ' ')
                     enc = quote(sub)
                     summary_cn += f'  - [{sub}](/solution/{file}/{enc}/README.md)\n'
@@ -285,96 +277,13 @@ def generate_summary():
         with open('./summary_en.md', 'w', encoding='utf-8') as f:
             f.write(summary_en)
 
-    @staticmethod
-    def remove_empty_folders():
-        """remove empty sub folders"""
-        sub_folders = [str(i * 100).zfill(4) + '-' + str(i * 100 + 99).zfill(4) for i in range(100)]
-
-        for item in sub_folders:
-            if not os.path.isdir(f'./{item}'):
-                continue
-            files = os.listdir(f'./{item}')
-            for f in files:
-                if os.path.isdir(f'./{item}/{f}'):
-                    if not os.listdir(f'./{item}/{f}'):
-                        os.rmdir(f'./{item}/{f}')
-
-    @staticmethod
-    def replace_content():
-        with open('./result.json', 'r', encoding='utf-8') as f:
-            result = f.read()
-            result = json.loads(result)
-            mapper = {item['frontend_question_id']: item for item in result}
-        sub_folders = [str(i * 100).zfill(4) + '-' + str(i * 100 + 99).zfill(4) for i in range(100)]
-        for item in sub_folders:
-            if not os.path.isdir(f'./{item}'):
-                continue
-            files = os.listdir(f'./{item}')
-            for f in files:
-                if os.path.isdir(f'./{item}/{f}'):
-                    with open(f'./{item}/{f}/README.md', 'r', encoding='utf-8') as f1:
-                        readme = f1.read()
-                    with open(f'./{item}/{f}/README_EN.md', 'r', encoding='utf-8') as f2:
-                        readme_en = f2.read()
-                    question = mapper[f[:4]]
-                    b = new_content.format(int(question['frontend_question_id']),
-                                           question["title_cn"],
-                                           question['url_cn'],
-                                           question['relative_path_en'],
-                                           question['content_cn'])
-                    index_cn = readme.index("## 解法")
-                    a = b + readme[index_cn:]
-                    with open(f'./{item}/{f}/README.md', 'w', encoding='utf-8') as f1:
-                        f1.write(a)
-                    index_en = readme_en.index("## Solutions")
-                    b = new_content_en.format(int(question['frontend_question_id']),
-                                              question["title_en"],
-                                              question['url_en'],
-                                              question['relative_path_cn'],
-                                              question['content_en'])
-                    a = b + readme_en[index_en:]
-                    with open(f'./{item}/{f}/README_EN.md', 'w', encoding='utf-8') as f2:
-                        f2.write(a)
-
-    @staticmethod
-    def download_image():
-        sub_folders = [str(i * 100).zfill(4) + '-' + str(i * 100 + 99).zfill(4) for i in range(100)]
-        for item in sub_folders:
-            if not os.path.isdir(f'./{item}'):
-                continue
-            files = os.listdir(f'./{item}')
-            for f in files:
-                if os.path.isdir(f'./{item}/{f}'):
-                    with open(f'./{item}/{f}/README.md', 'r', encoding='utf-8') as f1:
-                        readme = f1.read()
-                    with open(f'./{item}/{f}/README_EN.md', 'r', encoding='utf-8') as f2:
-                        readme_en = f2.read()
-
-                    res_cn = re.findall(r'src="(.*?)"', readme, re.S) or []
-                    for url in res_cn:
-                        file_name = os.path.basename(url)
-                        new_url = f'https://cdn.jsdelivr.net/gh/doocs/leetcode@main/solution/{item}/{quote(f)}/images/{file_name}'
-                        readme = readme.replace(url, new_url)
-                    with open(f'./{item}/{f}/README.md', 'w', encoding='utf-8') as f1:
-                        f1.write(readme)
-
-                    res_en = re.findall(r'src="(.*?)"', readme_en, re.S) or []
-                    for url in res_en:
-                        file_name = os.path.basename(url)
-                        new_url = f'https://cdn.jsdelivr.net/gh/doocs/leetcode@main/solution/{item}/{quote(f)}/images/{file_name}'
-                        readme_en = readme_en.replace(url, new_url)
-                    with open(f'./{item}/{f}/README_EN.md', 'w', encoding='utf-8') as f1:
-                        f1.write(readme_en)
-
 
 if __name__ == '__main__':
     spider = LCSpider()
 
     spider.get_all_questions()
     spider.save_result()
-    
+
     spider.generate_readme()
     spider.generate_question_readme()
     spider.generate_summary()
-    # spider.replace_content()
-    # spider.download_image()