From 0d7ed922511d6dfaf12b50020c9277604c14002e Mon Sep 17 00:00:00 2001
From: JesterHey <Jester_Huang@hnu.edu.cn>
Date: Tue, 12 Mar 2024 12:46:35 +0800
Subject: [PATCH] 	modified:   demo.py 	modified:   initialization.py 
 modified:   main.py

---
 demo.py           | 10 ++++++++--
 initialization.py | 17 ++++++++++++++++-
 main.py           |  5 ++++-
 3 files changed, 28 insertions(+), 4 deletions(-)

diff --git a/demo.py b/demo.py
index 0c40233..6e12363 100644
--- a/demo.py
+++ b/demo.py
@@ -1,2 +1,8 @@
-from loguru import logger
-logger.error('未知课程类型')
\ No newline at end of file
+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)
\ No newline at end of file
diff --git a/initialization.py b/initialization.py
index 0dde5f0..2de055d 100644
--- a/initialization.py
+++ b/initialization.py
@@ -3,4 +3,19 @@
 所以先进入个人中心检测各个类型课程数量是否存在小于10的,如果有，则进入首页点击一下视频一次
 '''
 # 本模块用于初始化，检测是否需要进入视频一次页以及进行进入视频一次页操作
-# 睡觉了，明天再写，哈哈。 おやすみなさいです、あしたまたね:)
\ No newline at end of file
+
+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]
\ No newline at end of file
diff --git a/main.py b/main.py
index cd6426a..ecc7d99 100644
--- a/main.py
+++ b/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()