', '', dify)
74 | dify = dify.replace('', '')
75 |
76 | prob[probID] = dify
77 | return dify
78 |
79 | except:
80 | print('Met a problem at {}'.format(probID))
81 | dify = '难度未知'
82 |
83 | prob[probID] = dify
84 | return dify
85 |
86 | else:
87 | return prob[probID]
88 |
89 | # ========== 获取个人信息页面 ==========
90 |
91 | def getUser(UserID):
92 |
93 | if config['hasStatisticsed'] == False:
94 | UserWeb = 'https://www.luogu.org/space/show?uid={}'.format(UserID)
95 | req = request.get(UserWeb)
96 | html = req.text
97 | base = req.content
98 | writeFile('log/U{}.html'.format(UserID), 'wb+', base)
99 | return html
100 |
101 | else:
102 | file = open('log/U{}.html'.format(UserID), 'r+', encoding='utf8')
103 | content = file.read()
104 | file.close()
105 | return content
106 |
107 | # ========== 列出已经AC的题目 ==========
108 |
109 | def listAC(UserID):
110 |
111 | print('List AC problems of U{}'.format(UserID))
112 |
113 | html = getUser(UserID)
114 | html = re.findall(r'[\s\S]*?
', html)[0]
115 | html = html.replace('\n', '')
116 | html = html.replace('
通过题目
\n', '')
117 | html = html.replace('\n', '')
118 | html = re.sub(r'\[', '', html)
119 | html = re.sub(r'\]', '', html)
120 | prob = re.split(r'\n', html)
121 |
122 | writeFile('log/AC.txt', 'w+', ", ".join(str(it) for it in prob))
123 |
124 | return prob
125 |
126 | # ========== 按题号统计题目难度 ==========
127 |
128 | def statistics(prob):
129 |
130 | ans = []
131 | for probID in prob:
132 | ans.append(getProblem(probID))
133 |
134 | out = ", ".join(str(it) for it in ans)
135 | writeFile('log/ans.txt', 'w+', out)
136 |
137 | return ans
138 |
139 | # ========== 统计输出结果 ==========
140 |
141 | def modify(set, content):
142 | result = content
143 | for it in re.findall(r'{\|[\s\S]*?\|}', content):
144 | try:
145 | # print('Modify "{}"'.format(it[2:-2]))
146 | result = result.replace(it, set[it[2:-2]])
147 | except:
148 | result = result.replace(it, '0')
149 | return result
150 |
151 | def modifyFile(oldPath, newPath, set):
152 | file = open(oldPath, 'r+', encoding='utf8')
153 | content = file.read()
154 | file.close()
155 | content = modify(set, content)
156 | file = open(newPath, 'w+', encoding='utf8')
157 | file.write(content)
158 | file.close()
159 |
160 | def result(UserID, ans):
161 |
162 | set = {}
163 | html = getUser(UserID)
164 |
165 | set['UserID'] = str(UserID)
166 | set['UserWeb'] = 'https://www.luogu.org/space/show?uid={}'.format(UserID)
167 | set['UserName'] = strFind('U{} '.format(UserID), '
', html)
168 | set['Submit'] = re.sub(r'<[\s\S]*?>', '', strFind('', '提交', html))
169 | set['SubmitReal'] = isThousand(set['Submit'])
170 | set['Accept'] = re.sub(r'<[\s\S]*?>', '', strFind('
91 |