modified: demo.py

modified:   initialization.py
	modified:   main.py

demo.py

@@ -1,2 +1,8 @@
-from loguru import logger
-logger.error('未知课程类型')
+from collections import *
+import json
+info = json.load(open('course_info.json','r'))
+cnt = Counter(info[k]['type'] for k in info.keys())
+print(cnt)
+cnt['必修'] = 1
+less_10 = [k for k, v in cnt.items() if v < 10]
+print(less_10)

initialization.py

@@ -3,4 +3,19 @@
 所以先进入个人中心检测各个类型课程数量是否存在小于10的,如果有，则进入首页点击一下视频一次
 '''
 # 本模块用于初始化，检测是否需要进入视频一次页以及进行进入视频一次页操作
-# 睡觉了，明天再写，哈哈。 おやすみなさいです、あしたまたね:)
+
+from DrissionPage import ChromiumPage
+from DrissionPage.common import *
+import json
+import time
+from logru import logger
+from get_info import get_info
+from collections import *
+
+# 读取课程数量信息
+get_info()
+class_info = json.load(open('course_info.json','r'))
+# 读取各个类别的课程数量
+cnt = Counter(class_info[k]['type'] for k in class_info.keys())
+# 检测是否存在小于10的课程数量
+less_10 = [k for k, v in cnt.items() if v < 10]

main.py

@@ -8,7 +8,10 @@ from DrissionPage import ChromiumPage
 from DrissionPage.common import *
 import time
 # 获取课程信息
-get_info()
+
+# 这里以后会放一个初始化模块
+
+get_info(first=False)
 # 执行刷课
 from kill_course import kill_course
 kill_course()