FilesExpand file tree

 master

/

ojhtml2markdown.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

71 lines (58 loc) · 2.05 KB

 master

/

ojhtml2markdown.py

Copy path

Top

File metadata and controls

• Code
• Blame

71 lines (58 loc) · 2.05 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

#!/usr/bin/env python3

"""Parse Leetcode/Lintcode html page to markdown."""

import frontmatter

import requests

import html2text

class YamlContent(object):

def __init__(self, metadata, content):

self.metadata_ = metadata

self.content_ = content

@property

def metadata(self):

return self.metadata_

@property

def content(self):

return self.content_

content = '# ' + title

yaml_content = YamlContent(metadata, content)

def leet_lint_url(url):

problem_slug = url.strip('/').split('/')[-1]

leetcode_url = 'https://leetcode.com/problems/{}/'.format(problem_slug)

lintcode_url = 'http://www.lintcode.com/en/problem/{}/'.format(problem_slug)

urls = {}

for url in [leetcode_url, lintcode_url]:

response = requests.head(url)

if response.status_code != 404:

if url.startswith('https://leetcode'):

urls['leetcode'] = url

elif url.startswith('http://www.lintcode'):

urls['lintcode'] = url

else:

print('cannot find url with: {}'.format(url))

return urls

def problem2md(problem):

metadata = {

'title': problem['title'],

'difficulty': problem['difficulty']

}

if problem['tags']:

metadata['tags'] = problem['tags']

description = problem['description']

h = html2text.HTML2Text()

description_md = h.handle(description)

lines = []

lines.append('# ' + problem['title'] + '\n')

lines.append('## Problem\n')

lines.append('### Metadata\n')

if problem['tags']:

lines.append('- tags: ' + ', '.join(problem['tags']))

lines.append('- difficulty: ' + problem['difficulty'])

urls = leet_lint_url(problem['url'])

for k, v in urls.items():

lines.append('- source({}): <{}>'.format(k, v))

lines.append('\n### Description\n')

lines.append(description_md)

content = '\n'.join(lines)

yaml_content = YamlContent(metadata, content)

problem_md = frontmatter.dumps(yaml_content, allow_unicode=True)

return problem_md