FilesExpand file tree

 master

/

main.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

63 lines (52 loc) · 2.18 KB

 master

/

main.py

Copy path

Top

File metadata and controls

• Code
• Blame

63 lines (52 loc) · 2.18 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

#!/usr/bin/env python

# -*- coding: utf-8 -*-

import os

import argparse

from datetime import datetime

import frontmatter

from slugify import slugify

from util import par_dir, mkdir_p

from leetcode import Leetcode

from lintcode import Lintcode

from ojhtml2markdown import problem2md

BASEDIR = os.path.abspath(os.path.dirname(__file__))

def curr_time():

return datetime.now().strftime("%Y-%m-%d_%H-%M-%S")

if __name__ == '__main__':

parser = argparse.ArgumentParser(description='Helper for GitBook algorithm')

parser.add_argument('--new', type=str, dest='new',

help='create new post with given leetcode/lintcode url.')

parser.add_argument('--dir', type=str, dest='dir',

help='create md under dir.')

parser.add_argument('--update', nargs='*', dest='update',

help='update post with given title in post and summary.')

parser.add_argument('--migrate', type=str, dest='migrate',

help='migrate old posts.')

parser.add_argument('--fix-summary', dest='fix_summary',

help='render new summary from posts.')

args = parser.parse_args()

print('Called with arguments: {}'.format(args))

ROOTDIR = par_dir(BASEDIR)

raw_url = args.new

problem_md = ''

problem_slug = ''

xxxcode = None

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

xxxcode = Leetcode()

elif raw_url.startswith('http://www.lintcode.com'):

xxxcode = Lintcode()

problem = xxxcode.get_problem_all(raw_url)

problem_slug = slugify(problem['title'], separator="_")

problem_md = problem2md(problem)

if args.dir:

post_dir = os.path.join(ROOTDIR, args.dir)

post_fn = os.path.join(post_dir, problem_slug + '.md')

summary_path = args.dir.strip('/').split('/')[-1] + '/' + problem_slug + '.md'

summary_line = '* [{title}]({path})'.format(title=problem['title'], path=summary_path)

print(summary_line)

mkdir_p(post_dir)

with open(post_fn, 'w', encoding='utf-8') as f:

print('create post file {}...'.format(post_fn))

f.write(problem_md)

if args.fix_summary:

update_summary(ROOTDIR)