#!/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))