update dir structure

leetcode_helper.py

@@ -3,6 +3,13 @@ import re
 import shutil
 import json
 import uuid
+import argparse
+import collections
+parser = argparse.ArgumentParser()
+parser.add_argument("--run", type=str, help="Decide to run which function")
+args = parser.parse_args()
+helper_function = args.run
+
 
 def get_files_path(file_dir, filetype='.txt'):
     """得到文件夹下的所有.txt文件的路径
@@ -175,5 +182,44 @@ def count_tag_class():
     print(tags_set)
 
 
-leetcode_helper_update_md()
-leetcode_helper_update_config()
\ No newline at end of file
+def count_exercises():
+    dirs = ['data/2.算法中阶', 'data/3.算法高阶']
+    exercises_ids = []
+    for dir in dirs:
+        dirs_ = os.listdir(dir)
+        algo_floor_dirs = []
+        for algo_floor_dir in dirs_:
+            leetcode_class_dir = os.path.join(dir, algo_floor_dir)
+            if os.path.isdir(leetcode_class_dir):
+                algo_floor_dirs.append(leetcode_class_dir)
+        exercises_dirs = []
+        for algo_floor_dir in algo_floor_dirs:
+            exercises_dirs_ = os.listdir(algo_floor_dir)
+            for exercises_dir_ in exercises_dirs_:
+                exercises_dir = os.path.join(algo_floor_dir, exercises_dir_)
+                if os.path.isdir(exercises_dir):
+                    exercises_dirs.append(exercises_dir)
+        for exercises_dir in exercises_dirs:
+            exercises_id = int(exercises_dir.split('/')[-1].split('_')[0])
+            exercises_ids.append(exercises_id)
+    try:
+        assert len(set(exercises_ids)) == len(exercises_ids)
+    except:
+        print(collections.Counter(exercises_ids))
+        print('------分割线-------')
+    dst_exercises_ids = [i for i in range(200)]
+    lacked_id = set(dst_exercises_ids) - set(exercises_ids)
+    print(lacked_id)
+        
+
+
+
+
+if helper_function == 'count_tag_class':
+    count_tag_class()
+if helper_function == 'count_exercises':
+    count_exercises()
+
+
+# leetcode_helper_update_md()
+# leetcode_helper_update_config()
\ No newline at end of file