|
|
'''
|
|
|
本模块用于获取作业的参数，包括：
|
|
|
1、关卡数
|
|
|
2、每关的参数
|
|
|
a:任务描述
|
|
|
b:编程要求
|
|
|
c:编辑器中的所有代码
|
|
|
3、课程id,实训id
|
|
|
4、写入本地json文件中
|
|
|
'''
|
|
|
|
|
|
#导入所需模块
|
|
|
import platform
|
|
|
import os
|
|
|
import json
|
|
|
import time
|
|
|
from selenium.webdriver import Chrome
|
|
|
from selenium.webdriver.chrome.options import Options
|
|
|
from selenium.webdriver.common.by import By
|
|
|
from selenium.webdriver.common.keys import Keys
|
|
|
import re
|
|
|
from lxml import etree
|
|
|
import time
|
|
|
import requests
|
|
|
from cloud import is_exist,download
|
|
|
global retry
|
|
|
retry = 0
|
|
|
#配置参数
|
|
|
opt = Options()
|
|
|
opt.add_experimental_option('detach', True)
|
|
|
opt.add_argument('--headless')
|
|
|
platf = platform.platform()
|
|
|
def is_practice(url:str) -> bool:
|
|
|
obj=re.compile(r'www.educoder.net/tasks')
|
|
|
if obj.search(url):
|
|
|
return True
|
|
|
else:
|
|
|
return False
|
|
|
def is_programming(url:str) -> bool:
|
|
|
obj=re.compile(r'www.educoder.net/myproblems')
|
|
|
if obj.search(url):
|
|
|
return True
|
|
|
else:
|
|
|
return False
|
|
|
# 另外，目前好像只有实训作业有这些参数，其他的作业例如编程作业就没有，所以先判断一下是否为实训作业，可以通过用户输入的url判断
|
|
|
# 主要是看educoder.net/后面是否有tasks，如果有，则是实训作业，否则，不是实训作业
|
|
|
#为方便main.py调用，将判断函数写入函数中，以下部分封装为函数
|
|
|
def get_parameters(url:str,user_name:str,password:str):
|
|
|
'''
|
|
|
用于获得实训作业的参数
|
|
|
url:实训网址
|
|
|
user_name:用户名
|
|
|
password:密码
|
|
|
'''
|
|
|
url = url
|
|
|
user_name = user_name
|
|
|
password = password
|
|
|
if is_practice(url):
|
|
|
#构造selenium对象
|
|
|
safari = Chrome(options=opt)
|
|
|
safari.get(url)
|
|
|
#模拟登录
|
|
|
safari.implicitly_wait(10)
|
|
|
safari.find_element(By.ID, 'login').send_keys(user_name)
|
|
|
safari.find_element(By.ID, 'password').send_keys(password)
|
|
|
safari.find_element(By.ID, 'password').send_keys(Keys.ENTER)
|
|
|
time.sleep(2)
|
|
|
#判断是否登录成功
|
|
|
try:
|
|
|
safari.find_element(By.XPATH,'//*[@id="task-left-panel"]/div[1]/a[1]')
|
|
|
except BaseException:
|
|
|
print('登录失败 请检查输入信息是否正确')
|
|
|
# 关闭浏览器
|
|
|
safari.quit()
|
|
|
# #重新调用login_ui.py
|
|
|
# if 'Windows' in platf:
|
|
|
# os.system('python login_ui.py')
|
|
|
# else:
|
|
|
# os.system('python3 login_ui.py')
|
|
|
#获取cookie，User-Agent
|
|
|
Cookie = safari.get_cookies()
|
|
|
User_Agent = safari.execute_script('return navigator.userAgent')
|
|
|
cookie = f'autologin_trustie={Cookie[1]["value"]}; _educoder_session={Cookie[0]["value"]}'
|
|
|
#先获取到shixun_id便于先判断云端文件是否存在
|
|
|
cur_url = url
|
|
|
identity = cur_url.split('/')[-1].split('?')[0]
|
|
|
id_url = f'https://data.educoder.net/api/tasks/{identity}.json?'
|
|
|
headers = {
|
|
|
'Cookie':cookie,
|
|
|
'User-Agent':User_Agent,
|
|
|
'Referer':cur_url
|
|
|
}
|
|
|
response = requests.get(url=id_url, headers=headers)
|
|
|
shixun_id = dict(response.json())['challenge']['shixun_id']
|
|
|
#判断云端文件是否存在
|
|
|
exist = is_exist(f'{shixun_id}.json')
|
|
|
if exist:
|
|
|
try:
|
|
|
print('云端文件已存在，正在下载')
|
|
|
download(f'{shixun_id}.json')
|
|
|
# 检测本地文件是否下载完成
|
|
|
while True:
|
|
|
try:
|
|
|
if os.path.exists(f'{shixun_id}.json'):
|
|
|
print('下载完成')
|
|
|
safari.quit()
|
|
|
return
|
|
|
break
|
|
|
except Exception as e:
|
|
|
print(e)
|
|
|
except Exception as e:
|
|
|
print(e)
|
|
|
else:#不存在，则继续执行本程序
|
|
|
print('云端文件不存在，正在爬取')
|
|
|
#获取关卡数
|
|
|
#点击展开关卡页面
|
|
|
time.sleep(3)
|
|
|
safari.find_element(By.XPATH,'//*[@id="task-left-panel"]/div[1]/a[1]').click()
|
|
|
#关卡数量由 class = "flex-container challenge-title space-between" 的元素数量决定
|
|
|
time.sleep(3)
|
|
|
htmltxt = safari.page_source
|
|
|
html = etree.HTML(htmltxt)
|
|
|
task_num = html.xpath('count(//*[@class="flex-container challenge-title space-between"])')
|
|
|
task_num = int(task_num)
|
|
|
print(f'关卡数量为{task_num}')
|
|
|
#回到第一关
|
|
|
time.sleep(3)
|
|
|
safari.find_element(By.XPATH,'//*[@id="task-left-panel"]/div[3]/div[3]/div/div/div/div/div[1]/div[1]/a').click()
|
|
|
#对于每一关，获取参数
|
|
|
#每一关的参数由以下元素组成：
|
|
|
'''
|
|
|
/html/body/div[1]/div/div/div/div[2]/section[1]/div[3]/div[3]/div/div/div/div/div[3]/div[1]/a
|
|
|
/html/body/div[1]/div/div/div/div[2]/section[1]/div[3]/div[3]/div/div/div/div/div[4]/div[1]/a
|
|
|
'''
|
|
|
obj1 = re.compile(r'<h3 id="任务描述">任务描述</h3><p>(?P<describe>.*?)</p>',re.S)
|
|
|
obj2 = re.compile(r'<h3 id="编程要求">编程要求</h3><p>(?P<require>.*?)</p>',re.S)
|
|
|
#初始化一个字典，用于存放所有关卡的参数
|
|
|
total = {}
|
|
|
i=1
|
|
|
try:
|
|
|
while i <= task_num:
|
|
|
cur_url=Referer = safari.current_url
|
|
|
identity = cur_url.split('/')[-1].split('?')[0]
|
|
|
id_url = f'https://data.educoder.net/api/tasks/{identity}.json?'
|
|
|
#获取课程id
|
|
|
headers = {
|
|
|
'Cookie':cookie,
|
|
|
'User-Agent':User_Agent,
|
|
|
'Referer':Referer
|
|
|
}
|
|
|
try:
|
|
|
response = requests.get(url=id_url,headers=headers)
|
|
|
challenge_id = dict(response.json())['challenge']['id']
|
|
|
shixun_id = dict(response.json())['challenge']['shixun_id']
|
|
|
except BaseException:
|
|
|
print('获取课程id失败')
|
|
|
#获取任务描述(如果存在的话)
|
|
|
time.sleep(3)
|
|
|
page_source = safari.page_source
|
|
|
describe = obj1.findall(page_source)
|
|
|
#获取编程要求(如果存在的话)
|
|
|
require = obj2.findall(page_source)
|
|
|
#获取编辑器中的代码,采用requests抓取https://data.educoder.net/api/tasks/{identity}/rep_content.json中的content中的content
|
|
|
# 然后然后，这个content是一个base64编码的字符串，需要解码
|
|
|
code = requests.get(f'https://data.educoder.net/api/tasks/{identity}/rep_content.json',headers=headers).json()['content']['content']
|
|
|
#把参数存入字典，再转换为json格式
|
|
|
task = {
|
|
|
'describe':describe[0] if len(describe) != 0 else '',
|
|
|
'require':require[0] if len(require) != 0 else '',
|
|
|
'code':code,
|
|
|
'verified': False, #这个参数是用来标记答案是否被用户认证为正确答案的，初始值为False
|
|
|
'last_modified': time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time())) #这个参数是用来标记答案最后一次被修改的时间，初始值为当前时间
|
|
|
}
|
|
|
#把每一关的参数存入总的字典中
|
|
|
total[challenge_id] = task
|
|
|
#去往下一关
|
|
|
safari.implicitly_wait(10)
|
|
|
if i == 1:
|
|
|
i += 1
|
|
|
safari.find_element(By.XPATH,f'//*[@id="task-right-panel"]/div[4]/div/div[2]/a').click()
|
|
|
elif i<task_num:
|
|
|
i += 1
|
|
|
safari.find_element(By.XPATH,f'//*[@id="task-right-panel"]/div[4]/div/div[2]/a[2]').click()
|
|
|
else:
|
|
|
i += 1
|
|
|
except Exception as e:
|
|
|
print(e)
|
|
|
print(f'{i}关参数获取参数失败')
|
|
|
#判断爬取到的代码是否存在空值或者键的数量是否与关卡数量相等，如果不相等，则说明爬取失败，需要重新爬取
|
|
|
|
|
|
|
|
|
# if len(total) != task_num:
|
|
|
# print('参数爬取失败，正在重新爬取')
|
|
|
# get_parameters(url,user_name,password)
|
|
|
# else:
|
|
|
# for i,j in total.items():
|
|
|
# if j['code'] == '':
|
|
|
# print('参数爬取失败，正在重新爬取')
|
|
|
# get_parameters(url,user_name,password)
|
|
|
|
|
|
|
|
|
#把参数写入本地json文件中，文件名字与shixun_name相同键为course_id，值为一个列表，列表中每个元素为一个字典，字典中包含每一关的参数
|
|
|
with open(f'{shixun_id}.json','w',encoding='utf-8') as f:
|
|
|
json.dump(total,f,ensure_ascii=False,indent=4)
|
|
|
print('参数爬取完成')
|
|
|
#关闭浏览器
|
|
|
safari.quit()
|
|
|
else:
|
|
|
print('这不是一个实训作业')
|
|
|
return
|
|
|
|
|
|
def get_parameters_of_programming(url:str,user_name:str,password:str):
|
|
|
'''
|
|
|
用于获得编程作业的参数
|
|
|
url:实训网址
|
|
|
user_name:用户名
|
|
|
password:密码
|
|
|
'''
|
|
|
url = url
|
|
|
user_name = user_name
|
|
|
password = password
|
|
|
# 检查是否为编程作业
|
|
|
if is_programming(url):
|
|
|
# 构造selenium对象
|
|
|
#构造selenium对象
|
|
|
safari = Chrome(options=opt)
|
|
|
safari.get(url)
|
|
|
#模拟登录
|
|
|
safari.implicitly_wait(10)
|
|
|
safari.find_element(By.ID, 'login').send_keys(user_name)
|
|
|
safari.find_element(By.ID, 'password').send_keys(password)
|
|
|
safari.find_element(By.ID, 'password').send_keys(Keys.ENTER)
|
|
|
time.sleep(2)
|
|
|
#判断是否登录成功
|
|
|
try:
|
|
|
safari.find_element(By.XPATH,'//*[@id="root"]/div/div/div/div/div/div/div/section[1]/div/div[4]')
|
|
|
except BaseException:
|
|
|
print('登录失败 请检查输入信息是否正确')
|
|
|
# 关闭浏览器
|
|
|
safari.quit()
|
|
|
# #重新调用login_ui.py
|
|
|
# if 'Windows' in platf:
|
|
|
# os.system('python login_ui.py')
|
|
|
# else:
|
|
|
# os.system('python3 login_ui.py')
|
|
|
#获取cookie，User-Agent
|
|
|
Cookie = safari.get_cookies()
|
|
|
User_Agent = safari.execute_script('return navigator.userAgent')
|
|
|
cookie = f'autologin_trustie={Cookie[1]["value"]}; _educoder_session={Cookie[0]["value"]}'
|
|
|
heders = {
|
|
|
'Cookie':cookie,
|
|
|
'User-Agent':User_Agent,
|
|
|
'Referer':url
|
|
|
}
|
|
|
# 由于编程作业往往是题库的题目组合的，没有直接编程作业的总ID，所以要先访问每个题目界面，获得相应的ID，题干，代码等参数
|
|
|
# 最终上传时，记得在要区别于已经上传的实训作业，可以在文件名前加上'pro'前缀
|
|
|
|
|
|
# 获得题目数量
|
|
|
# 展开关卡页面
|
|
|
safari.find_element(By.XPATH,'//*[@id="root"]/div/div/div/div/div/div/div/section[1]/div/div[4]').click()
|
|
|
# <div class="list___PXTsq ">
|
|
|
# 可见，关卡数量是根据class="list___PXTsq "的元素数量决定的，直接用xpath获取即可
|
|
|
time.sleep(1)
|
|
|
htmltxt = safari.page_source
|
|
|
html = etree.HTML(htmltxt)
|
|
|
task_num = html.xpath('count(//*[@class="list___PXTsq "])')
|
|
|
task_num = int(task_num) + 1
|
|
|
# 回到第一关
|
|
|
# //*[@id="educoder"]/body/div[6]/div/div[3]/div/div/div[2]/div[1]
|
|
|
# //*[@id="educoder"]/body/div[7]/div/div[3]/div/div/div[2]/div[2]
|
|
|
for i in range(1,10):
|
|
|
try:
|
|
|
if safari.find_element(By.XPATH,f'//*[@id="educoder"]/body/div[{i}]/div/div[3]/div/div/div[2]/div[1]'):
|
|
|
safari.find_element(By.XPATH,f'//*[@id="educoder"]/body/div[{i}]/div/div[3]/div/div/div[2]/div[1]').click()
|
|
|
break
|
|
|
except:
|
|
|
continue
|
|
|
i = 1
|
|
|
while i<=task_num:
|
|
|
# 对于每一关，获取参数
|
|
|
cur_url = safari.current_url
|
|
|
identity = cur_url.split('/')[-1].split('?')[0]
|
|
|
resp = requests.get(f'https://data.educoder.net/api/myproblems/{identity}.json',headers=heders)
|
|
|
# 提取resp中的题目id，题干，代码等参数
|
|
|
problem_data = resp.json()
|
|
|
# 初始化一个字典，用于存放所有关卡的参数
|
|
|
total = {}
|
|
|
# 题目id
|
|
|
pro_id = problem_data['hack']['id']
|
|
|
# 获取id后，判断云端是否存在该文件，如果存在，则跳过，如果不存在，则继续执行本程序
|
|
|
exist = is_exist(f'pro_{pro_id}.json')
|
|
|
if exist:
|
|
|
print('云端文件已存在，正在下载')
|
|
|
download(f'pro_{pro_id}.json')
|
|
|
print(f'pro_{pro_id}.json下载完成')
|
|
|
continue
|
|
|
else:
|
|
|
print('云端文件不存在，正在爬取')
|
|
|
# 题干
|
|
|
describe = problem_data['hack']['description']
|
|
|
# 代码(base64编码)
|
|
|
code = problem_data['hack']['code']
|
|
|
total = {
|
|
|
'id':pro_id,
|
|
|
'describe':describe,
|
|
|
'code':code,
|
|
|
'verified': False, #这个参数是用来标记答案是否被用户认证为正确答案的，初始值为False(暂时没啥用)
|
|
|
'last_modified': time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time())) #这个参数是用来标记答案最后一次被修改的时间，初始值为当前时间
|
|
|
}
|
|
|
# 写入本地json，命名为pro_pro_id.json
|
|
|
with open(f'pro_{pro_id}.json','w',encoding='utf-8') as f:
|
|
|
json.dump(total,f,ensure_ascii=False)
|
|
|
print(f'{pro_id}完成')
|
|
|
# 去往下一关
|
|
|
try:
|
|
|
i+=1
|
|
|
safari.implicitly_wait(10)
|
|
|
safari.find_element(By.ID,'oj-next').click()
|
|
|
time.sleep(2)
|
|
|
except BaseException:
|
|
|
print('参数爬取完成') # 完成后，本地会有若干个pro_pro_id.json文件
|
|
|
safari.quit()
|
|
|
else:
|
|
|
print('不是编程作业')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
url = 'https://www.educoder.net/myproblems/9kwnlzvcegsa?type=1'
|
|
|
user_name = 'hnu202311020126'
|
|
|
password = 'hzy123456'
|
|
|
get_parameters_of_programming(url,user_name,password)
|
|
|
print('结束测试') |
