Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly
Appearance settings

Expand file tree

/

leetcode.py

Copy path
More file actions
More file actions

Latest commit

 

History

History
History
78 lines (63 loc) · 2.29 KB
/

leetcode.py

Copy path

File metadata and controls

78 lines (63 loc) · 2.29 KB
Raw
Copy raw file
Download raw file
Open symbols panel
Edit and raw actions
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
#!/usr/bin/env python
# -*- coding: utf-8 -*-
from selenium import webdriver
class Leetcode(object):
def __init__(self):
chrome_options = webdriver.ChromeOptions()
chrome_options.add_argument('--headless')
chrome_options.add_argument('--disable-gpu')
self.driver = webdriver.Chrome(chrome_options=chrome_options)
def open_url(self, url):
self.url = url
print('open URL: {}'.format(url))
self.driver.get(url)
def teardown(self):
self.driver.close()
def get_title(self):
print('get title...')
raw_title = self.driver.title
title = raw_title[:-len(' - LeetCode')].strip()
return title
def get_description(self):
print('get description...')
elem = self.driver.find_element_by_class_name('question-description')
return elem.get_attribute('innerHTML')
def get_difficulty(self):
print('get difficulty...')
elem = self.driver.find_element_by_class_name('difficulty-label')
return elem.get_attribute('innerHTML')
def get_tags(self):
print('get tags...')
tags_id = self.driver.find_element_by_id('tags-topics')
tags_id_a = tags_id.find_elements_by_tag_name('a')
tags = []
for i in tags_id_a:
tag = i.get_attribute('innerHTML')
tags.append(tag)
return tags
def _clean_url(self, url):
new_url = ['https:/', 'leetcode.com', 'problems']
problem_slug = url[len('https://'):].strip('/').split('/')[2]
new_url.append(problem_slug)
return '/'.join(new_url)
def get_problem_all(self, url):
"""获取所有细节"""
print('get all the problem detail...')
self.open_url(url)
title = self.get_title()
difficulty = self.get_difficulty()
tags = self.get_tags()
description = self.get_description()
problem = {
'title': title,
'difficulty': difficulty,
'tags': tags,
'description': description,
'url': self._clean_url(url)
}
self.teardown()
return problem
if __name__ == '__main__':
url = 'https://leetcode.com/problems/palindrome-number'
leetcode = Leetcode()
print(leetcode.get_problem_all(url))
Morty Proxy This is a proxified and sanitized view of the page, visit original site.