forked from rajpratik71/algorithm-exercise
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathlintcode.py
More file actions
95 lines (79 loc) · 2.8 KB
/
Copy pathlintcode.py
File metadata and controls
95 lines (79 loc) · 2.8 KB
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
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# from pyquery import PyQuery as pq
import requests
class Lintcode(object):
def __init__(self):
self.driver = None
def open_url(self, url):
print('open URL: {}'.format(url))
url = url.strip('description')
url = url.strip('/')
self.url = url
lintcode_unique_name = url.split('/')[-1]
req_url = 'https://www.lintcode.com/api/problems/detail/?unique_name_or_alias={}&_format=detail'.format(lintcode_unique_name)
self.driver = requests.get(req_url).json()
def get_title(self):
print('get title...')
title = self.driver['title']
return title
def get_description(self):
print('get description...')
desc = self.driver['description']
notice = self.driver['notice']
clarification = self.driver['clarification']
example = self.driver['example']
challenge = self.driver['challenge']
desc_full = desc
if notice:
desc_full += '\n\n#### Notice\n\n' + notice
if clarification:
desc_full += '\n\n#### Clarification\n\n' + clarification
if example:
desc_full += '\n\n#### Example\n\n' + example
if challenge:
desc_full += '\n\n#### Challenge\n\n' + challenge
return desc_full
def get_difficulty(self):
print('get difficulty...')
mapping = {1: 'Easy', 2: 'Medium', 3: 'Hard'}
difficulty = mapping.get(self.driver['level'], 'unknown')
return difficulty
def get_tags(self):
print('get tags...')
tags = []
for i in self.driver['tags']:
if i['alias']:
tags.append(i['alias'])
else:
tags.append(i['name'])
return tags
def _get_related(self):
print('get related...')
related = self.driver('.m-t-lg:last')
return related
def _clean_url(self, url):
new_url = ['http:/', 'www.lintcode.com', 'en/problem']
problem_slug = url[len('http://'):].strip('/').split('/')[3]
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.url
}
return problem
if __name__ == '__main__':
url = 'https://www.lintcode.com/problem/topological-sorting'
lintcode = Lintcode()
print(lintcode.get_problem_all(url))