FilesExpand file tree

 master

/

lintcode.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

78 lines (63 loc) · 2.17 KB

 master

/

lintcode.py

Copy path

Top

File metadata and controls

• Code
• Blame

78 lines (63 loc) · 2.17 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 pyquery import PyQuery as pq

class Lintcode(object):

def __init__(self):

self.driver = None

def open_url(self, url):

self.url = url

print('open URL: {}'.format(url))

self.driver = pq(url=url)

def get_title(self):

print('get title...')

title = self.driver('title').text()

return title

def get_description(self):

print('get description...')

desc_pq = self.driver('#description')

desc_html = desc_pq('.m-t-lg:nth-child(1)').html()

example_html = desc_pq('.m-t-lg:nth-child(2)').html()

return desc_html + example_html

def get_difficulty(self):

print('get difficulty...')

progress_bar = self.driver('.progress-bar')

original_title = progress_bar.attr('data-original-title')

splits = original_title.strip().split(' ')

difficulty = splits[1]

ac_rate = splits[-1]

return difficulty

def get_tags(self):

print('get tags...')

tags = []

for i in self.driver('#tags.tags a'):

tags.append(i.text)

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._clean_url(url)

}

return problem

if __name__ == '__main__':

url = 'http://www.lintcode.com/en/problem/palindrome-number/'

leetcode = Lintcode()

print(leetcode.get_problem_all(url))