fix none new parameter

billryan · billryan · commit 750174bd725b · 2018-07-25T01:07:27.000+08:00
diff --git a/scripts/main.py b/scripts/main.py
@@ -11,6 +11,7 @@
 from util import par_dir, mkdir_p
 from leetcode import Leetcode
 from lintcode import Lintcode
+from summary import update_summary
 from ojhtml2markdown import problem2md
 
 BASEDIR = os.path.abspath(os.path.dirname(__file__))
@@ -36,19 +37,20 @@ def curr_time():
     print('Called with arguments: {}'.format(args))
 
     ROOTDIR = par_dir(BASEDIR)
-    raw_url = args.new
-    problem_md = ''
-    problem_slug = ''
-    xxxcode = None
-    convert_desc = True
-    if raw_url.startswith('https://leetcode'):
-        xxxcode = Leetcode()
-    elif raw_url.startswith('https://www.lintcode.com'):
-        xxxcode = Lintcode()
-        convert_desc = False
-    problem = xxxcode.get_problem_all(raw_url)
-    problem_slug = slugify(problem['title'], separator="_")
-    problem_md = problem2md(problem, convert_desc)
+    if args.new:
+        raw_url = args.new
+        problem_md = ''
+        problem_slug = ''
+        xxxcode = None
+        convert_desc = True
+        if raw_url.startswith('https://leetcode'):
+            xxxcode = Leetcode()
+        elif raw_url.startswith('https://www.lintcode.com'):
+            xxxcode = Lintcode()
+            convert_desc = False
+        problem = xxxcode.get_problem_all(raw_url)
+        problem_slug = slugify(problem['title'], separator="_")
+        problem_md = problem2md(problem, convert_desc)
 
     if args.dir:
         post_dir = os.path.join(ROOTDIR, args.dir)
