liusoon
diff --git a/‎leetcode_generate.py
Lines changed: 14 additions & 12 deletions b/‎leetcode_generate.py
Lines changed: 14 additions & 12 deletions
@@ -6,20 +6,22 @@
 # Usage:  Leetcode solution downloader and auto generate readme
 #
 import requests
-import configparser
 import os
+import configparser
 import json
 import time
 import datetime
 import re
 import sys
 import html
 
+from pathlib import Path
 from selenium import webdriver
 from collections import namedtuple, OrderedDict
 
-HOME = os.getcwd()
-CONFIG_FILE = os.path.join(HOME, 'config.cfg')
+HOME = Path.cwd()
+SOLUTION_FOLDER = Path.joinpath(HOME, 'solutions')
+CONFIG_FILE = Path.joinpath(HOME, 'config.cfg')
 COOKIE_PATH = 'cookies.json'
 BASE_URL = 'https://leetcode.com'
 # If you have proxy, change PROXIES below
@@ -85,8 +87,8 @@ def rep_unicode_in_code(code):
 
 
 def check_and_make_dir(dirname):
-    if not os.path.exists(dirname):
-        os.mkdir(dirname)
+    if not Path.exists(dirname):
+        Path.mkdir(dirname)
 
 
 ProgLang = namedtuple('ProgLang', ['language', 'ext', 'annotation'])
@@ -292,7 +294,7 @@ def _generate_items_from_api(self, json_data):
     def is_login(self):
         """ validate if the cookie exists and not overtime """
         api_url = self.base_url + '/api/problems/algorithms/'  # NOQA
-        if not os.path.exists(COOKIE_PATH):
+        if not Path.exists(COOKIE_PATH):
             return False
 
         with open(COOKIE_PATH, 'r') as f:
@@ -462,15 +464,15 @@ def _download_code_by_quiz(self, quiz):
             )
             return
 
-        dirname = '{id}-{title}'.format(id=str(qid).zfill(3), title=qtitle)
-        print('begin download ' + dirname)
-        check_and_make_dir(dirname)
-        path = os.path.join(HOME, dirname)
+        qname = '{id}-{title}'.format(id=str(qid).zfill(3), title=qtitle)
+        print('begin download ' + qname)
+        path = Path.joinpath(SOLUTION_FOLDER, ,qname)
+        check_and_make_dir(path)
         for slt in slts:
             fname = '{title}.{ext}'.format(
                 title=qtitle, ext=self.prolangdict[slt['lang']].ext
             )
-            filename = os.path.join(path, fname)
+            filename = Path.joinpath(path, fname)
             content = self._get_code_with_anno(slt)
             import codecs
 
@@ -549,7 +551,7 @@ def write_readme(self):
                 language = ':lock:'
             else:
                 if item.solutions:
-                    dirname = '{id}-{title}'.format(
+                    dirname = 'solutions/{id}-{title}'.format(
                         id=str(item.question_id).zfill(3),
                         title=item.question__title_slug,
                     )