python實現合工大試題庫自動刷題

具體代碼參見：https://github.com/Jie-OY/python-funny

語言： python 3.5 (2.7也行)

庫：requests, re, xlrd

 

# coding= utf-8
import re
import requests
import xlrd

save_url = "http://tkkc.hfut.edu.cn/student/exam/manageExam.do?1479131327464&method=saveAnswer"
# index用於提示題目序號
index = 1
headers = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; WOW64; rv:51.0) Gecko/20100101 Firefox/41.0",
           "Host": "tkkc.hfut.edu.cn",
           "X-Requested-With": "XMLHttpRequest",
           }

ses = requests.session()
ID = input("請輸入學號\n")
Pwd = input("請輸入密碼\n")
logInfo = {
    "logname": ID,
    "password": Pwd
}
login_url = "http://tkkc.hfut.edu.cn/login.do?"
res = ses.post(login_url, data=logInfo, headers=headers)

# 用於存放excel中question,answer鍵值對的字典
result = dict()


# retries默認為2，表示嘗試次數。以防某種原因，某次連接失敗
def craw(url, retries=2):
    try:
        b = ses.post(url, headers=headers)
        b.encoding = 'utf-8'
        d = b.text
        title = re.findall(r' (.*?)","', d, re.S)[0]
        return title
    except Exception as e:
        print(e)
        if retries > 0:
            return craw(url, retries=retries - 1)
        else:
            print("get failed", index)
            return ''


# 從字典中根據題目找到並返回答案
def answer_func(t):
    return result.get(title, "Not Found")


# 將找到的答案提交給服務器
def submit(ans, id, id2, id3, id4, index, retries=2):
    dx = ["false", "false", "false", "false", "false"]
    try:
        if ans.find('A') != -1:
            dx[0] = "true"
        if ans.find('B') != -1:
            dx[1] = "true"
        if ans.find('C') != -1:
            dx[2] = "true"
        if ans.find('D') != -1:
            dx[3] = "true"
        if ans.find('E') != -1:
            dx[4] = "true"
        if ans.find('正確') != -1:
            ans = "A"
        if ans.find('錯誤') != -1:
            ans = "B"
        data2 = {"examReplyId": id3,
                 "examStudentExerciseId": id2,
                 "exerciseId": id,
                 "examId": id4,
                 "DXanswer": ans,
                 "PDanswer": ans,
                 "DuoXanswerA": dx[0],
                 "DuoXanswerB": dx[1],
                 "DuoXanswerC": dx[2],
                 "DuoXanswerD": dx[3],
                 "DuoXanswerE": dx[4]}
        body = ses.post(save_url, data=data2, headers=headers)
        wb_data = body.text
        print(wb_data, index)
    except Exception as e:
        print(e)
        if retries > 0:
            return submit(ans, id, id2, id3, id4, index, retries=retries - 1)
        else:
            print("get failed", index)
            return ''


# 此變量用於判斷用戶是否要繼續刷課
finished = 0

while finished == 0:
    start_url = input("請輸入測試頁面URL\n")
    
    myfile = xlrd.open_workbook('exercise.xls')
    lenOfXls = len(myfile.sheets())

    # 讀取XLS中的題目和答案，存進字典（將這段程序放在這，是因為當用戶有多門試題庫時，刷完一門，切換到另一門時，不用關閉程序只需切換題庫Excel即可）
    for x in range(0, lenOfXls):
        xls = myfile.sheets()[x]
        for i in range(1, xls.nrows):
            title = xls.cell(i, 0).value
            if x != 2:
                answer = xls.cell(i, 7).value
            else:
                answer = xls.cell(i, 2).value
            result[title] = answer

    body = ses.get(start_url, headers=headers)
    body.encoding = 'utf-8'
    wb_data = body.text
    # print(wb_data)

    urlId = re.findall(r'do\?(.*?)&method', start_url, re.S)[0]

    eval = re.findall(r'eval(.*?)]\);', wb_data, re.S)[0]

    examReplyId = re.findall(r'examReplyId=(.*?)&examId', wb_data, re.S)[0]

    examId = re.findall(r'<input type="hidden" name="examId" id="examId" value="(.*?)" />', wb_data, re.S)[0]

    exerciseId = re.findall(r'exerciseId":(.*?),', eval, re.S)

    examSEId = re.findall(r'examStudentExerciseId":(.*?),', eval, re.S)

    examStudentExerciseId = re.findall(r'"examStudentExerciseId":(.*?),"exerciseId"',
                                       wb_data, re.S)[0]

    print(examStudentExerciseId)
    examStudentExerciseId = int(examStudentExerciseId)

    # id對應exerciseID,id2對應examStudetExerciseId
    for id in exerciseId:
        next_url = r"http://tkkc.hfut.edu.cn/student/exam/manageExam.do?%s&method=getExerciseInfo&examReplyId=%s&exerciseId=%s&examStudentExerciseId=%d" % (
            urlId, examReplyId, id, examStudentExerciseId)
        title = craw(next_url)
        ans = answer_func(title)
        submit(ans, id, examStudentExerciseId, examReplyId, examId, index)
        # time.sleep(1)
        index += 1
        examStudentExerciseId = examStudentExerciseId + 1
    # input函數獲取到的為字符串，所以進行Type conversion
    finished = int(input("繼續請輸入0，退出請輸入1\n"))