-
Notifications
You must be signed in to change notification settings - Fork 10
/
Copy pathyzuCourseBot.py
234 lines (192 loc) · 9.02 KB
/
yzuCourseBot.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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
'''
Date : 2019/09
Author: Doem
E-mail: [email protected]
'''
import os
import cv2
import time
import requests
import numpy as np
import configparser
from bs4 import BeautifulSoup
from keras.models import load_model
class CourseBot:
def __init__(self, account, password):
self.account = account
self.password = password
self.coursesDB = {}
# for keras
self.model = load_model('model.h5')
self.n_classes = '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ'
# for requests
self.session = requests.Session()
self.session.headers['User-Agent'] = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/74.0.3729.169 Safari/537.36'
self.loginUrl = 'https://isdna1.yzu.edu.tw/CnStdSel/Index.aspx'
self.captchaUrl = 'https://isdna1.yzu.edu.tw/CnStdSel/SelRandomImage.aspx'
self.courseListUrl = 'https://isdna1.yzu.edu.tw/CnStdSel/SelCurr/CosList.aspx'
self.courseSelectUrl = 'https://isdna1.yzu.edu.tw/CnStdSel/SelCurr/CurrMainTrans.aspx?mSelType=SelCos&mUrl='
self.loginPayLoad = {
'__VIEWSTATE': '',
'__VIEWSTATEGENERATOR': '',
'__EVENTVALIDATION': '',
'DPL_SelCosType': '',
'Txt_User': self.account,
'Txt_Password': self.password,
'Txt_CheckCode': '',
'btnOK': '確定'
}
self.selectPayLoad = {}
def predict(self, img):
prediction = self.model.predict(np.array([img]))
predicStr = ""
for pred in prediction:
predicStr += self.n_classes[np.argmax(pred[0])]
return predicStr
def captchaOCR(self):
captchaImg = cv2.imread('captcha.png') / 255.0
return self.predict(captchaImg)
# login into system and get session
def login(self):
while True:
# clear Session object
self.session.cookies.clear()
# download and recognize captch
with self.session.get(self.captchaUrl, stream= True) as captchaHtml:
with open('captcha.png', 'wb') as img:
img.write(captchaHtml.content)
captcha = self.captchaOCR()
# get login data
loginHtml = self.session.get(self.loginUrl)
# check if system is open
if '選課系統尚未開放!' in loginHtml.text:
self.log('選課系統尚未開放!')
continue
# use BeautifulSoup to parse html
parser = BeautifulSoup(loginHtml.text, 'lxml')
# update login payload
self.loginPayLoad['__VIEWSTATE'] = parser.select("#__VIEWSTATE")[0]['value']
self.loginPayLoad['__VIEWSTATEGENERATOR'] = parser.select("#__VIEWSTATEGENERATOR")[0]['value']
self.loginPayLoad['__EVENTVALIDATION'] = parser.select("#__EVENTVALIDATION")[0]['value']
self.loginPayLoad['DPL_SelCosType'] = parser.select("#DPL_SelCosType option")[1]['value']
self.loginPayLoad['Txt_CheckCode'] = captcha
result = self.session.post(self.loginUrl, data= self.loginPayLoad)
if ("parent.location ='SelCurr.aspx?Culture=zh-tw'" in result.text): #成功登入訊息可能一直改，挑個不太能改的
self.log('Login Successful! {}'.format(captcha))
break
elif ("資料庫發生異常" in result.text): # 僅比較成功登入及帳號密碼錯誤的訊息，不確定是否還有其他種情況也符合這個條件
self.log('帳號或密碼錯誤，請重新確認。')
elif ("您未在此階段選課時程之內!請於時程內選課!!" in result.text):
self.log('您未在此階段選課時程之內!請於時程內選課!!')
else:
self.log("Login Failed, Re-try!")
continue
exit(0)
def getCourseDB(self, depts):
for dept in depts:
# use BeautifulSoup to parse html
html = self.session.get(self.courseListUrl)
if "異常登入" in html.text:
self.log("異常登入，休息10分鐘!")
time.sleep(600) # sleep 10 min
continue
parser = BeautifulSoup(html.text, 'lxml')
self.selectPayLoad[dept] = {
'__EVENTTARGET': 'DPL_Degree',
'__EVENTARGUMENT': '',
'__LASTFOCUS': '',
'__VIEWSTATE': parser.select("#__VIEWSTATE")[0]['value'],
'__VIEWSTATEGENERATOR': parser.select("#__VIEWSTATEGENERATOR")[0]['value'],
'__VIEWSTATEENCRYPTED': '',
'__EVENTVALIDATION': parser.select("#__EVENTVALIDATION")[0]['value'],
'Hidden1': '',
'Hid_SchTime': '',
'DPL_DeptName': dept,
'DPL_Degree': '6',
}
# use BeautifulSoup to parse html
html = self.session.post(self.courseListUrl, data= self.selectPayLoad[dept])
if "Error" in html.text:
self.log('Wrong coursesList, please check it again!')
exit(0)
parser = BeautifulSoup(html.text, 'lxml')
# parse and save courses information
courseList = parser.select("#CosListTable input")
for courseInfo in courseList:
tokens = courseInfo.attrs['name'].split(',') # SelCos,CS354,A,1,F,3,Y,Chinese,CS354,A,3 電腦與網路安全概論
key = tokens[1] + tokens[2]
courseName = '{} {}'.format(key, tokens[-1].split(' ')[1])
self.coursesDB[key] = {
'name': courseName,
'mUrl': courseInfo.attrs['name']
}
# self.log(self.coursesDB[key])
self.log('Get {} Data Completed!'.format(dept))
def selectCourses(self, coursesList, delay = 0):
while len(coursesList) > 0:
for course in coursesList.copy():
tokens = course.split(',')
dept = tokens[0]
key = tokens[1]
# check if the classID is legal
if key not in self.coursesDB:
self.log('{} is not a legal classID'.format(key))
coursesList.remove(course)
continue
# simulte click button
html = self.session.post(self.courseListUrl, data= self.selectPayLoad[dept])
parser = BeautifulSoup(html.text, 'lxml')
selectPayLoad = {
'__EVENTTARGET': '',
'__EVENTARGUMENT': '',
'__LASTFOCUS': '',
'__VIEWSTATE': parser.select("#__VIEWSTATE")[0]['value'],
'__VIEWSTATEGENERATOR': parser.select("#__VIEWSTATEGENERATOR")[0]['value'],
'__VIEWSTATEENCRYPTED': '',
'__EVENTVALIDATION': parser.select("#__EVENTVALIDATION")[0]['value'],
'Hidden1': '',
'Hid_SchTime': '',
'DPL_DeptName': dept,
'DPL_Degree': '6',
self.coursesDB[key]['mUrl'] + '.x': '0',
self.coursesDB[key]['mUrl'] + '.y': '0'
}
self.session.post(self.courseListUrl, data= selectPayLoad)
# select course
html = self.session.get(self.courseSelectUrl + self.coursesDB[key]['mUrl'] + ' ,B,')
# check if successful
parser = BeautifulSoup(html.text, 'lxml')
alertMsg = parser.select("script")[0].string.split(';')[0]
self.log('{} {}'.format(self.coursesDB[key]['name'], alertMsg[7:-2]))
if "加選訊息：" in alertMsg or "已選過" in alertMsg:
coursesList.remove(course)
elif "please log on again!" in alertMsg:
self.login()
time.sleep(delay)
def log(self, msg):
print(time.strftime("[%Y-%m-%d %H:%M:%S]", time.localtime()), msg)
if __name__ == '__main__':
configFilename = 'accounts.ini'
if not os.path.isfile(configFilename):
with open(configFilename, 'a') as f:
f.writelines(["[Default]\n", "Account= your account\n", "Password= your password"])
print('input your username and password in accounts.ini')
exit()
# get account info fomr ini config file
config = configparser.ConfigParser()
config.read(configFilename)
Account = config['Default']['Account']
Password = config['Default']['Password']
# the courses you want to select, format: '`deptId`,`courseId``classId`'
coursesList = [
'304,CS250B',
# '304,CS310A',
# '901,LS239A',
]
# Time Parameter, sleep n seconds
delay = 1
depts = set([i.split(',')[0] for i in coursesList])
myBot = CourseBot(Account, Password)
myBot.login()
myBot.getCourseDB(depts)
myBot.selectCourses(coursesList, delay)