|
|
from random import uniform, randint
|
|
|
|
|
|
from bs4 import BeautifulSoup
|
|
|
from loguru import logger
|
|
|
|
|
|
from configs import configs
|
|
|
from .devtools import Browser
|
|
|
from .spider import Spider
|
|
|
|
|
|
PLACEHOLDER = '. '
|
|
|
_paper_config = configs['paper']
|
|
|
|
|
|
|
|
|
async def list_books(detail):
|
|
|
async with Spider() as spider:
|
|
|
await spider.login(**configs['user'])
|
|
|
courses = await spider.get_courses()
|
|
|
|
|
|
for cr in courses:
|
|
|
if detail:
|
|
|
hint = f'{cr["courseName"]} ({cr["teacherName"]})'
|
|
|
else:
|
|
|
hint = cr['courseName']
|
|
|
print(hint)
|
|
|
|
|
|
books = await spider.get_books(cr["courseId"])
|
|
|
for book in books:
|
|
|
if detail:
|
|
|
hint = f'> [{cr["courseId"]}#{book["bookId"]}] {book["bookName"]} ({book["percent"]}%)'
|
|
|
else:
|
|
|
hint = f'> {book["bookName"]}'
|
|
|
print(PLACEHOLDER + hint)
|
|
|
|
|
|
|
|
|
async def _random(spider, paper_id): # 随机的分数和学习时长
|
|
|
paper = BeautifulSoup((await spider.get_paper(paper_id))['paperData'], 'lxml-xml')
|
|
|
questions = paper.select('element[knowledge]:has(> question_type)')
|
|
|
if not questions:
|
|
|
return 100, 5
|
|
|
|
|
|
total = 0
|
|
|
score, time = 0, 0
|
|
|
for q in questions:
|
|
|
q_type = int(q.select_one('question_type').text)
|
|
|
q_score = float(q.select_one('question_score').text)
|
|
|
total += q_score
|
|
|
|
|
|
ranges = _paper_config['random-score']
|
|
|
if q_type <= len(ranges) and (r := ranges[q_type-1]) is not None:
|
|
|
if not isinstance(ranges[q_type-1], list):
|
|
|
r = [r, r]
|
|
|
score += q_score * uniform(*r)
|
|
|
else:
|
|
|
q_no = q.select_one('question_no').text
|
|
|
logger.warning(f'T{q_no}：未知的题型！')
|
|
|
if (r := _paper_config['defaults']) == 'pause':
|
|
|
score += float(input('请手动输入该题得分 [0-1]：'))
|
|
|
else: # defaults
|
|
|
score += q_score * uniform(*r)
|
|
|
|
|
|
time += randint(*_paper_config['random-time'])
|
|
|
|
|
|
return int(100 * score / total), time
|
|
|
|
|
|
|
|
|
async def _flash(course_id, book_id, spider):
|
|
|
async def dfs(node, depth=0):
|
|
|
task = node.task
|
|
|
print(PLACEHOLDER * depth + task['name'])
|
|
|
if _paper_config['skip-finished'] and task['unitStudyPercent'] == 100:
|
|
|
print(PLACEHOLDER * depth + '# Skipped')
|
|
|
return
|
|
|
if 'paperId' in task: # 如果有任务则提交
|
|
|
chapter_id = task['chapterId']
|
|
|
task_id = task['id']
|
|
|
score, time = await _random(spider, task['paperId'])
|
|
|
result = await page.submit(book_id, chapter_id, task_id, score, time, 100, user_id)
|
|
|
if result['wasThrown'] or not result['result']['value']:
|
|
|
logger.warning(f'任务 {task["name"]} 可能提交失败，请留意最终结果！')
|
|
|
for ch in node.child:
|
|
|
await dfs(ch, depth + 1)
|
|
|
|
|
|
browser = await Browser.connect()
|
|
|
page = await browser.wait_for_page(r'https?://pc\.ismartin\.com.*')
|
|
|
|
|
|
# With Spider
|
|
|
await spider.login(**configs['user'])
|
|
|
user_id = (await spider.user_info())['data']['uid']
|
|
|
|
|
|
book_type = (await spider.book_info(book_id))['bookType']
|
|
|
root = await spider.get_tasks(book_id, book_type, course_id)
|
|
|
await dfs(root)
|
|
|
|
|
|
|
|
|
async def flash_by_id(identity):
|
|
|
async with Spider() as spider:
|
|
|
await _flash(*identity.split('#'), spider)
|
|
|
|
|
|
|
|
|
async def flash_current(): # 对当前课程或书籍执行刷课
|
|
|
browser = await Browser.connect()
|
|
|
course_id, book_id = await browser.get_current()
|
|
|
async with Spider() as spider:
|
|
|
if book_id:
|
|
|
await _flash(course_id, book_id, spider)
|
|
|
else:
|
|
|
await spider.login(**configs['user'])
|
|
|
books = await spider.get_books(course_id)
|
|
|
for book in books:
|
|
|
await _flash(course_id, book['bookId'], spider)
|
|
|
|
|
|
|
|
|
async def flash_all():
|
|
|
async with Spider() as spider:
|
|
|
await spider.login(**configs['user'])
|
|
|
courses = await spider.get_courses()
|
|
|
for course in courses:
|
|
|
course_id = course['courseId']
|
|
|
books = await spider.get_books(course_id)
|
|
|
for book in books:
|
|
|
await _flash(course_id, book['bookId'], spider)
|
