add MAX_DIGIT_LEN

bonfy · bonfy · commit 48a562c4d5d8 · 2019-04-08T13:03:42.000+08:00
diff --git a/leetcode_generate.py b/leetcode_generate.py
@@ -20,6 +20,7 @@
 from collections import namedtuple, OrderedDict
 
 HOME = Path.cwd()
+MAX_DIGIT_LEN = 4 # 1000+ PROBLEMS
 SOLUTION_FOLDER_NAME = 'solutions'
 SOLUTION_FOLDER = Path.joinpath(HOME, SOLUTION_FOLDER_NAME)
 CONFIG_FILE = Path.joinpath(HOME, 'config.cfg')
@@ -466,7 +467,7 @@ def _download_code_by_quiz(self, quiz):
             )
             return
 
-        qname = '{id}-{title}'.format(id=str(qid).zfill(3), title=qtitle)
+        qname = '{id}-{title}'.format(id=str(qid).zfill(MAX_DIGIT_LEN), title=qtitle)
         print('begin download ' + qname)
         path = Path.joinpath(SOLUTION_FOLDER, qname)
         check_and_make_dir(path)
@@ -555,7 +556,7 @@ def write_readme(self):
                 if item.solutions:
                     dirname = '{folder}/{id}-{title}'.format(
                         folder=SOLUTION_FOLDER_NAME,
-                        id=str(item.question_id).zfill(3),
+                        id=str(item.question_id).zfill(MAX_DIGIT_LEN),
                         title=item.question__title_slug,
                     )
                     language = ''
