-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathcheck.py
103 lines (90 loc) · 4.58 KB
/
check.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
# -*- encoding: utf-8 -*-
from json import loads
from enum import Enum
from rich import print
from rich.table import Table
from datetime import datetime
from selenium.webdriver.common.by import By
from custom.xuexi_chrome import XuexiChrome
class CheckResType(Enum):
NULL = 0
ARTICLE = 1
VIDEO = 2
ARTICLE_AND_VIDEO = 3
DAILY_EXAM = 4
WEEKLY_EXAM = 5
SPECIAL_EXAM = 6
def check_task(browser: XuexiChrome):
"""
检查任务项并返回给主程序
:param browser: browser
:return: CheckResType：任务类型
"""
# table = PrettyTable(["每日登录", "选读文章", "视频数量", "视频时长", "每日答题", "每周答题", "专项答题", "今日累计积分", "成长总积分"])
table = Table(show_header=True, header_style="bold black",title_justify='center')
table.add_column("每日登录", justify='center')
table.add_column("选读文章", justify='center')
table.add_column("视听学习", justify='center')
table.add_column("每日答题", justify='center')
table.add_column("今日累积积分", justify='center')
table.add_column("成长总积分", justify='center')
table_row = []
settings_path = 'data/settings.json'
with open(settings_path, 'r', encoding='utf-8') as f:
settings = f.read()
settings = loads(settings)
exam_temp_path = './data/exam_temp.json'
with open(exam_temp_path, 'r', encoding='utf-8') as f:
exam_temp = f.read()
exam_temp = loads(exam_temp)
res = CheckResType.NULL
browser.xuexi_get('https://www.xuexi.cn/index.html')
browser.xuexi_get('https://pc.xuexi.cn/points/my-points.html')
# 获取各任务项底部按钮
task_buttons = browser.find_elements(by=By.CLASS_NAME, value='big')
# 获取各任务项积分
scores = browser.find_elements(by=By.CLASS_NAME, value='my-points-card-text')
for score in scores:
table_row.append(score.text.strip())
# 今日积分
today_points = browser.find_elements(by=By.CLASS_NAME, value='my-points-points')[1]
table_row.append(today_points.text.strip())
# 总积分
all_points = browser.find_elements(by=By.CLASS_NAME, value='my-points-points')[0]
table_row.append(all_points.text.strip())
# 打印表格
table.add_row(table_row[0],
table_row[1],
table_row[2],
table_row[3],
table_row[4],
table_row[5])
print(table)
if settings['浏览文章'] == "true" and scores[1].text != '12分/12分':
res = CheckResType.ARTICLE
if settings['观看视频'] == "true" and (scores[2].text != '12分/12分' ):
if res == CheckResType.ARTICLE:
res = CheckResType.ARTICLE_AND_VIDEO
else:
res = CheckResType.VIDEO
# 检查设置文件
if settings['自动答题'] != 'true':
return res
day_of_week = str(datetime.now().isoweekday())
if settings['每日答题'] == 'true' and res == CheckResType.NULL and task_buttons[3].text != '已完成':
if settings['答题时间设置']['是否启用(关闭则每天都答题)'] != 'true' or (settings['答题时间设置']['是否启用(关闭则每天都答题)'] == 'true' and day_of_week in settings['答题时间设置']['答题类型(数字代表星期几)']['每日答题']):
res = CheckResType.DAILY_EXAM
special_exam_path = './data/special_exam.json'
date = datetime.now().strftime('%Y%m%d')
# judge file exists
with open(special_exam_path, 'r', encoding='utf-8') as f:
special_exam = f.read()
special_exam = loads(special_exam)
# 专项答题
if exam_temp['SPECIAL_EXAM'] == 'true' and date not in special_exam and settings['专项答题'] == 'true' and res == CheckResType.NULL:
if settings['答题时间设置']['是否启用(关闭则每天都答题)'] != 'true' or (settings['答题时间设置']['是否启用(关闭则每天都答题)'] == 'true' and day_of_week in settings['答题时间设置']['答题类型(数字代表星期几)']['专项答题']):
res = CheckResType.SPECIAL_EXAM
# if exam_temp['WEEKLY_EXAM'] == 'true' and settings['每周答题'] == 'true' and res == CheckResType.NULL and len(task_buttons) >=6 and task_buttons[5].text != '已完成':
# if settings['答题时间设置']['是否启用(关闭则每天都答题)'] != 'true' or (settings['答题时间设置']['是否启用(关闭则每天都答题)'] == 'true' and day_of_week in settings['答题时间设置']['答题类型(数字代表星期几)']['每周答题']):
# res = CheckResType.WEEKLY_EXAM
return res