-
Notifications
You must be signed in to change notification settings - Fork 1
/
leetcode.py
171 lines (133 loc) · 8.23 KB
/
leetcode.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
"""
@project: PyCharm
@file: leetcode.py
@author: CompetitiveLin
@time: 2023/09/01 10:58
@mail: [email protected]
"""
import sys
import requests
from requests.adapters import HTTPAdapter
requests.packages.urllib3.disable_warnings()
import os
import json
import time
# 登录
def login(LEETCODE_SESSION):
session = requests.Session() # 建立会话
session.mount('http://', HTTPAdapter(max_retries=6)) # 超时重试次数
session.mount('https://', HTTPAdapter(max_retries=6)) # 超时重试次数
session.encoding = "utf-8" # 编码格式
session.cookies.set('LEETCODE_SESSION', LEETCODE_SESSION)
return session
# 获取某个题目的提交记录
def get_submission_list(slug, session):
url = 'https://leetcode.cn/graphql/'
payload = json.dumps({
"operationName": "submissions",
"variables": {
"offset": 0,
"limit": 40,
"lastKey": "null",
"questionSlug": slug
},
"query": "query submissions($offset: Int!, $limit: Int!, $lastKey: String, $questionSlug: String!) {\n submissionList(offset: $offset, limit: $limit, lastKey: $lastKey, questionSlug: $questionSlug) {\n lastKey\n hasNext\n submissions {\n id\n statusDisplay\n lang\n runtime\n timestamp\n url\n isPending\n memory\n __typename\n }\n __typename\n }\n}\n"
})
headers = {"content-type": "application/json", "origin": "https://leetcode.cn", "referer": "https://leetcode.cn/progress/", "user-agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_12_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/81.0.4044.122 Safari/537.36"}
# 发起请求，失败重试10次，每次间隔2秒
for try_count in range(0, 10):
try:
r = session.post(url, data=payload, headers=headers, verify=False)
response_data = json.loads(r.text)
return response_data
except:
print("获取题目为{}的提交记录失败，重试中，重试次数：{}/10".format(slug, try_count + 1))
time.sleep(2)
continue
# 走到这里说明出异常了
raise Exception("获取题目为{}的提交记录失败，请检查错误".format(slug))
# 获取所有通过的题目列表
def get_accepted_problems(session):
url = 'https://leetcode.cn/graphql/'
payload = json.dumps({
"operationName": "userProfileQuestions",
"variables": {
"status": "ACCEPTED",
"skip": 0,
"first": 200000, # 一次返回的数据量
"sortField": "LAST_SUBMITTED_AT",
"sortOrder": "DESCENDING",
"difficulty": [
]
},
"query": "query userProfileQuestions($status: StatusFilterEnum!, $skip: Int!, $first: Int!, $sortField: SortFieldEnum!, $sortOrder: SortingOrderEnum!, $keyword: String, $difficulty: [DifficultyEnum!]) {\n userProfileQuestions(status: $status, skip: $skip, first: $first, sortField: $sortField, sortOrder: $sortOrder, keyword: $keyword, difficulty: $difficulty) {\n totalNum\n questions {\n translatedTitle\n frontendId\n titleSlug\n title\n difficulty\n lastSubmittedAt\n numSubmitted\n lastSubmissionSrc {\n sourceType\n ... on SubmissionSrcLeetbookNode {\n slug\n title\n pageId\n __typename\n }\n __typename\n }\n __typename\n }\n __typename\n }\n}\n"
})
headers = {"content-type": "application/json", "origin": "https://leetcode.cn", "referer": "https://leetcode.cn/progress/", "user-agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_12_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/81.0.4044.122 Safari/537.36"}
r = session.post(url, data=payload, headers=headers, verify=False)
response_data = json.loads(r.text)
return response_data['data']['userProfileQuestions']
# 生成Markdown文本
def generate_markdown_text(response_data, session):
response_data = response_data['questions']
# 相关介绍
markdown_text = "## 相关介绍\n\n"
markdown_text += "这是一个 [LeetCode](https://leetcode.cn/problemset/all/) 题目自动统计及分析程序，可自动统计所有提交通过的题目，并以 Markdown 的形式展示。\n\n"
markdown_text += "根据个人需求，目前只考虑获取**提交次数**和**重刷次数**这两个指标，以便更好地进行刷题。\n\n"
markdown_text += "采用 GitHub Actions 进行自动化部署，无需本地服务器资源。\n\n"
markdown_text += "## 使用教程\n\n"
markdown_text += "1. 由模板项目[生成](https://github.com/CompetitiveLin/leetcode-revise/generate)自己的项目\n\n"
markdown_text += "2. 点击生成项目下的 Settings -> Secrets -> Actions -> New repository secret，分别添加以下 Secrets：\n"
markdown_text += " - Name：`LEETCODE_SESSION`\n"
markdown_text += " - Secret：已登录 LeetCode 账号的浏览器中 Cookie 名为 LEETCODE_SESSION 的值\n\n"
markdown_text += "3. 回到项目首页并进入 Actions 部分，在左侧点击`Github LeetCode Bot`，再点击蓝色提示框中的 `Run workflow`，最后点击绿色的 `Run workflow` 按钮即可\n\n"
markdown_text += "## 补充说明\n\n"
markdown_text += "默认配置为每12小时更新一次，可根据需求修改 [action.yml](.github/workflows/action.yml#L9) 文件的第 `9` 行。如有其他需求，欢迎提交PR。\n\n"
markdown_text += "> 重刷次数的计算规则为：累计所有提交通过且互为不同一天的记录次数\n\n"
markdown_text += "## 数据统计\n\n"
markdown_text += "| 最近提交时间 | 题目 | 题目难度 | 提交次数| 重刷次数 |\n| ---- | ---- | ---- | ---- | ---- |\n"
for index, sub_data in enumerate(response_data):
# 显示进度
print('{}/{}'.format(index + 1, len(response_data)))
# 获取一些必要的信息
lastSubmittedAt = time.strftime("%Y-%m-%d %H:%M", time.localtime(sub_data['lastSubmittedAt']))
translatedTitle = "#{} {}".format(sub_data['frontendId'], sub_data['translatedTitle'])
frontendId = sub_data['frontendId']
difficulty = sub_data['difficulty']
titleSlug = sub_data['titleSlug']
numSubmitted = sub_data['numSubmitted']
numSubmitted = str(numSubmitted)
url = "https://leetcode.cn/problems/{}".format(sub_data['titleSlug'])
# 获取重刷次数
# 规则定义如下：提交通过的时间 与 之前提交通过的时间 不为同一天 即认为是重新刷了一遍
submission_dict = get_submission_list(titleSlug, session)
submission_list = submission_dict['data']['submissionList']['submissions']
submission_accepted_dict = {}
for submission in submission_list:
status = submission['statusDisplay']
if (status == 'Accepted'):
submission_time = time.strftime("%Y-%m-%d", time.localtime(int(submission['timestamp'])))
if submission_time in submission_accepted_dict.keys():
submission_accepted_dict[submission_time] += 1
else:
submission_accepted_dict[submission_time] = 1
# 重刷次数
count = len(submission_accepted_dict)
if count > 1:
count = "**" + str(count) + "**"
else:
count = str(count)
# 更新Markdown文本
markdown_text += "| " + lastSubmittedAt + " | " + "[" + translatedTitle + "]" + "(" + url + ")" + " | " + difficulty + " | " + numSubmitted + " | " + count + " |" + "\n"
return markdown_text
if __name__ == '__main__':
session = login(sys.argv[1]) # 登录，参数为已登录 LeetCode 账号的浏览器中 Cookie 名为 LEETCODE_SESSION 的值
response_data = get_accepted_problems(session) # 获取所有通过的题目列表
if response_data:
markdown_text = generate_markdown_text(response_data, session) # 生成Markdown文本
# 更新README.md文件
with open("README.md", "w") as f:
f.write(markdown_text)
else:
raise Exception("Cookies 已过期，请更新 LEETCODE_SESSION")