FilesExpand file tree

 master

/

lintcode.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

95 lines (79 loc) · 2.8 KB

 master

/

lintcode.py

Copy path

Top

File metadata and controls

• Code
• Blame

95 lines (79 loc) · 2.8 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

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))