output to specific dir

billryan · billryan · commit de92acb19a2b · 2018-03-11T22:44:42.000+08:00
diff --git a/scripts/main.py b/scripts/main.py
@@ -5,6 +5,9 @@
 import argparse
 from datetime import datetime
 
+import frontmatter
+from slugify import slugify
+
 from util import par_dir, mkdir_p
 from leetcode import Leetcode
 from ojhtml2markdown import problem2md
@@ -20,6 +23,8 @@ def curr_time():
     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',
@@ -31,8 +36,24 @@ def curr_time():
 
     ROOTDIR = par_dir(BASEDIR)
     raw_url = args.new
+    problem_md = ''
+    problem_slug = ''
     if raw_url.startswith('https://leetcode'):
         leetcode = Leetcode()
         problem = leetcode.get_problem_all(raw_url)
+        problem_slug = slugify(problem['title'], separator="_")
         problem_md = problem2md(problem)
-        print(problem_md)
+    
+    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)
diff --git a/zh-hans/math_and_bit_manipulation/palindrome_number.md b/zh-hans/math_and_bit_manipulation/palindrome_number.md
@@ -37,7 +37,7 @@ There is a more generic way of solving this problem.
 
 ## 题解1 - 循环处理首尾数字
 
-题意为判断数字是否为回文，要求为不允许使用额外空间，也就是说不能使用类似数字转字符串的方法直接判断。既然不能使用和数字等长的数组空间，那不借助数组来循环判断首尾数字是否相等总是可以的。接下来的问题就转化为怎么获取数字的首尾数字，获取整数的末尾是非常容易的，对10取模即可，那如何获取整数的首部数字呢？用当前整数除以10的幂(幂的大小和整数的宽度一样)即可。确定好初始和循环终止条件即可。
+题意为判断数字是否为回文，要求为不允许使用额外空间，也就是说不能使用类似数字转字符串的方法直接判断。既然不能使用和数字等长的数组空间，那不借助数组来循环判断首尾数字是否相等总是可以的。接下来的问题就转化为怎么获取数字的首尾数字，获取整数的末尾是非常容易的，对10取模即可，那如何获取整数的首部数字呢？用当前整数除以10的幂(幂的大小和整数的宽度一样)即可。确定好初始和循环终止条件即可。
 
 ### Java
 
@@ -64,7 +64,7 @@ class Solution {
 
 ### 源码分析
 
-对于32位整数来说，初始化最大的除数为 1000000000, 循环找出适合当前的最大的除数，随后算出首尾的数字并对其进行比对，循环退出条件为首尾不匹配或者除数为1(比对至最后一位).
+对于32位整数来说，初始化最大的除数为 1000000000, 循环找出适合当前的最大的除数，随后算出首尾的数字并对其进行比对，循环退出条件为首尾不匹配或者除数为1(比对至最后一位).
 
 ### 复杂度分析
 
@@ -102,4 +102,4 @@ class Solution {
 
 ### 复杂度分析
 
-空间复杂度为 $$O(1)$$, 时间复杂度为 $$logN$$.
+空间复杂度为 $$O(1)$$, 时间复杂度为 $$logN$$.
