126 lines
4.4 KiB
Python
126 lines
4.4 KiB
Python
import os
|
||
import configparser
|
||
import requests
|
||
|
||
class Ipgw_login(object):
|
||
def __init__(self):
|
||
self.stu_ID = ""
|
||
self.stu_password = ""
|
||
self.login_url = 'https://pass.neu.edu.cn/tpass/login?service=http://ipgw.neu.edu.cn/srun_portal_sso?ac_id={}'
|
||
|
||
|
||
def config(self,Fname="config.ini"):
|
||
# 获取当前的路径
|
||
filepath = os.path.split(os.path.realpath(__file__))[0]
|
||
filename = os.path.join(filepath, Fname)
|
||
# print(filename)
|
||
|
||
# 获取账号，密码
|
||
conf = configparser.ConfigParser()
|
||
conf.read(filename)
|
||
|
||
self.stu_ID = conf.get("info","StudentID")
|
||
self.stu_password = conf.get("info","password")
|
||
|
||
# print(self.stu_ID,self.stu_password)
|
||
|
||
|
||
def login_with_acid(self,ac_id):
|
||
# 访问统一登录获取lt
|
||
session = requests.Session()
|
||
|
||
get_pass_page = session.get(self.login_url.format(ac_id))
|
||
|
||
# 用于更新cookie:mysession
|
||
temp = session.get("https://ipgw.neu.edu.cn/checkacid")
|
||
|
||
if get_pass_page.status_code != 200:
|
||
return False, f'访问pass.neu.edu.cn失败，状态码：{get_pass_page.status_code}'
|
||
|
||
# text = ...<input type="hidden" id="lt" name="lt" value="LT-29360-**********-tpass" />\r\n\t\t\t
|
||
text = get_pass_page.text
|
||
# print(text)
|
||
# 获取lt (login token)
|
||
target = '<input type="hidden" id="lt" name="lt" value="'
|
||
# half = LT-29360-**********-tpass" />\r\n\t\t\t
|
||
half = text[text.index(target) + len(target):]
|
||
# lt = LT-29360-**********-tpass
|
||
lt = half[:half.index('"')]
|
||
|
||
# print(lt)
|
||
|
||
# 获取execution
|
||
target = '<input type="hidden" name="execution" value="'
|
||
half = text[text.index(target) + len(target):]
|
||
execution = half[:half.index('"')]
|
||
|
||
# 拼接rsa
|
||
rsa = self.stu_ID + self.stu_password + lt
|
||
ul = len(self.stu_ID)
|
||
pl = len(self.stu_password)
|
||
|
||
# 获取用于sso链接，其中含有ticket
|
||
get_sso_href = session.post(self.login_url.format(ac_id),
|
||
allow_redirects=False, # 禁用转跳
|
||
data={'rsa': rsa,
|
||
'ul': ul,
|
||
'pl': pl,
|
||
'lt': lt,
|
||
'execution': execution,
|
||
'_eventId': 'submit'})
|
||
|
||
if get_sso_href.status_code != 302:
|
||
return False, f'访问pass.neu.edu.cn失败，状态码：{get_sso_href.status_code}'
|
||
else:
|
||
# 重定向url
|
||
# print(get_pass_page.headers)
|
||
href = dict(get_sso_href.headers).get("Location")
|
||
|
||
session.headers.update({
|
||
'Proxy-Connection':'keep-alive'
|
||
})
|
||
|
||
sso_login = session.get(href.replace('http://ipgw.neu.edu.cn/','http://ipgw.neu.edu.cn/v1/'))
|
||
|
||
if 'success' in sso_login.text:
|
||
return True, None
|
||
else:
|
||
return False, f'sso登录返回{sso_login.text}'
|
||
|
||
|
||
def login(self):
|
||
"""
|
||
登录ipgw，要点：
|
||
① 以get方式访问https://pass.neu.edu.cn页面，响应中含有lt (login token)
|
||
② 以post方式访问https://pass.neu.edu.cn页面，提交学号密码lt等参数，响应中含有用于单点登录(sso)的ticket
|
||
③ 访问http://ipgw.neu.edu.cn/v1/srun_portal_sso?ac_id=xx&ticket=xxx，提交ticket即可上网
|
||
其中使用无线网时ac_id=15，使用有线连接时ac_id=1
|
||
|
||
:param student_id: 学号
|
||
:param password: 密码
|
||
:return: (bool,reason) 若第一项为True则登录成功，否则登录失败，此时第二项会说明原因
|
||
"""
|
||
wifi_acid = 15 # 无线网登录
|
||
lan_acid = 1 # 有线网登录
|
||
|
||
results = {}
|
||
for acid in [wifi_acid, lan_acid]:
|
||
print("acid:",acid)
|
||
|
||
result, message = self.login_with_acid(acid)
|
||
if result:
|
||
return True, message
|
||
results[acid] = message
|
||
|
||
# print(results)
|
||
return False, results
|
||
|
||
def main():
|
||
_login = Ipgw_login()
|
||
_login.config()
|
||
result, message = _login.login()
|
||
# print(result, message)
|
||
|
||
if __name__ == '__main__':
|
||
main()
|